Index: chrome/browser/ui/website_settings/website_settings_unittest.cc |
diff --git a/chrome/browser/ui/website_settings/website_settings_unittest.cc b/chrome/browser/ui/website_settings/website_settings_unittest.cc |
index f979fcc66e78ff39a4ba678c6573bc97194fc6ff..7426a87f66b3d1acc3f01e5833f122631974a8e0 100644 |
--- a/chrome/browser/ui/website_settings/website_settings_unittest.cc |
+++ b/chrome/browser/ui/website_settings/website_settings_unittest.cc |
@@ -118,6 +118,8 @@ class WebsiteSettingsTest : public ChromeRenderViewHostTestHarness { |
EXPECT_CALL(*mock_ui, SetFirstVisit(base::string16())); |
} |
+ void SetURL(std::string url) { url_ = GURL(url); } |
+ |
const GURL& url() const { return url_; } |
MockCertStore* cert_store() { return &cert_store_; } |
int cert_id() { return cert_id_; } |
@@ -391,3 +393,23 @@ TEST_F(WebsiteSettingsTest, ShowInfoBar) { |
infobar_service()->RemoveInfoBar(infobar_service()->infobar_at(0)); |
} |
+ |
+TEST_F(WebsiteSettingsTest, AboutBlankPage) { |
+ SetURL("about:blank"); |
+ SetDefaultUIExpectations(mock_ui()); |
+ EXPECT_EQ(WebsiteSettings::SITE_CONNECTION_STATUS_INTERNAL_PAGE, |
+ website_settings()->site_connection_status()); |
+ EXPECT_EQ(WebsiteSettings::SITE_IDENTITY_STATUS_INTERNAL_PAGE, |
+ website_settings()->site_identity_status()); |
+ EXPECT_EQ(base::string16(), website_settings()->organization_name()); |
+} |
+ |
+TEST_F(WebsiteSettingsTest, InternalPage) { |
+ SetURL("chrome://bookmarks"); |
+ SetDefaultUIExpectations(mock_ui()); |
+ EXPECT_EQ(WebsiteSettings::SITE_CONNECTION_STATUS_INTERNAL_PAGE, |
+ website_settings()->site_connection_status()); |
+ EXPECT_EQ(WebsiteSettings::SITE_IDENTITY_STATUS_INTERNAL_PAGE, |
+ website_settings()->site_identity_status()); |
+ EXPECT_EQ(base::string16(), website_settings()->organization_name()); |
+} |