Fixes Permission Bubbles never responding to duplicate requests

chrome/browser/permissions/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/permissions/permission_context_base.h"
 
+#include <vector>
+
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/macros.h"
 #include "base/metrics/field_trial.h"
 #include "base/test/mock_entropy_provider.h"
 #include "build/build_config.h"
 #include "chrome/browser/content_settings/host_content_settings_map_factory.h"
 #include "chrome/browser/infobars/infobar_service.h"
 #include "chrome/browser/permissions/permission_queue_controller.h"
 #include "chrome/browser/permissions/permission_request_id.h"
 #include "chrome/browser/permissions/permission_util.h"
 #include "chrome/browser/ui/website_settings/permission_bubble_manager.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/content_settings/core/browser/host_content_settings_map.h"
 #include "components/content_settings/core/common/content_settings.h"
 #include "components/content_settings/core/common/content_settings_types.h"
 #include "components/variations/variations_associated_data.h"
 #include "content/public/browser/permission_type.h"
 #include "content/public/browser/render_frame_host.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"
 
 #if defined(OS_ANDROID)
 #include "chrome/browser/permissions/permission_queue_controller.h"
 #else
 #include "chrome/browser/ui/website_settings/permission_bubble_manager.h"
 #endif
 
 const char* kPermissionsKillSwitchFieldStudy =
     PermissionContextBase::kPermissionsKillSwitchFieldStudy;
 const char* kPermissionsKillSwitchBlockedValue =
     PermissionContextBase::kPermissionsKillSwitchBlockedValue;
 const char kPermissionsKillSwitchTestGroup[] = "TestGroup";
 
 class TestPermissionContext : public PermissionContextBase {
  public:
   TestPermissionContext(Profile* profile,
                         const content::PermissionType permission_type,
                         const ContentSettingsType content_settings_type)
       : PermissionContextBase(profile, permission_type, content_settings_type),
-        permission_set_(false),
-        permission_granted_(false),
         tab_context_updated_(false),
         field_trial_list_(
             new base::FieldTrialList(new base::MockEntropyProvider)) {}
 
   ~TestPermissionContext() override {}
 
 #if defined(OS_ANDROID)
   PermissionQueueController* GetInfoBarController() {
     return GetQueueController();
   }
 #endif
 
-  bool permission_granted() {
-    return permission_granted_;
-  }
-
-  bool permission_set() {
-    return permission_set_;
-  }
+  const std::vector<ContentSetting>& decisions() { return decisions_; }
 
   bool tab_context_updated() {
     return tab_context_updated_;
   }
 
   void TrackPermissionDecision(ContentSetting content_setting) {
-    permission_set_ = true;
-    permission_granted_ = content_setting == CONTENT_SETTING_ALLOW;
+    decisions_.push_back(content_setting);
   }
 
   void ResetFieldTrialList() {
     // Destroy the existing FieldTrialList before creating a new one to avoid
     // a DCHECK.
     field_trial_list_.reset();
     field_trial_list_.reset(new base::FieldTrialList(
         new base::MockEntropyProvider));
     variations::testing::ClearAllVariationParams();
   }
@@ skipping 10 matching lines @@
                         const GURL& requesting_origin,
                         bool allowed) override {
     tab_context_updated_ = true;
   }
 
   bool IsRestrictedToSecureOrigins() const override {
     return false;
   }
 
  private:
