OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <algorithm> | 5 #include <algorithm> |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/pickle.h" | 8 #include "base/pickle.h" |
9 #include "base/time.h" | 9 #include "base/time.h" |
10 #include "chrome/browser/history/history_backend.h" | 10 #include "chrome/browser/history/history_backend.h" |
(...skipping 108 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
119 | 119 |
120 std::string empty; | 120 std::string empty; |
121 WriteToEntry(cache, kMalwareURL, kMalwareHeaders, kMalwareData); | 121 WriteToEntry(cache, kMalwareURL, kMalwareHeaders, kMalwareData); |
122 WriteToEntry(cache, kLandingURL, kLandingHeaders, kLandingData); | 122 WriteToEntry(cache, kLandingURL, kLandingHeaders, kLandingData); |
123 } | 123 } |
124 | 124 |
125 void QuitUIMessageLoop() { | 125 void QuitUIMessageLoop() { |
126 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 126 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
127 BrowserThread::PostTask(BrowserThread::UI, | 127 BrowserThread::PostTask(BrowserThread::UI, |
128 FROM_HERE, | 128 FROM_HERE, |
129 MessageLoop::QuitClosure()); | 129 base::MessageLoop::QuitClosure()); |
130 } | 130 } |
131 | 131 |
132 // Lets us provide a MockURLRequestContext with an HTTP Cache we pre-populate. | 132 // Lets us provide a MockURLRequestContext with an HTTP Cache we pre-populate. |
133 // Also exposes the constructor. | 133 // Also exposes the constructor. |
134 class MalwareDetailsWrap : public MalwareDetails { | 134 class MalwareDetailsWrap : public MalwareDetails { |
135 public: | 135 public: |
136 MalwareDetailsWrap( | 136 MalwareDetailsWrap( |
137 SafeBrowsingUIManager* ui_manager, | 137 SafeBrowsingUIManager* ui_manager, |
138 WebContents* web_contents, | 138 WebContents* web_contents, |
139 const SafeBrowsingUIManager::UnsafeResource& unsafe_resource, | 139 const SafeBrowsingUIManager::UnsafeResource& unsafe_resource, |
(...skipping 67 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
207 return lhs->id() < rhs->id(); | 207 return lhs->id() < rhs->id(); |
208 } | 208 } |
209 | 209 |
210 std::string WaitForSerializedReport(MalwareDetails* report) { | 210 std::string WaitForSerializedReport(MalwareDetails* report) { |
211 BrowserThread::PostTask( | 211 BrowserThread::PostTask( |
212 BrowserThread::IO, | 212 BrowserThread::IO, |
213 FROM_HERE, | 213 FROM_HERE, |
214 base::Bind(&MalwareDetails::FinishCollection, report)); | 214 base::Bind(&MalwareDetails::FinishCollection, report)); |
215 // Wait for the callback (SendSerializedMalwareDetails). | 215 // Wait for the callback (SendSerializedMalwareDetails). |
216 DVLOG(1) << "Waiting for SendSerializedMalwareDetails"; | 216 DVLOG(1) << "Waiting for SendSerializedMalwareDetails"; |
217 MessageLoop::current()->Run(); | 217 base::MessageLoop::current()->Run(); |
218 return ui_manager_->GetSerialized(); | 218 return ui_manager_->GetSerialized(); |
219 } | 219 } |
220 | 220 |
221 HistoryService* history_service() { | 221 HistoryService* history_service() { |
222 return HistoryServiceFactory::GetForProfile(profile(), | 222 return HistoryServiceFactory::GetForProfile(profile(), |
223 Profile::EXPLICIT_ACCESS); | 223 Profile::EXPLICIT_ACCESS); |
224 } | 224 } |
225 | 225 |
226 protected: | 226 protected: |
227 void InitResource(UnsafeResource* resource, | 227 void InitResource(UnsafeResource* resource, |
(...skipping 186 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
414 child_node.url = GURL(kDOMChildURL); | 414 child_node.url = GURL(kDOMChildURL); |
415 child_node.tag_name = "iframe"; | 415 child_node.tag_name = "iframe"; |
416 child_node.parent = GURL(kDOMParentURL); | 416 child_node.parent = GURL(kDOMParentURL); |
417 params.push_back(child_node); | 417 params.push_back(child_node); |
418 SafeBrowsingHostMsg_MalwareDOMDetails_Node parent_node; | 418 SafeBrowsingHostMsg_MalwareDOMDetails_Node parent_node; |
419 parent_node.url = GURL(kDOMParentURL); | 419 parent_node.url = GURL(kDOMParentURL); |
420 parent_node.children.push_back(GURL(kDOMChildURL)); | 420 parent_node.children.push_back(GURL(kDOMChildURL)); |
421 params.push_back(parent_node); | 421 params.push_back(parent_node); |
422 report->OnReceivedMalwareDOMDetails(params); | 422 report->OnReceivedMalwareDOMDetails(params); |
423 | 423 |
424 MessageLoop::current()->RunUntilIdle(); | 424 base::MessageLoop::current()->RunUntilIdle(); |
425 | 425 |
426 std::string serialized = WaitForSerializedReport(report); | 426 std::string serialized = WaitForSerializedReport(report); |
427 ClientMalwareReportRequest actual; | 427 ClientMalwareReportRequest actual; |
428 actual.ParseFromString(serialized); | 428 actual.ParseFromString(serialized); |
429 | 429 |
430 ClientMalwareReportRequest expected; | 430 ClientMalwareReportRequest expected; |
431 expected.set_malware_url(kMalwareURL); | 431 expected.set_malware_url(kMalwareURL); |
432 expected.set_page_url(kLandingURL); | 432 expected.set_page_url(kLandingURL); |
433 expected.set_referrer_url(""); | 433 expected.set_referrer_url(""); |
434 | 434 |
(...skipping 111 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
546 BrowserThread::PostTask( | 546 BrowserThread::PostTask( |
547 BrowserThread::IO, FROM_HERE, | 547 BrowserThread::IO, FROM_HERE, |
548 base::Bind(&FillCache, | 548 base::Bind(&FillCache, |
549 make_scoped_refptr(profile()->GetRequestContext()))); | 549 make_scoped_refptr(profile()->GetRequestContext()))); |
550 | 550 |
551 // The cache collection starts after the IPC from the DOM is fired. | 551 // The cache collection starts after the IPC from the DOM is fired. |
552 std::vector<SafeBrowsingHostMsg_MalwareDOMDetails_Node> params; | 552 std::vector<SafeBrowsingHostMsg_MalwareDOMDetails_Node> params; |
553 report->OnReceivedMalwareDOMDetails(params); | 553 report->OnReceivedMalwareDOMDetails(params); |
554 | 554 |
555 // Let the cache callbacks complete | 555 // Let the cache callbacks complete |
556 MessageLoop::current()->RunUntilIdle(); | 556 base::MessageLoop::current()->RunUntilIdle(); |
557 | 557 |
558 DVLOG(1) << "Getting serialized report"; | 558 DVLOG(1) << "Getting serialized report"; |
559 std::string serialized = WaitForSerializedReport(report); | 559 std::string serialized = WaitForSerializedReport(report); |
560 ClientMalwareReportRequest actual; | 560 ClientMalwareReportRequest actual; |
561 actual.ParseFromString(serialized); | 561 actual.ParseFromString(serialized); |
562 | 562 |
563 ClientMalwareReportRequest expected; | 563 ClientMalwareReportRequest expected; |
564 expected.set_malware_url(kMalwareURL); | 564 expected.set_malware_url(kMalwareURL); |
565 expected.set_page_url(kLandingURL); | 565 expected.set_page_url(kLandingURL); |
566 expected.set_referrer_url(""); | 566 expected.set_referrer_url(""); |
(...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
617 ui_manager_.get(), web_contents(), resource, | 617 ui_manager_.get(), web_contents(), resource, |
618 profile()->GetRequestContext()); | 618 profile()->GetRequestContext()); |
619 | 619 |
620 // No call to FillCache | 620 // No call to FillCache |
621 | 621 |
622 // The cache collection starts after the IPC from the DOM is fired. | 622 // The cache collection starts after the IPC from the DOM is fired. |
623 std::vector<SafeBrowsingHostMsg_MalwareDOMDetails_Node> params; | 623 std::vector<SafeBrowsingHostMsg_MalwareDOMDetails_Node> params; |
624 report->OnReceivedMalwareDOMDetails(params); | 624 report->OnReceivedMalwareDOMDetails(params); |
625 | 625 |
626 // Let the cache callbacks complete | 626 // Let the cache callbacks complete |
627 MessageLoop::current()->RunUntilIdle(); | 627 base::MessageLoop::current()->RunUntilIdle(); |
628 | 628 |
629 DVLOG(1) << "Getting serialized report"; | 629 DVLOG(1) << "Getting serialized report"; |
630 std::string serialized = WaitForSerializedReport(report); | 630 std::string serialized = WaitForSerializedReport(report); |
631 ClientMalwareReportRequest actual; | 631 ClientMalwareReportRequest actual; |
632 actual.ParseFromString(serialized); | 632 actual.ParseFromString(serialized); |
633 | 633 |
634 ClientMalwareReportRequest expected; | 634 ClientMalwareReportRequest expected; |
635 expected.set_malware_url(kMalwareURL); | 635 expected.set_malware_url(kMalwareURL); |
636 expected.set_page_url(kLandingURL); | 636 expected.set_page_url(kLandingURL); |
637 expected.set_referrer_url(""); | 637 expected.set_referrer_url(""); |
(...skipping 28 matching lines...) Expand all Loading... |
666 UnsafeResource resource; | 666 UnsafeResource resource; |
667 InitResource(&resource, true, GURL(kMalwareURL)); | 667 InitResource(&resource, true, GURL(kMalwareURL)); |
668 scoped_refptr<MalwareDetailsWrap> report = new MalwareDetailsWrap( | 668 scoped_refptr<MalwareDetailsWrap> report = new MalwareDetailsWrap( |
669 ui_manager_.get(), web_contents(), resource, NULL); | 669 ui_manager_.get(), web_contents(), resource, NULL); |
670 | 670 |
671 // The redirects collection starts after the IPC from the DOM is fired. | 671 // The redirects collection starts after the IPC from the DOM is fired. |
672 std::vector<SafeBrowsingHostMsg_MalwareDOMDetails_Node> params; | 672 std::vector<SafeBrowsingHostMsg_MalwareDOMDetails_Node> params; |
673 report->OnReceivedMalwareDOMDetails(params); | 673 report->OnReceivedMalwareDOMDetails(params); |
674 | 674 |
675 // Let the redirects callbacks complete. | 675 // Let the redirects callbacks complete. |
676 MessageLoop::current()->RunUntilIdle(); | 676 base::MessageLoop::current()->RunUntilIdle(); |
677 | 677 |
678 std::string serialized = WaitForSerializedReport(report); | 678 std::string serialized = WaitForSerializedReport(report); |
679 ClientMalwareReportRequest actual; | 679 ClientMalwareReportRequest actual; |
680 actual.ParseFromString(serialized); | 680 actual.ParseFromString(serialized); |
681 | 681 |
682 ClientMalwareReportRequest expected; | 682 ClientMalwareReportRequest expected; |
683 expected.set_malware_url(kMalwareURL); | 683 expected.set_malware_url(kMalwareURL); |
684 expected.set_page_url(kLandingURL); | 684 expected.set_page_url(kLandingURL); |
685 expected.set_referrer_url(""); | 685 expected.set_referrer_url(""); |
686 | 686 |
687 ClientMalwareReportRequest::Resource* pb_resource = expected.add_resources(); | 687 ClientMalwareReportRequest::Resource* pb_resource = expected.add_resources(); |
688 pb_resource->set_id(0); | 688 pb_resource->set_id(0); |
689 pb_resource->set_url(kLandingURL); | 689 pb_resource->set_url(kLandingURL); |
690 pb_resource = expected.add_resources(); | 690 pb_resource = expected.add_resources(); |
691 pb_resource->set_id(1); | 691 pb_resource->set_id(1); |
692 pb_resource->set_parent_id(2); | 692 pb_resource->set_parent_id(2); |
693 pb_resource->set_url(kMalwareURL); | 693 pb_resource->set_url(kMalwareURL); |
694 pb_resource = expected.add_resources(); | 694 pb_resource = expected.add_resources(); |
695 pb_resource->set_id(2); | 695 pb_resource->set_id(2); |
696 pb_resource->set_parent_id(3); | 696 pb_resource->set_parent_id(3); |
697 pb_resource->set_url(kSecondRedirectURL); | 697 pb_resource->set_url(kSecondRedirectURL); |
698 pb_resource = expected.add_resources(); | 698 pb_resource = expected.add_resources(); |
699 pb_resource->set_id(3); | 699 pb_resource->set_id(3); |
700 pb_resource->set_url(kFirstRedirectURL); | 700 pb_resource->set_url(kFirstRedirectURL); |
701 | 701 |
702 VerifyResults(actual, expected); | 702 VerifyResults(actual, expected); |
703 } | 703 } |
OLD | NEW |