| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "chrome/browser/safe_browsing/ui_manager.h" | 5 #include "chrome/browser/safe_browsing/ui_manager.h" |
| 6 | 6 |
| 7 #include "base/run_loop.h" | 7 #include "base/run_loop.h" |
| 8 #include "chrome/browser/safe_browsing/safe_browsing_blocking_page.h" | 8 #include "chrome/browser/safe_browsing/safe_browsing_blocking_page.h" |
| 9 #include "chrome/browser/safe_browsing/safe_browsing_service.h" | 9 #include "chrome/browser/safe_browsing/safe_browsing_service.h" |
| 10 #include "chrome/browser/safe_browsing/ui_manager.h" | 10 #include "chrome/browser/safe_browsing/ui_manager.h" |
| (...skipping 316 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 327 } | 327 } |
| 328 | 328 |
| 329 private: | 329 private: |
| 330 bool visible_security_state_changed_ = false; | 330 bool visible_security_state_changed_ = false; |
| 331 DISALLOW_COPY_AND_ASSIGN(SecurityStateWebContentsDelegate); | 331 DISALLOW_COPY_AND_ASSIGN(SecurityStateWebContentsDelegate); |
| 332 }; | 332 }; |
| 333 | 333 |
| 334 // A test blocking page that does not create windows. | 334 // A test blocking page that does not create windows. |
| 335 class TestSafeBrowsingBlockingPage : public SafeBrowsingBlockingPage { | 335 class TestSafeBrowsingBlockingPage : public SafeBrowsingBlockingPage { |
| 336 public: | 336 public: |
| 337 TestSafeBrowsingBlockingPage(SafeBrowsingUIManager* manager, | 337 TestSafeBrowsingBlockingPage(BaseSafeBrowsingUIManager* manager, |
| 338 content::WebContents* web_contents, | 338 content::WebContents* web_contents, |
| 339 const GURL& main_frame_url, | 339 const GURL& main_frame_url, |
| 340 const UnsafeResourceList& unsafe_resources) | 340 const UnsafeResourceList& unsafe_resources) |
| 341 : SafeBrowsingBlockingPage(manager, | 341 : SafeBrowsingBlockingPage(manager, |
| 342 web_contents, | 342 web_contents, |
| 343 main_frame_url, | 343 main_frame_url, |
| 344 unsafe_resources) { | 344 unsafe_resources, |
| 345 nullptr) { |
| 345 // Don't delay details at all for the unittest. | 346 // Don't delay details at all for the unittest. |
| 346 threat_details_proceed_delay_ms_ = 0; | 347 threat_details_proceed_delay_ms_ = 0; |
| 347 DontCreateViewForTesting(); | 348 DontCreateViewForTesting(); |
| 348 } | 349 } |
| 349 }; | 350 }; |
| 350 | 351 |
| 351 // A factory that creates TestSafeBrowsingBlockingPages. | 352 // A factory that creates TestSafeBrowsingBlockingPages. |
| 352 class TestSafeBrowsingBlockingPageFactory | 353 class TestSafeBrowsingBlockingPageFactory |
| 353 : public SafeBrowsingBlockingPageFactory { | 354 : public SafeBrowsingBlockingPageFactory { |
| 354 public: | 355 public: |
| 355 TestSafeBrowsingBlockingPageFactory() {} | 356 TestSafeBrowsingBlockingPageFactory() {} |
| 356 ~TestSafeBrowsingBlockingPageFactory() override {} | 357 ~TestSafeBrowsingBlockingPageFactory() override {} |
| 357 | 358 |
| 358 SafeBrowsingBlockingPage* CreateSafeBrowsingPage( | 359 SafeBrowsingBlockingPage* CreateSafeBrowsingPage( |
| 359 SafeBrowsingUIManager* delegate, | 360 BaseSafeBrowsingUIManager* delegate, |
| 360 content::WebContents* web_contents, | 361 content::WebContents* web_contents, |
| 361 const GURL& main_frame_url, | 362 const GURL& main_frame_url, |
| 362 const SafeBrowsingBlockingPage::UnsafeResourceList& unsafe_resources) | 363 const SafeBrowsingBlockingPage::UnsafeResourceList& unsafe_resources) |
| 363 override { | 364 override { |
| 364 return new TestSafeBrowsingBlockingPage(delegate, web_contents, | 365 return new TestSafeBrowsingBlockingPage(delegate, web_contents, |
| 365 main_frame_url, unsafe_resources); | 366 main_frame_url, unsafe_resources); |
| 366 } | 367 } |
| 367 }; | 368 }; |
| 368 | 369 |
| 369 } // namespace | 370 } // namespace |
| (...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 404 SimulateBlockingPageDone(resources, true); | 405 SimulateBlockingPageDone(resources, true); |
| 405 EXPECT_TRUE(delegate.visible_security_state_changed()); | 406 EXPECT_TRUE(delegate.visible_security_state_changed()); |
| 406 | 407 |
| 407 waiter.WaitForCallback(); | 408 waiter.WaitForCallback(); |
| 408 EXPECT_TRUE(waiter.callback_called()); | 409 EXPECT_TRUE(waiter.callback_called()); |
| 409 EXPECT_TRUE(waiter.proceed()); | 410 EXPECT_TRUE(waiter.proceed()); |
| 410 EXPECT_TRUE(IsWhitelisted(resource)); | 411 EXPECT_TRUE(IsWhitelisted(resource)); |
| 411 } | 412 } |
| 412 | 413 |
| 413 } // namespace safe_browsing | 414 } // namespace safe_browsing |
| OLD | NEW |