Chromium Code Reviews| 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 "chrome/browser/website_settings.h" | 5 #include "chrome/browser/ui/website_settings/website_settings.h" |
| 6 | 6 |
| 7 #include "base/at_exit.h" | 7 #include "base/at_exit.h" |
| 8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" |
| 9 #include "base/string16.h" | 9 #include "base/string16.h" |
| 10 #include "base/utf_string_conversions.h" | 10 #include "base/utf_string_conversions.h" |
| 11 #include "chrome/browser/content_settings/host_content_settings_map.h" | 11 #include "chrome/browser/content_settings/host_content_settings_map.h" |
| 12 #include "chrome/browser/content_settings/tab_specific_content_settings.h" | 12 #include "chrome/browser/content_settings/tab_specific_content_settings.h" |
| 13 #include "chrome/browser/ui/website_settings_ui.h" | 13 #include "chrome/browser/ui/website_settings/website_settings_ui.h" |
| 14 #include "chrome/common/content_settings.h" | 14 #include "chrome/common/content_settings.h" |
| 15 #include "chrome/common/content_settings_types.h" | 15 #include "chrome/common/content_settings_types.h" |
| 16 #include "chrome/test/base/chrome_render_view_host_test_harness.h" | 16 #include "chrome/test/base/chrome_render_view_host_test_harness.h" |
| 17 #include "chrome/test/base/testing_profile.h" | 17 #include "chrome/test/base/testing_profile.h" |
| 18 #include "content/public/browser/cert_store.h" | 18 #include "content/public/browser/cert_store.h" |
| 19 #include "content/public/common/ssl_status.h" | 19 #include "content/public/common/ssl_status.h" |
| 20 #include "content/test/test_browser_thread.h" | 20 #include "content/test/test_browser_thread.h" |
| 21 #include "net/base/cert_status_flags.h" | 21 #include "net/base/cert_status_flags.h" |
| 22 #include "net/base/ssl_connection_status_flags.h" | 22 #include "net/base/ssl_connection_status_flags.h" |
| 23 #include "net/base/test_certificate_data.h" | 23 #include "net/base/test_certificate_data.h" |
| (...skipping 26 matching lines...) Expand all Loading... | |
| 50 class MockCertStore : public content::CertStore { | 50 class MockCertStore : public content::CertStore { |
| 51 public: | 51 public: |
| 52 virtual ~MockCertStore() {} | 52 virtual ~MockCertStore() {} |
| 53 MOCK_METHOD2(StoreCert, int(net::X509Certificate*, int)); | 53 MOCK_METHOD2(StoreCert, int(net::X509Certificate*, int)); |
| 54 MOCK_METHOD2(RetrieveCert, bool(int, scoped_refptr<net::X509Certificate>*)); | 54 MOCK_METHOD2(RetrieveCert, bool(int, scoped_refptr<net::X509Certificate>*)); |
| 55 }; | 55 }; |
| 56 | 56 |
| 57 class MockWebsiteSettingsUI : public WebsiteSettingsUI { | 57 class MockWebsiteSettingsUI : public WebsiteSettingsUI { |
| 58 public: | 58 public: |
| 59 virtual ~MockWebsiteSettingsUI() {} | 59 virtual ~MockWebsiteSettingsUI() {} |
| 60 MOCK_METHOD1(SetPresenter, void(WebsiteSettings* presenter)); | |
| 61 MOCK_METHOD1(SetCookieInfo, void(const CookieInfoList& cookie_info_list)); | 60 MOCK_METHOD1(SetCookieInfo, void(const CookieInfoList& cookie_info_list)); |
| 62 MOCK_METHOD1(SetPermissionInfo, | 61 MOCK_METHOD1(SetPermissionInfo, |
| 63 void(const PermissionInfoList& permission_info_list)); | 62 void(const PermissionInfoList& permission_info_list)); |
| 64 MOCK_METHOD1(SetIdentityInfo, void(const IdentityInfo& identity_info)); | 63 MOCK_METHOD1(SetIdentityInfo, void(const IdentityInfo& identity_info)); |
| 65 MOCK_METHOD1(SetFirstVisit, void(const string16& first_visit)); | 64 MOCK_METHOD1(SetFirstVisit, void(const string16& first_visit)); |
| 66 }; | 65 }; |
| 67 | 66 |
| 68 class WebsiteSettingsTest : public ChromeRenderViewHostTestHarness { | 67 class WebsiteSettingsTest : public ChromeRenderViewHostTestHarness { |
| 69 public: | 68 public: |
| 70 WebsiteSettingsTest() | 69 WebsiteSettingsTest() |
| (...skipping 25 matching lines...) Expand all Loading... | |
| 96 | 95 |
| 97 tab_specific_content_settings_.reset( | 96 tab_specific_content_settings_.reset( |
| 98 new TabSpecificContentSettings(contents())); | 97 new TabSpecificContentSettings(contents())); |
| 99 | 98 |
| 100 // Setup the mock cert store. | 99 // Setup the mock cert store. |
| 101 EXPECT_CALL(cert_store_, RetrieveCert(cert_id_, _) ) | 100 EXPECT_CALL(cert_store_, RetrieveCert(cert_id_, _) ) |
| 102 .Times(AnyNumber()) | 101 .Times(AnyNumber()) |
| 103 .WillRepeatedly(DoAll(SetArgPointee<1>(cert_), Return(true))); | 102 .WillRepeatedly(DoAll(SetArgPointee<1>(cert_), Return(true))); |
| 104 | 103 |
| 105 // Setup mock ui. | 104 // Setup mock ui. |
| 106 mock_ui_ = new MockWebsiteSettingsUI(); | 105 mock_ui_.reset(new MockWebsiteSettingsUI()); |
| 107 } | 106 } |
| 108 | 107 |
| 109 virtual void TearDown() { | 108 virtual void TearDown() { |
| 110 ASSERT_TRUE(website_settings_) << "No WebsiteSettings instance created."; | 109 ASSERT_TRUE(website_settings_.get()) |
| 111 // Call OnUIClosing to destroy the |website_settings| object. | 110 << "No WebsiteSettings instance created."; |
| 112 EXPECT_CALL(*mock_ui_, SetPresenter(NULL)); | |
| 113 website_settings_->OnUIClosing(); | |
| 114 RenderViewHostTestHarness::TearDown(); | 111 RenderViewHostTestHarness::TearDown(); |
| 112 website_settings_.reset(NULL); | |
|
tfarina
2012/06/03 16:22:24
nit: website_settings_.reset();
markusheintz_
2012/06/04 16:33:54
Done.
| |
| 115 } | 113 } |
| 116 | 114 |
| 117 void SetDefaultUIExpectations(MockWebsiteSettingsUI* mock_ui) { | 115 void SetDefaultUIExpectations(MockWebsiteSettingsUI* mock_ui) { |
| 118 // During creation |WebsiteSettings| makes the following calls to the ui. | 116 // During creation |WebsiteSettings| makes the following calls to the ui. |
| 119 EXPECT_CALL(*mock_ui, SetPermissionInfo(_)); | 117 EXPECT_CALL(*mock_ui, SetPermissionInfo(_)); |
| 120 EXPECT_CALL(*mock_ui, SetIdentityInfo(_)); | 118 EXPECT_CALL(*mock_ui, SetIdentityInfo(_)); |
| 121 EXPECT_CALL(*mock_ui, SetPresenter(_)); | |
| 122 EXPECT_CALL(*mock_ui, SetCookieInfo(_)); | 119 EXPECT_CALL(*mock_ui, SetCookieInfo(_)); |
| 123 EXPECT_CALL(*mock_ui, SetFirstVisit(string16())); | 120 EXPECT_CALL(*mock_ui, SetFirstVisit(string16())); |
| 124 } | 121 } |
| 125 | 122 |
| 126 const GURL& url() const { return url_; } | 123 const GURL& url() const { return url_; } |
| 127 MockCertStore* cert_store() { return &cert_store_; } | 124 MockCertStore* cert_store() { return &cert_store_; } |
| 128 int cert_id() { return cert_id_; } | 125 int cert_id() { return cert_id_; } |
| 129 MockWebsiteSettingsUI* mock_ui() { return mock_ui_; } | 126 MockWebsiteSettingsUI* mock_ui() { return mock_ui_.get(); } |
| 130 const SSLStatus& ssl() { return ssl_; } | 127 const SSLStatus& ssl() { return ssl_; } |
| 131 TabSpecificContentSettings* tab_specific_content_settings() { | 128 TabSpecificContentSettings* tab_specific_content_settings() { |
| 132 return tab_specific_content_settings_.get(); | 129 return tab_specific_content_settings_.get(); |
| 133 } | 130 } |
| 134 | 131 |
| 135 WebsiteSettings* website_settings() { | 132 WebsiteSettings* website_settings() { |
| 136 if (!website_settings_) { | 133 if (!website_settings_.get()) { |
| 137 website_settings_ = new WebsiteSettings( | 134 website_settings_.reset(new WebsiteSettings( |
| 138 mock_ui(), profile(), tab_specific_content_settings_.get(), url(), | 135 mock_ui(), profile(), tab_specific_content_settings_.get(), url(), |
| 139 ssl(), cert_store()); | 136 ssl(), cert_store())); |
| 140 } | 137 } |
| 141 return website_settings_; | 138 return website_settings_.get(); |
| 142 } | 139 } |
| 143 | 140 |
| 144 SSLStatus ssl_; | 141 SSLStatus ssl_; |
| 145 | 142 |
| 146 private: | 143 private: |
| 147 // The |Website_settings| object deletes itself after the OnUIClosing() | 144 scoped_ptr<WebsiteSettings> website_settings_; |
| 148 // method is called. | |
| 149 WebsiteSettings* website_settings_; // Weak pointer. | |
| 150 // The UI is owned by |web_site_settings_|. | 145 // The UI is owned by |web_site_settings_|. |
|
tfarina
2012/06/03 16:22:24
nit: Is this comment still valid?
markusheintz_
2012/06/04 16:33:54
Sorry of course not. Removed.
| |
| 151 MockWebsiteSettingsUI* mock_ui_; // Weak pointer. | 146 scoped_ptr<MockWebsiteSettingsUI> mock_ui_; |
| 152 int cert_id_; | 147 int cert_id_; |
| 153 scoped_refptr<net::X509Certificate> cert_; | 148 scoped_refptr<net::X509Certificate> cert_; |
| 154 content::TestBrowserThread browser_thread_; | 149 content::TestBrowserThread browser_thread_; |
| 155 scoped_ptr<TabSpecificContentSettings> tab_specific_content_settings_; | 150 scoped_ptr<TabSpecificContentSettings> tab_specific_content_settings_; |
| 156 MockCertStore cert_store_; | 151 MockCertStore cert_store_; |
| 157 GURL url_; | 152 GURL url_; |
| 158 }; | 153 }; |
| 159 | 154 |
| 160 } // namespace | 155 } // namespace |
| 161 | 156 |
| (...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 199 url(), url(), CONTENT_SETTINGS_TYPE_GEOLOCATION, ""); | 194 url(), url(), CONTENT_SETTINGS_TYPE_GEOLOCATION, ""); |
| 200 EXPECT_EQ(setting, CONTENT_SETTING_ALLOW); | 195 EXPECT_EQ(setting, CONTENT_SETTING_ALLOW); |
| 201 setting = content_settings->GetContentSetting( | 196 setting = content_settings->GetContentSetting( |
| 202 url(), url(), CONTENT_SETTINGS_TYPE_NOTIFICATIONS, ""); | 197 url(), url(), CONTENT_SETTINGS_TYPE_NOTIFICATIONS, ""); |
| 203 EXPECT_EQ(setting, CONTENT_SETTING_ALLOW); | 198 EXPECT_EQ(setting, CONTENT_SETTING_ALLOW); |
| 204 } | 199 } |
| 205 | 200 |
| 206 TEST_F(WebsiteSettingsTest, OnSiteDataAccessed) { | 201 TEST_F(WebsiteSettingsTest, OnSiteDataAccessed) { |
| 207 EXPECT_CALL(*mock_ui(), SetPermissionInfo(_)); | 202 EXPECT_CALL(*mock_ui(), SetPermissionInfo(_)); |
| 208 EXPECT_CALL(*mock_ui(), SetIdentityInfo(_)); | 203 EXPECT_CALL(*mock_ui(), SetIdentityInfo(_)); |
| 209 EXPECT_CALL(*mock_ui(), SetPresenter(_)); | |
| 210 EXPECT_CALL(*mock_ui(), SetFirstVisit(string16())); | 204 EXPECT_CALL(*mock_ui(), SetFirstVisit(string16())); |
| 211 EXPECT_CALL(*mock_ui(), SetCookieInfo(_)).Times(2); | 205 EXPECT_CALL(*mock_ui(), SetCookieInfo(_)).Times(2); |
| 212 | 206 |
| 213 website_settings()->OnSiteDataAccessed(); | 207 website_settings()->OnSiteDataAccessed(); |
| 214 } | 208 } |
| 215 | 209 |
| 216 TEST_F(WebsiteSettingsTest, HTTPConnection) { | 210 TEST_F(WebsiteSettingsTest, HTTPConnection) { |
| 217 SetDefaultUIExpectations(mock_ui()); | 211 SetDefaultUIExpectations(mock_ui()); |
| 218 EXPECT_EQ(WebsiteSettings::SITE_CONNECTION_STATUS_UNENCRYPTED, | 212 EXPECT_EQ(WebsiteSettings::SITE_CONNECTION_STATUS_UNENCRYPTED, |
| 219 website_settings()->site_connection_status()); | 213 website_settings()->site_connection_status()); |
| (...skipping 99 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 319 ssl_.connection_status = status; | 313 ssl_.connection_status = status; |
| 320 | 314 |
| 321 SetDefaultUIExpectations(mock_ui()); | 315 SetDefaultUIExpectations(mock_ui()); |
| 322 | 316 |
| 323 EXPECT_EQ(WebsiteSettings::SITE_CONNECTION_STATUS_ENCRYPTED_ERROR, | 317 EXPECT_EQ(WebsiteSettings::SITE_CONNECTION_STATUS_ENCRYPTED_ERROR, |
| 324 website_settings()->site_connection_status()); | 318 website_settings()->site_connection_status()); |
| 325 EXPECT_EQ(WebsiteSettings::SITE_IDENTITY_STATUS_CERT, | 319 EXPECT_EQ(WebsiteSettings::SITE_IDENTITY_STATUS_CERT, |
| 326 website_settings()->site_identity_status()); | 320 website_settings()->site_identity_status()); |
| 327 EXPECT_EQ(string16(), website_settings()->organization_name()); | 321 EXPECT_EQ(string16(), website_settings()->organization_name()); |
| 328 } | 322 } |
| OLD | NEW |