Index: extensions/browser/policy_check_unittest.cc |
diff --git a/extensions/browser/policy_check_unittest.cc b/extensions/browser/policy_check_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..8b8af3057d2d42c91d3265f7c6d245d8c4781fa4 |
--- /dev/null |
+++ b/extensions/browser/policy_check_unittest.cc |
@@ -0,0 +1,122 @@ |
+// Copyright 2017 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 "extensions/browser/policy_check.h" |
+ |
+#include <vector> |
+ |
+#include "base/memory/ref_counted.h" |
+#include "base/strings/string16.h" |
+#include "base/strings/utf_string_conversions.h" |
+#include "content/public/test/test_browser_context.h" |
+#include "extensions/browser/extension_system.h" |
+#include "extensions/browser/extensions_test.h" |
+#include "extensions/browser/management_policy.h" |
+#include "extensions/browser/mock_extension_system.h" |
+#include "extensions/browser/preload_check.h" |
+#include "extensions/browser/preload_check_test_util.h" |
+#include "extensions/browser/test_extensions_browser_client.h" |
+#include "extensions/common/constants.h" |
+#include "extensions/common/extension.h" |
+#include "testing/gmock/include/gmock/gmock.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace extensions { |
+ |
+namespace { |
+ |
+const char kDummyPolicyError[] = "Cannot install extension"; |
+ |
+class ManagementPolicyMock : public ManagementPolicy::Provider { |
+ public: |
+ ManagementPolicyMock(const Extension* extension, bool may_load) |
+ : extension_(extension), may_load_(may_load) {} |
+ |
+ std::string GetDebugPolicyProviderName() const override { |
+ return "ManagementPolicyMock"; |
+ } |
+ |
+ bool UserMayLoad(const Extension* extension, |
+ base::string16* error) const override { |
+ EXPECT_EQ(extension_, extension); |
+ if (!may_load_) |
+ *error = base::ASCIIToUTF16(kDummyPolicyError); |
+ return may_load_; |
+ } |
+ |
+ private: |
+ const Extension* extension_; |
+ bool may_load_; |
+}; |
+ |
+class TestExtensionSystem : public MockExtensionSystem { |
+ public: |
+ explicit TestExtensionSystem(content::BrowserContext* context) |
+ : MockExtensionSystem(context) {} |
+ ~TestExtensionSystem() override {} |
+ |
+ ManagementPolicy* management_policy() override { return &management_policy_; } |
+ |
+ private: |
+ ManagementPolicy management_policy_; |
+}; |
+ |
+} // namespace |
+ |
+class PolicyCheckTest : public ExtensionsTest { |
+ public: |
+ PolicyCheckTest() { |
+ // Replace the MockExtensionSystemFactory set by ExtensionsTest. |
+ extensions_browser_client()->set_extension_system_factory(&factory_); |
+ } |
+ |
+ ~PolicyCheckTest() override {} |
+ |
+ void SetUp() override { |
+ ExtensionsTest::SetUp(); |
+ |
+ base::DictionaryValue manifest_dict; |
+ manifest_dict.SetString("name", "dummy name"); |
+ manifest_dict.SetString("version", "1"); |
+ std::string error; |
+ |
+ extension_ = Extension::Create(base::FilePath(), Manifest::UNPACKED, |
+ manifest_dict, Extension::NO_FLAGS, &error); |
+ EXPECT_TRUE(extension_.get()) << error; |
+ } |
+ |
+ protected: |
+ scoped_refptr<Extension> extension_; |
+ PreloadCheckRunner runner_; |
+ |
+ private: |
+ MockExtensionSystemFactory<TestExtensionSystem> factory_; |
+}; |
+ |
+// Tests an allowed extension. |
+TEST_F(PolicyCheckTest, PolicySuccess) { |
+ PolicyCheck policy_check(browser_context(), extension_); |
+ runner_.Run(&policy_check); |
+ EXPECT_TRUE(runner_.called()); |
+ EXPECT_EQ(0u, runner_.errors().size()); |
+ EXPECT_TRUE(policy_check.GetErrorMessage().empty()); |
+} |
+ |
+// Tests a disallowed extension. |
+TEST_F(PolicyCheckTest, PolicyFailure) { |
+ ManagementPolicyMock policy(extension_.get(), false); |
+ ExtensionSystem::Get(browser_context()) |
+ ->management_policy() |
+ ->RegisterProvider(&policy); |
+ |
+ PolicyCheck policy_check(browser_context(), extension_); |
+ runner_.Run(&policy_check); |
+ EXPECT_TRUE(runner_.called()); |
+ EXPECT_THAT(runner_.errors(), testing::UnorderedElementsAre( |
+ PreloadCheck::DISALLOWED_BY_POLICY)); |
+ EXPECT_EQ(base::ASCIIToUTF16(kDummyPolicyError), |
+ policy_check.GetErrorMessage()); |
+} |
+ |
+} // namespace extensions |