Page Info: Hide default permissions with a value of Ask if the default is Ask.

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 <string>
 #include <vector>
 
 #include "base/at_exit.h"
@@ skipping 55 matching lines @@
   // Clear cipher suite bits (the 16 lowest bits).
   connection_status &= ~net::SSL_CONNECTION_CIPHERSUITE_MASK;
   return cipher_suite | connection_status;
 }
 
 class MockWebsiteSettingsUI : public WebsiteSettingsUI {
  public:
   virtual ~MockWebsiteSettingsUI() {}
   MOCK_METHOD1(SetCookieInfo, void(const CookieInfoList& cookie_info_list));
   MOCK_METHOD0(SetPermissionInfoStub, void());
+  MOCK_METHOD1(UpdatePermissionButton,
+               void(VisiblePermissions visible_permissions));
   MOCK_METHOD1(SetIdentityInfo, void(const IdentityInfo& identity_info));
 
   void SetPermissionInfo(
       const PermissionInfoList& permission_info_list,
       ChosenObjectInfoList chosen_object_info_list) override {
     SetPermissionInfoStub();
     if (set_permission_info_callback_) {
       set_permission_info_callback_.Run(permission_info_list,
                                         std::move(chosen_object_info_list));
     }
@@ skipping 37 matching lines @@
         << "No WebsiteSettings instance created.";
     RenderViewHostTestHarness::TearDown();
     website_settings_.reset();
   }
 
   void SetDefaultUIExpectations(MockWebsiteSettingsUI* mock_ui) {
     // During creation |WebsiteSettings| makes the following calls to the ui.
     EXPECT_CALL(*mock_ui, SetPermissionInfoStub());
     EXPECT_CALL(*mock_ui, SetIdentityInfo(_));
     EXPECT_CALL(*mock_ui, SetCookieInfo(_));
+    EXPECT_CALL(*mock_ui, UpdatePermissionButton(_));
   }
 
   void SetURL(const std::string& url) { url_ = GURL(url); }
 
   void SetPermissionInfo(const PermissionInfoList& permission_info_list,
                          ChosenObjectInfoList chosen_object_info_list) {
     last_chosen_object_info_.clear();
     for (auto& chosen_object_info : chosen_object_info_list)
       last_chosen_object_info_.push_back(std::move(chosen_object_info));
   }
@@ skipping 481 matching lines @@
   EXPECT_EQ(0u, infobar_service()->infobar_count());
   website_settings()->OnUIClosing();
   EXPECT_EQ(0u, infobar_service()->infobar_count());
 }
 
 TEST_F(WebsiteSettingsTest, ShowInfoBar) {
   EXPECT_CALL(*mock_ui(), SetIdentityInfo(_));
   EXPECT_CALL(*mock_ui(), SetCookieInfo(_));
 
   EXPECT_CALL(*mock_ui(), SetPermissionInfoStub()).Times(2);
+  EXPECT_CALL(*mock_ui(), UpdatePermissionButton(_)).Times(2);
 
   EXPECT_EQ(0u, infobar_service()->infobar_count());
   website_settings()->OnSitePermissionChanged(
       CONTENT_SETTINGS_TYPE_GEOLOCATION, CONTENT_SETTING_ALLOW);
   website_settings()->OnUIClosing();
   ASSERT_EQ(1u, infobar_service()->infobar_count());
 
   infobar_service()->RemoveInfoBar(infobar_service()->infobar_at(0));
 }
 #endif
@@ skipping 71 matching lines @@
     histograms.ExpectBucketCount(
         kGenericHistogram,
         WebsiteSettings::WebsiteSettingsAction::WEBSITE_SETTINGS_OPENED, 2);
 
     histograms.ExpectTotalCount(test.histogram_name, 2);
     histograms.ExpectBucketCount(
         test.histogram_name,
         WebsiteSettings::WebsiteSettingsAction::WEBSITE_SETTINGS_OPENED, 2);
   }
 }