chrome/renderer/extensions/renderer_permissions_policy_delegate_unittest.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/command_line.h"
   6 #include "chrome/common/chrome_switches.h"
   7 #include "chrome/common/extensions/extension_constants.h"
   8 #include "chrome/renderer/extensions/dispatcher.h"
   9 #include "chrome/renderer/extensions/renderer_permissions_policy_delegate.h"
  10 #include "content/public/test/mock_render_process_host.h"
  11 #include "content/public/test/mock_render_thread.h"
  12 #include "extensions/common/extension.h"
  13 #include "extensions/common/extension_builder.h"
  14 #include "extensions/common/extension_messages.h"
  15 #include "extensions/common/permissions/permissions_data.h"
  16 #include "testing/gtest/include/gtest/gtest.h"
  17
  18 namespace extensions {
  19
  20 namespace {
  21
  22 class RendererPermissionsPolicyDelegateTest : public testing::Test {
  23  public:
  24   RendererPermissionsPolicyDelegateTest() {
  25   }
  26   virtual void SetUp() {
  27     testing::Test::SetUp();
  28     render_thread_.reset(new content::MockRenderThread());
  29     extension_dispatcher_.reset(new Dispatcher());
  30     policy_delegate_.reset(
  31         new RendererPermissionsPolicyDelegate(extension_dispatcher_.get()));
  32   }
  33  protected:
  34   scoped_ptr<Dispatcher> extension_dispatcher_;
  35   scoped_ptr<RendererPermissionsPolicyDelegate> policy_delegate_;
  36   scoped_ptr<content::MockRenderThread> render_thread_;
  37 };
  38
  39 scoped_refptr<const Extension> CreateTestExtension(const std::string& id) {
  40   return ExtensionBuilder()
  41       .SetManifest(DictionaryBuilder()
  42           .Set("name", "Extension with ID " + id)
  43           .Set("version", "1.0")
  44           .Set("manifest_version", 2)
  45           .Set("permissions", ListBuilder().Append("<all_urls>")))
  46       .SetID(id)
  47       .Build();
  48 }
  49
  50 }  // namespace
  51
  52 // Tests that CanExecuteScriptOnPage returns false for the signin process,
  53 // all else being equal.
  54 TEST_F(RendererPermissionsPolicyDelegateTest, CannotScriptSigninProcess) {
  55   GURL kSigninUrl(
  56       "https://accounts.google.com/ServiceLogin?service=chromiumsync");
  57   scoped_refptr<const Extension> extension(CreateTestExtension("a"));
  58   std::string error;
  59
  60   EXPECT_TRUE(PermissionsData::CanExecuteScriptOnPage(
  61       extension.get(), kSigninUrl, kSigninUrl, -1, NULL, -1, &error)) << error;
  62   // Pretend we are in the signin process. We should not be able to execute
  63   // script.
  64   CommandLine::ForCurrentProcess()->AppendSwitch(switches::kSigninProcess);
  65   EXPECT_FALSE(PermissionsData::CanExecuteScriptOnPage(
  66       extension.get(), kSigninUrl, kSigninUrl, -1, NULL, -1, &error)) << error;
  67 }
  68
  69 // Tests that CanExecuteScriptOnPage returns false for the any process
  70 // which hosts the webstore.
  71 TEST_F(RendererPermissionsPolicyDelegateTest, CannotScriptWebstore) {
  72   GURL kAnyUrl("http://example.com/");
  73   scoped_refptr<const Extension> extension(CreateTestExtension("a"));
  74   std::string error;
  75
  76   EXPECT_TRUE(PermissionsData::CanExecuteScriptOnPage(
  77       extension.get(), kAnyUrl, kAnyUrl, -1, NULL, -1, &error)) << error;
  78
  79   // Pretend we are in the webstore process. We should not be able to execute
  80   // script.
  81   scoped_refptr<const Extension> webstore_extension(
  82       CreateTestExtension(extension_misc::kWebStoreAppId));
  83   extension_dispatcher_->OnLoadedInternal(webstore_extension);
  84   extension_dispatcher_->OnActivateExtension(extension_misc::kWebStoreAppId);
  85   EXPECT_FALSE(PermissionsData::CanExecuteScriptOnPage(
  86       extension.get(), kAnyUrl, kAnyUrl, -1, NULL, -1, &error)) << error;
  87 }
  88
  89 }  // namespace extensions