| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 | 5 |
| 6 #include "chrome/browser/prefs/pref_service.h" | 6 #include "chrome/browser/prefs/pref_service.h" |
| 7 #include "chrome/browser/profiles/profile.h" | 7 #include "chrome/browser/profiles/profile.h" |
| 8 #include "chrome/browser/safe_browsing/malware_details.h" | 8 #include "chrome/browser/safe_browsing/malware_details.h" |
| 9 #include "chrome/browser/safe_browsing/safe_browsing_blocking_page.h" | 9 #include "chrome/browser/safe_browsing/safe_browsing_blocking_page.h" |
| 10 #include "chrome/common/pref_names.h" | 10 #include "chrome/common/pref_names.h" |
| 11 #include "chrome/test/base/chrome_render_view_host_test_harness.h" |
| 11 #include "content/browser/browser_thread.h" | 12 #include "content/browser/browser_thread.h" |
| 12 #include "content/browser/renderer_host/test_render_view_host.h" | |
| 13 #include "content/browser/tab_contents/navigation_entry.h" | 13 #include "content/browser/tab_contents/navigation_entry.h" |
| 14 #include "content/browser/tab_contents/test_tab_contents.h" | 14 #include "content/browser/tab_contents/test_tab_contents.h" |
| 15 #include "content/common/view_messages.h" | 15 #include "content/common/view_messages.h" |
| 16 | 16 |
| 17 static const char* kGoogleURL = "http://www.google.com/"; | 17 static const char* kGoogleURL = "http://www.google.com/"; |
| 18 static const char* kGoodURL = "http://www.goodguys.com/"; | 18 static const char* kGoodURL = "http://www.goodguys.com/"; |
| 19 static const char* kBadURL = "http://www.badguys.com/"; | 19 static const char* kBadURL = "http://www.badguys.com/"; |
| 20 static const char* kBadURL2 = "http://www.badguys2.com/"; | 20 static const char* kBadURL2 = "http://www.badguys2.com/"; |
| 21 static const char* kBadURL3 = "http://www.badguys3.com/"; | 21 static const char* kBadURL3 = "http://www.badguys3.com/"; |
| 22 | 22 |
| (...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 60 | 60 |
| 61 virtual SafeBrowsingBlockingPage* CreateSafeBrowsingPage( | 61 virtual SafeBrowsingBlockingPage* CreateSafeBrowsingPage( |
| 62 SafeBrowsingService* service, | 62 SafeBrowsingService* service, |
| 63 TabContents* tab_contents, | 63 TabContents* tab_contents, |
| 64 const SafeBrowsingBlockingPage::UnsafeResourceList& unsafe_resources) { | 64 const SafeBrowsingBlockingPage::UnsafeResourceList& unsafe_resources) { |
| 65 return new TestSafeBrowsingBlockingPage(service, tab_contents, | 65 return new TestSafeBrowsingBlockingPage(service, tab_contents, |
| 66 unsafe_resources); | 66 unsafe_resources); |
| 67 } | 67 } |
| 68 }; | 68 }; |
| 69 | 69 |
| 70 class SafeBrowsingBlockingPageTest : public RenderViewHostTestHarness, | 70 class SafeBrowsingBlockingPageTest : public ChromeRenderViewHostTestHarness, |
| 71 public SafeBrowsingService::Client { | 71 public SafeBrowsingService::Client { |
| 72 public: | 72 public: |
| 73 // The decision the user made. | 73 // The decision the user made. |
| 74 enum UserResponse { | 74 enum UserResponse { |
| 75 PENDING, | 75 PENDING, |
| 76 OK, | 76 OK, |
| 77 CANCEL | 77 CANCEL |
| 78 }; | 78 }; |
| 79 | 79 |
| 80 SafeBrowsingBlockingPageTest() | 80 SafeBrowsingBlockingPageTest() |
| 81 : ui_thread_(BrowserThread::UI, MessageLoop::current()), | 81 : ui_thread_(BrowserThread::UI, MessageLoop::current()), |
| 82 io_thread_(BrowserThread::IO, MessageLoop::current()) { | 82 io_thread_(BrowserThread::IO, MessageLoop::current()) { |
| 83 ResetUserResponse(); | 83 ResetUserResponse(); |
| 84 service_ = new TestSafeBrowsingService(); | 84 service_ = new TestSafeBrowsingService(); |
| 85 } | 85 } |
| 86 | 86 |
| 87 virtual void SetUp() { | 87 virtual void SetUp() { |
| 88 RenderViewHostTestHarness::SetUp(); | 88 ChromeRenderViewHostTestHarness::SetUp(); |
| 89 SafeBrowsingBlockingPage::RegisterFactory(&factory_); | 89 SafeBrowsingBlockingPage::RegisterFactory(&factory_); |
| 90 ResetUserResponse(); | 90 ResetUserResponse(); |
| 91 } | 91 } |
| 92 | 92 |
| 93 // SafeBrowsingService::Client implementation. | 93 // SafeBrowsingService::Client implementation. |
| 94 virtual void OnUrlCheckResult(const GURL& url, | 94 virtual void OnUrlCheckResult(const GURL& url, |
| 95 SafeBrowsingService::UrlCheckResult result) { | 95 SafeBrowsingService::UrlCheckResult result) { |
| 96 } | 96 } |
| 97 virtual void OnBlockingPageComplete(bool proceed) { | 97 virtual void OnBlockingPageComplete(bool proceed) { |
| 98 if (proceed) | 98 if (proceed) |
| (...skipping 486 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 585 | 585 |
| 586 EXPECT_TRUE(profile->GetPrefs()->GetBoolean( | 586 EXPECT_TRUE(profile->GetPrefs()->GetBoolean( |
| 587 prefs::kSafeBrowsingReportingEnabled)); | 587 prefs::kSafeBrowsingReportingEnabled)); |
| 588 | 588 |
| 589 // Simulate the user uncheck the report agreement checkbox. | 589 // Simulate the user uncheck the report agreement checkbox. |
| 590 sb_interstitial->SetReportingPreference(false); | 590 sb_interstitial->SetReportingPreference(false); |
| 591 | 591 |
| 592 EXPECT_FALSE(profile->GetPrefs()->GetBoolean( | 592 EXPECT_FALSE(profile->GetPrefs()->GetBoolean( |
| 593 prefs::kSafeBrowsingReportingEnabled)); | 593 prefs::kSafeBrowsingReportingEnabled)); |
| 594 } | 594 } |
| OLD | NEW |