Change tests to prepare for turning on the permission bubble flag.

chrome/browser/content_settings/permission_context_base_unittest.cc

 // Copyright 2014 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/content_settings/permission_context_base.h"
 
 #include "base/bind.h"
 #include "chrome/browser/content_settings/host_content_settings_map.h"
 #include "chrome/browser/content_settings/permission_queue_controller.h"
 #include "chrome/browser/content_settings/permission_request_id.h"
 #include "chrome/browser/infobars/infobar_service.h"
+#include "chrome/browser/ui/website_settings/permission_bubble_manager.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/web_contents.h"
 #include "content/public/test/mock_render_process_host.h"
 #include "content/public/test/web_contents_tester.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 class PermissionContextBaseTests : public ChromeRenderViewHostTestHarness {
  protected:
   PermissionContextBaseTests() {}
   virtual ~PermissionContextBaseTests() {}
 
+  // Accept or dismiss the permission bubble.
+  void RespondPermissionBubble(bool accept) {
+    PermissionBubbleManager* manager =
+        PermissionBubbleManager::FromWebContents(web_contents());
+    if (accept)
+      manager->Accept();
+    else
+      manager->Closing();
+  }
+
  private:
   // ChromeRenderViewHostTestHarness:
   virtual void SetUp() OVERRIDE {
     ChromeRenderViewHostTestHarness::SetUp();
     InfoBarService::CreateForWebContents(web_contents());
+    PermissionBubbleManager::CreateForWebContents(web_contents());

[leng, 2014/07/23 16:29:10]

Is this a no-op for !PermissionBubbleManager::Enabled()?

[Greg Billock, 2014/07/29 16:44:13]

Not quite a no-op, but shouldn't interfere.

   }
 
   DISALLOW_COPY_AND_ASSIGN(PermissionContextBaseTests);
 };
 
 class TestPermissionContext : public PermissionContextBase {
  public:
   TestPermissionContext(Profile* profile,
                   const ContentSettingsType permission_type)
    : PermissionContextBase(profile, permission_type),
@@ skipping 48 matching lines @@
   const PermissionRequestID id(
       web_contents()->GetRenderProcessHost()->GetID(),
       web_contents()->GetRenderViewHost()->GetRoutingID(),
       -1, GURL());
   permission_context.RequestPermission(
       web_contents(),
       id, url, true,
       base::Bind(&TestPermissionContext::TrackPermissionDecision,
                  base::Unretained(&permission_context)));
 
-  permission_context.GetInfoBarController()->OnPermissionSet(
-      id, url, url, true, true);
+  if (!PermissionBubbleManager::Enabled()) {

[Miguel Garcia, 2014/07/25 12:17:35]

Can you create a Respond(bool value)  method that does infobars or bubbles and
then just use it in both tests in stead of having one if per test?

[Greg Billock, 2014/07/29 16:44:13]

Done.

+    permission_context.GetInfoBarController()->OnPermissionSet(
+        id, url, url, true, true);
+  } else {
+    RespondPermissionBubble(true);
+  }
   EXPECT_TRUE(permission_context.permission_set());
   EXPECT_TRUE(permission_context.permission_granted());
   EXPECT_TRUE(permission_context.tab_context_updated());
 
   ContentSetting setting =
       profile()->GetHostContentSettingsMap()->GetContentSetting(
           url.GetOrigin(), url.GetOrigin(),
           CONTENT_SETTINGS_TYPE_PUSH_MESSAGING, std::string());
   EXPECT_EQ(CONTENT_SETTING_ALLOW , setting);
 };
 
 // Simulates clicking Dismiss (X in the infobar.
 // The permission should be denied but not saved for future use.
 TEST_F(PermissionContextBaseTests, TestAskAndDismiss) {
   TestPermissionContext permission_context(profile(),
                       CONTENT_SETTINGS_TYPE_MIDI_SYSEX);
   GURL url("http://www.google.es");
   content::WebContentsTester::For(web_contents())->NavigateAndCommit(url);
 
   const PermissionRequestID id(
       web_contents()->GetRenderProcessHost()->GetID(),
       web_contents()->GetRenderViewHost()->GetRoutingID(),
       -1, GURL());
   permission_context.RequestPermission(
       web_contents(),
       id, url, true,
       base::Bind(&TestPermissionContext::TrackPermissionDecision,
                  base::Unretained(&permission_context)));
 
-  permission_context.GetInfoBarController()->OnPermissionSet(
-      id, url, url, false, false);
+  if (!PermissionBubbleManager::Enabled()) {
+    permission_context.GetInfoBarController()->OnPermissionSet(
+        id, url, url, false, false);
+  } else {
+    RespondPermissionBubble(false);
+  }
   EXPECT_TRUE(permission_context.permission_set());
   EXPECT_FALSE(permission_context.permission_granted());
   EXPECT_TRUE(permission_context.tab_context_updated());
 
   ContentSetting setting =
       profile()->GetHostContentSettingsMap()->GetContentSetting(
           url.GetOrigin(), url.GetOrigin(),
           CONTENT_SETTINGS_TYPE_MIDI_SYSEX, std::string());
   EXPECT_EQ(CONTENT_SETTING_ASK , setting);
 };