(Views only) Prompt the user to reload the website after changing site permissions.

chrome/browser/ui/website_settings/website_settings_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/ui/website_settings/website_settings.h"
 
 #include "base/at_exit.h"
 #include "base/message_loop.h"
 #include "base/string16.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/content_settings/host_content_settings_map.h"
 #include "chrome/browser/content_settings/tab_specific_content_settings.h"
+#include "chrome/browser/infobars/infobar_tab_helper.h"
 #include "chrome/browser/ui/website_settings/website_settings_ui.h"
 #include "chrome/common/content_settings.h"
 #include "chrome/common/content_settings_types.h"
 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/public/browser/cert_store.h"
 #include "content/public/common/ssl_status.h"
 #include "content/public/test/test_browser_thread.h"
 #include "net/base/cert_status_flags.h"
 #include "net/base/ssl_connection_status_flags.h"
@@ skipping 42 matching lines @@
 };
 
 class WebsiteSettingsTest : public ChromeRenderViewHostTestHarness {
  public:
   WebsiteSettingsTest()
       : website_settings_(NULL),
         mock_ui_(NULL),
         cert_id_(0),
         browser_thread_(content::BrowserThread::UI, &message_loop_),
         tab_specific_content_settings_(NULL),
+        infobar_tab_helper_(NULL),
         url_("http://www.example.com") {
   }
 
   virtual ~WebsiteSettingsTest() {
   }
 
   virtual void SetUp() {
     ChromeRenderViewHostTestHarness::SetUp();
     // Setup stub SSLStatus.
     ssl_.security_style = content::SECURITY_STYLE_UNAUTHENTICATED;
 
     // Create the certificate.
     cert_id_ = 1;
     base::Time start_date = base::Time::Now();
     base::Time expiration_date = base::Time::FromInternalValue(
         start_date.ToInternalValue() + base::Time::kMicrosecondsPerWeek);
     cert_ = new net::X509Certificate("subject",
                                      "issuer",
                                      start_date,
                                      expiration_date);
 
     tab_specific_content_settings_.reset(
         new TabSpecificContentSettings(contents()));
+    infobar_tab_helper_.reset(new InfoBarTabHelper(contents()));
 
     // Setup the mock cert store.
     EXPECT_CALL(cert_store_, RetrieveCert(cert_id_, _) )
         .Times(AnyNumber())
         .WillRepeatedly(DoAll(SetArgPointee<1>(cert_), Return(true)));
 
     // Setup mock ui.
     mock_ui_.reset(new MockWebsiteSettingsUI());
   }
 
@@ skipping 13 matching lines @@
   }
 
   const GURL& url() const { return url_; }
   MockCertStore* cert_store() { return &cert_store_; }
   int cert_id() { return cert_id_; }
   MockWebsiteSettingsUI* mock_ui() { return mock_ui_.get(); }
   const SSLStatus& ssl() { return ssl_; }
   TabSpecificContentSettings* tab_specific_content_settings() {
     return tab_specific_content_settings_.get();
   }
+  InfoBarTabHelper* infobar_tab_helper() { return infobar_tab_helper_.get(); }
 
   WebsiteSettings* website_settings() {
     if (!website_settings_.get()) {
       website_settings_.reset(new WebsiteSettings(
-          mock_ui(), profile(), tab_specific_content_settings_.get(), url(),
-          ssl(), cert_store()));
+          mock_ui(), profile(), tab_specific_content_settings_.get(),
+          infobar_tab_helper_.get(), url(), ssl(), cert_store()));
     }
     return website_settings_.get();
   }
 
   SSLStatus ssl_;
 
  private:
   scoped_ptr<WebsiteSettings> website_settings_;
   scoped_ptr<MockWebsiteSettingsUI> mock_ui_;
   int cert_id_;
   scoped_refptr<net::X509Certificate> cert_;
   content::TestBrowserThread browser_thread_;
   scoped_ptr<TabSpecificContentSettings> tab_specific_content_settings_;
+  scoped_ptr<InfoBarTabHelper> infobar_tab_helper_;
   MockCertStore cert_store_;
   GURL url_;
 };
 
 }  // namespace
 
 TEST_F(WebsiteSettingsTest, OnPermissionsChanged) {
   // Setup site permissions.
   HostContentSettingsMap* content_settings =
       profile()->GetHostContentSettingsMap();
@@ skipping 152 matching lines @@
   ssl_.connection_status = status;
 
   SetDefaultUIExpectations(mock_ui());
 
   EXPECT_EQ(WebsiteSettings::SITE_CONNECTION_STATUS_ENCRYPTED_ERROR,
             website_settings()->site_connection_status());
   EXPECT_EQ(WebsiteSettings::SITE_IDENTITY_STATUS_CERT,
             website_settings()->site_identity_status());
   EXPECT_EQ(string16(), website_settings()->organization_name());
 }
+
+TEST_F(WebsiteSettingsTest, NoInfoBar) {
+  SetDefaultUIExpectations(mock_ui());
+  EXPECT_EQ(0u, infobar_tab_helper()->infobar_count());
+  website_settings()->OnUIClosing();
+  EXPECT_EQ(0u, infobar_tab_helper()->infobar_count());
+}
+
+TEST_F(WebsiteSettingsTest, ShowInfoBar) {
+  SetDefaultUIExpectations(mock_ui());
+  EXPECT_EQ(0u, infobar_tab_helper()->infobar_count());
+  website_settings()->OnSitePermissionChanged(
+      CONTENT_SETTINGS_TYPE_GEOLOCATION, CONTENT_SETTING_ALLOW);
+  website_settings()->OnUIClosing();
+  EXPECT_EQ(1u, infobar_tab_helper()->infobar_count());
+
+  infobar_tab_helper()->RemoveInfoBar(
+      infobar_tab_helper()->GetInfoBarDelegateAt(0));
+}