Split GetPermissionStatus into GetPermissionStatusForFrame/Worker

chrome/browser/permissions/permission_manager_unittest.cc

 // Copyright 2015 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_manager.h"
 
 #include "base/macros.h"
 #include "build/build_config.h"
 #include "chrome/browser/content_settings/host_content_settings_map_factory.h"
 #include "chrome/browser/permissions/permission_manager_factory.h"
 #include "chrome/browser/permissions/permission_result.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 "content/public/browser/permission_type.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using blink::mojom::PermissionStatus;
 using content::PermissionType;
 
 namespace {
 
 class PermissionManagerTestingProfile final : public TestingProfile {
  public:
   PermissionManagerTestingProfile() {}
   ~PermissionManagerTestingProfile() override {}
 
   PermissionManager* GetPermissionManager() override {
     return PermissionManagerFactory::GetForProfile(this);
   }
 
   DISALLOW_COPY_AND_ASSIGN(PermissionManagerTestingProfile);
 };
 
 }  // anonymous namespace
 
-class PermissionManagerTest : public testing::Test {
+class PermissionManagerTest : public ChromeRenderViewHostTestHarness {
  public:
   void OnPermissionChange(PermissionStatus permission) {
     callback_called_ = true;
     callback_result_ = permission;
   }
 
  protected:
   PermissionManagerTest()
       : url_("https://example.com"),
         other_url_("https://foo.com"),
         callback_called_(false),
         callback_result_(PermissionStatus::ASK) {}
 
+  void SetUp() override {
+    ChromeRenderViewHostTestHarness::SetUp();
+    profile_.reset(new PermissionManagerTestingProfile);
+    NavigateAndCommit(url_);
+  }
+
+  void TearDown() override {
+    profile_.reset();
+    ChromeRenderViewHostTestHarness::TearDown();
+  }
+
   PermissionManager* GetPermissionManager() {
-    return profile_.GetPermissionManager();
+    return profile_->GetPermissionManager();
   }
 
   HostContentSettingsMap* GetHostContentSettingsMap() {
-    return HostContentSettingsMapFactory::GetForProfile(&profile_);
+    return HostContentSettingsMapFactory::GetForProfile(profile_.get());
   }
 
   void CheckPermissionStatus(PermissionType type,
                              PermissionStatus expected) {
-    EXPECT_EQ(expected, GetPermissionManager()->GetPermissionStatus(
-                            type, url_.GetOrigin(), url_.GetOrigin()));
+    EXPECT_EQ(expected, GetPermissionManager()->GetPermissionStatusForFrame(
+                            type, main_rfh(), url_.GetOrigin()));
   }
 
   void CheckPermissionResult(ContentSettingsType type,
                              ContentSetting expected_status,
                              PermissionStatusSource expected_status_source) {
-    PermissionResult result = GetPermissionManager()->GetPermissionStatus(
-        type, url_.GetOrigin(), url_.GetOrigin());
+    PermissionResult result =
+        GetPermissionManager()->GetPermissionStatusForFrame(type, main_rfh(),
+                                                            url_.GetOrigin());
     EXPECT_EQ(expected_status, result.content_setting);
     EXPECT_EQ(expected_status_source, result.source);
   }
 
   void SetPermission(ContentSettingsType type, ContentSetting value) {
-    HostContentSettingsMapFactory::GetForProfile(&profile_)
+    HostContentSettingsMapFactory::GetForProfile(profile_.get())
         ->SetContentSettingDefaultScope(url_, url_, type, std::string(), value);
   }
 
   const GURL& url() const {
     return url_;
   }
 
   const GURL& other_url() const {
     return other_url_;
   }
 
   bool callback_called() const {
     return callback_called_;
   }
 
   PermissionStatus callback_result() const { return callback_result_; }
 
   void Reset() {
     callback_called_ = false;
     callback_result_ = PermissionStatus::ASK;
   }
 
  private:
   const GURL url_;
   const GURL other_url_;
   bool callback_called_;
   PermissionStatus callback_result_;
-  content::TestBrowserThreadBundle thread_bundle_;
-  PermissionManagerTestingProfile profile_;
+  std::unique_ptr<PermissionManagerTestingProfile> profile_;
 };
 
 TEST_F(PermissionManagerTest, GetPermissionStatusDefault) {
   CheckPermissionStatus(PermissionType::MIDI_SYSEX, PermissionStatus::ASK);
   CheckPermissionStatus(PermissionType::PUSH_MESSAGING, PermissionStatus::ASK);
   CheckPermissionStatus(PermissionType::NOTIFICATIONS, PermissionStatus::ASK);
   CheckPermissionStatus(PermissionType::GEOLOCATION, PermissionStatus::ASK);
 #if defined(OS_ANDROID)
   CheckPermissionStatus(PermissionType::PROTECTED_MEDIA_IDENTIFIER,
                         PermissionStatus::ASK);
@@ skipping 258 matching lines @@
   CheckPermissionStatus(PermissionType::GEOLOCATION, PermissionStatus::ASK);
   GetHostContentSettingsMap()->SetContentSettingDefaultScope(
       url(), url(), CONTENT_SETTINGS_TYPE_GEOLOCATION, std::string(),
       CONTENT_SETTING_ALLOW);
   CheckPermissionStatus(PermissionType::GEOLOCATION, PermissionStatus::GRANTED);
 
   EXPECT_FALSE(callback_called());
 
   GetPermissionManager()->UnsubscribePermissionStatusChange(subscription_id);
 }
+
+// Test that permissions that are allowed in workers are queried correctly.
+TEST_F(PermissionManagerTest, AllowedInWorker) {
+  // Notifications is allowed from workers but should ASK by default. This
+  // should return DENIED in a worker because it can't prompt.
+  EXPECT_EQ(PermissionStatus::DENIED,
+            GetPermissionManager()->GetPermissionStatusForWorker(
+                PermissionType::NOTIFICATIONS, url()));
+
+  // If it is allowed, it should return GRANTED.
+  GetHostContentSettingsMap()->SetContentSettingDefaultScope(
+      url(), url(), CONTENT_SETTINGS_TYPE_NOTIFICATIONS, std::string(),
+      CONTENT_SETTING_ALLOW);
+  EXPECT_EQ(PermissionStatus::GRANTED,
+            GetPermissionManager()->GetPermissionStatusForWorker(
+                PermissionType::NOTIFICATIONS, url()));
+
+  // Geolocation isn't allowed from workers so should return DENIED even if the
+  // user has allowed it.
+  GetHostContentSettingsMap()->SetContentSettingDefaultScope(
+      url(), url(), CONTENT_SETTINGS_TYPE_GEOLOCATION, std::string(),
+      CONTENT_SETTING_ALLOW);
+  EXPECT_EQ(PermissionStatus::DENIED,
+            GetPermissionManager()->GetPermissionStatusForWorker(
+                PermissionType::GEOLOCATION, url()));
+}