-   bool permission_set_;
-   bool permission_granted_;
-   bool tab_context_updated_;
-   scoped_ptr<base::FieldTrialList> field_trial_list_;
+  std::vector<ContentSetting> decisions_;
+  bool tab_context_updated_;
+  scoped_ptr<base::FieldTrialList> field_trial_list_;
 };
 
 class PermissionContextBaseTests : public ChromeRenderViewHostTestHarness {
  protected:
   PermissionContextBaseTests() {}
 
   // Accept or dismiss the permission bubble or infobar.
   void RespondToPermission(TestPermissionContext* context,
                            const PermissionRequestID& id,
                            const GURL& url,
-                           bool accept) {
+                           ContentSetting response) {
+    DCHECK(response == CONTENT_SETTING_ALLOW ||
+           response == CONTENT_SETTING_BLOCK ||
+           response == CONTENT_SETTING_ASK);
 #if defined(OS_ANDROID)
-    context->GetInfoBarController()->OnPermissionSet(id, url, url, accept,
-                                                     accept);
+    bool update_content_setting = response != CONTENT_SETTING_ASK;
+    bool allowed = response == CONTENT_SETTING_ALLOW;
+    context->GetInfoBarController()->OnPermissionSet(
+        id, url, url, update_content_setting, allowed);
 #else
     PermissionBubbleManager* manager =
         PermissionBubbleManager::FromWebContents(web_contents());
-    if (accept)
-      manager->Accept();
-    else
-      manager->Closing();
+    switch (response) {
+      case CONTENT_SETTING_ALLOW:
+        manager->Accept();
+        break;
+      case CONTENT_SETTING_BLOCK:
+        manager->Deny();
+        break;
+      case CONTENT_SETTING_ASK:
+        manager->Closing();
+        break;
+      default:
+        NOTREACHED();
+    }
 #endif
   }
 
   void TestAskAndGrant_TestContent() {
     TestPermissionContext permission_context(
         profile(), content::PermissionType::NOTIFICATIONS,
         CONTENT_SETTINGS_TYPE_NOTIFICATIONS);
     GURL url("http://www.google.com");
-    content::WebContentsTester::For(web_contents())->NavigateAndCommit(url);
+    NavigateAndCommit(url);
 
     const PermissionRequestID id(
         web_contents()->GetRenderProcessHost()->GetID(),
         web_contents()->GetMainFrame()->GetRoutingID(),
         -1);
     permission_context.RequestPermission(
         web_contents(),
         id, url, true,
         base::Bind(&TestPermissionContext::TrackPermissionDecision,
                    base::Unretained(&permission_context)));
 
-    RespondToPermission(&permission_context, id, url, true);
-    EXPECT_TRUE(permission_context.permission_set());
-    EXPECT_TRUE(permission_context.permission_granted());
+    RespondToPermission(&permission_context, id, url, CONTENT_SETTING_ALLOW);
+    EXPECT_EQ(1u, permission_context.decisions().size());
+    EXPECT_EQ(CONTENT_SETTING_ALLOW, permission_context.decisions()[0]);
     EXPECT_TRUE(permission_context.tab_context_updated());
     EXPECT_EQ(CONTENT_SETTING_ALLOW,
               permission_context.GetContentSettingFromMap(url, url));
   }
 
   void TestAskAndDismiss_TestContent() {
     TestPermissionContext permission_context(
         profile(), content::PermissionType::MIDI_SYSEX,
         CONTENT_SETTINGS_TYPE_MIDI_SYSEX);
     GURL url("http://www.google.es");
-    content::WebContentsTester::For(web_contents())->NavigateAndCommit(url);
+    NavigateAndCommit(url);
 
     const PermissionRequestID id(
         web_contents()->GetRenderProcessHost()->GetID(),
         web_contents()->GetMainFrame()->GetRoutingID(),
         -1);
     permission_context.RequestPermission(
         web_contents(),
         id, url, true,
         base::Bind(&TestPermissionContext::TrackPermissionDecision,
                    base::Unretained(&permission_context)));
 
-    RespondToPermission(&permission_context, id, url, false);
-    EXPECT_TRUE(permission_context.permission_set());
-    EXPECT_FALSE(permission_context.permission_granted());
+    RespondToPermission(&permission_context, id, url, CONTENT_SETTING_ASK);
+    EXPECT_EQ(1u, permission_context.decisions().size());
+    EXPECT_EQ(CONTENT_SETTING_ASK, permission_context.decisions()[0]);
     EXPECT_TRUE(permission_context.tab_context_updated());
     EXPECT_EQ(CONTENT_SETTING_ASK,
               permission_context.GetContentSettingFromMap(url, url));
   }
 
   void TestRequestPermissionInvalidUrl(
       content::PermissionType permission_type,
       ContentSettingsType content_settings_type) {
     TestPermissionContext permission_context(profile(), permission_type,
                                              content_settings_type);
     GURL url;
     ASSERT_FALSE(url.is_valid());
-    content::WebContentsTester::For(web_contents())->NavigateAndCommit(url);
+    NavigateAndCommit(url);
 
     const PermissionRequestID id(
         web_contents()->GetRenderProcessHost()->GetID(),
         web_contents()->GetMainFrame()->GetRoutingID(),
         -1);
     permission_context.RequestPermission(
         web_contents(),
         id, url, true,
         base::Bind(&TestPermissionContext::TrackPermissionDecision,
                    base::Unretained(&permission_context)));
 
-    EXPECT_TRUE(permission_context.permission_set());
-    EXPECT_FALSE(permission_context.permission_granted());
+    EXPECT_EQ(1u, permission_context.decisions().size());
+    EXPECT_EQ(CONTENT_SETTING_BLOCK, permission_context.decisions()[0]);
     EXPECT_TRUE(permission_context.tab_context_updated());
     EXPECT_EQ(CONTENT_SETTING_ASK,
               permission_context.GetContentSettingFromMap(url, url));
   }
 
   void TestGrantAndRevoke_TestContent(content::PermissionType permission_type,
                                       ContentSettingsType content_settings_type,
                                       ContentSetting expected_default) {
     TestPermissionContext permission_context(profile(), permission_type,
                                              content_settings_type);
     GURL url("https://www.google.com");
-    content::WebContentsTester::For(web_contents())->NavigateAndCommit(url);
+    NavigateAndCommit(url);
 
     const PermissionRequestID id(
         web_contents()->GetRenderProcessHost()->GetID(),
         web_contents()->GetMainFrame()->GetRoutingID(),
         -1);
     permission_context.RequestPermission(
         web_contents(),
         id, url, true,
         base::Bind(&TestPermissionContext::TrackPermissionDecision,
                    base::Unretained(&permission_context)));
 
-    RespondToPermission(&permission_context, id, url, true);
-    EXPECT_TRUE(permission_context.permission_set());
-    EXPECT_TRUE(permission_context.permission_granted());
+    RespondToPermission(&permission_context, id, url, CONTENT_SETTING_ALLOW);
+    EXPECT_EQ(1u, permission_context.decisions().size());
+    EXPECT_EQ(CONTENT_SETTING_ALLOW, permission_context.decisions()[0]);
     EXPECT_TRUE(permission_context.tab_context_updated());
     EXPECT_EQ(CONTENT_SETTING_ALLOW,
               permission_context.GetContentSettingFromMap(url, url));
 
     // Try to reset permission.
     permission_context.ResetPermission(url.GetOrigin(), url.GetOrigin());
     ContentSetting setting_after_reset =
         permission_context.GetContentSettingFromMap(url, url);
     ContentSetting default_setting =
         HostContentSettingsMapFactory::GetForProfile(profile())
@@ skipping 13 matching lines @@
     params[PermissionUtil::GetPermissionString(permission_type)] =
         kPermissionsKillSwitchBlockedValue;
     variations::AssociateVariationParams(
         kPermissionsKillSwitchFieldStudy, kPermissionsKillSwitchTestGroup,
         params);
     base::FieldTrialList::CreateFieldTrial(kPermissionsKillSwitchFieldStudy,
                                            kPermissionsKillSwitchTestGroup);
     EXPECT_TRUE(permission_context.IsPermissionKillSwitchOn());
   }
 
+  // Don't call this more than once in the same test, as it persists data to
+  // HostContentSettingsMap.
+  void TestParallelRequests(ContentSetting response) {
+    TestPermissionContext permission_context(
+        profile(), content::PermissionType::NOTIFICATIONS,
+        CONTENT_SETTINGS_TYPE_NOTIFICATIONS);
+    GURL url("http://www.google.com");
+    NavigateAndCommit(url);
+
+    const PermissionRequestID id0(
+        web_contents()->GetRenderProcessHost()->GetID(),
+        web_contents()->GetMainFrame()->GetRoutingID(), 0);
+    const PermissionRequestID id1(
+        web_contents()->GetRenderProcessHost()->GetID(),
+        web_contents()->GetMainFrame()->GetRoutingID(), 1);
+
+    permission_context.RequestPermission(
+        web_contents(), id0, url, true,
+        base::Bind(&TestPermissionContext::TrackPermissionDecision,
+                   base::Unretained(&permission_context)));
+    permission_context.RequestPermission(
+        web_contents(), id1, url, true,
+        base::Bind(&TestPermissionContext::TrackPermissionDecision,
+                   base::Unretained(&permission_context)));
+
+    EXPECT_EQ(0u, permission_context.decisions().size());
+
+    RespondToPermission(&permission_context, id0, url, response);
+
+    EXPECT_EQ(2u, permission_context.decisions().size());
+    EXPECT_EQ(response, permission_context.decisions()[0]);
+    EXPECT_EQ(response, permission_context.decisions()[1]);
+    EXPECT_TRUE(permission_context.tab_context_updated());
+
+    EXPECT_EQ(response, permission_context.GetContentSettingFromMap(url, url));
+  }
+
  private:
   // ChromeRenderViewHostTestHarness:
   void SetUp() override {
     ChromeRenderViewHostTestHarness::SetUp();
 #if defined(OS_ANDROID)
     InfoBarService::CreateForWebContents(web_contents());
 #else
     PermissionBubbleManager::CreateForWebContents(web_contents());
 #endif
   }
@@ skipping 88 matching lines @@
 #if defined(OS_ANDROID) || defined(OS_CHROMEOS)
   TestGlobalPermissionsKillSwitch(
       content::PermissionType::PROTECTED_MEDIA_IDENTIFIER,
       CONTENT_SETTINGS_TYPE_PROTECTED_MEDIA_IDENTIFIER);
 #endif
   TestGlobalPermissionsKillSwitch(content::PermissionType::AUDIO_CAPTURE,
                                   CONTENT_SETTINGS_TYPE_MEDIASTREAM_MIC);
   TestGlobalPermissionsKillSwitch(content::PermissionType::VIDEO_CAPTURE,
                                   CONTENT_SETTINGS_TYPE_MEDIASTREAM_CAMERA);
 }
+
+TEST_F(PermissionContextBaseTests, TestParallelRequestsAllowed) {
+  TestParallelRequests(CONTENT_SETTING_ALLOW);
+}
+
+TEST_F(PermissionContextBaseTests, TestParallelRequestsBlocked) {
+  TestParallelRequests(CONTENT_SETTING_BLOCK);
+}
+
+TEST_F(PermissionContextBaseTests, TestParallelRequestsDismissed) {
+  TestParallelRequests(CONTENT_SETTING_ASK);
+}