1// Copyright 2012 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/memory/scoped_ptr.h"
6#include "base/message_loop/message_loop.h"
7#include "base/strings/string_number_conversions.h"
8#include "base/values.h"
9#include "chrome/browser/extensions/blacklist.h"
10#include "chrome/browser/extensions/extension_management.h"
11#include "chrome/browser/extensions/standard_management_policy_provider.h"
12#include "chrome/browser/extensions/test_extension_prefs.h"
13#include "content/public/test/test_browser_thread.h"
14#include "extensions/browser/extension_prefs.h"
15#include "extensions/common/manifest_constants.h"
16#include "testing/gtest/include/gtest/gtest.h"
17
18namespace extensions {
19
20class StandardManagementPolicyProviderTest : public testing::Test {
21 public:
22  StandardManagementPolicyProviderTest()
23      : ui_thread_(content::BrowserThread::UI, &message_loop_),
24        file_thread_(content::BrowserThread::FILE, &message_loop_),
25        prefs_(message_loop_.message_loop_proxy().get()),
26        settings_(new ExtensionManagement(prefs()->pref_service())),
27        provider_(settings_.get()) {}
28
29 protected:
30  ExtensionPrefs* prefs() {
31    return prefs_.prefs();
32  }
33
34  scoped_refptr<const Extension> CreateExtension(Manifest::Location location,
35                                                 bool required) {
36    base::DictionaryValue values;
37    values.SetString(manifest_keys::kName, "test");
38    values.SetString(manifest_keys::kVersion, "0.1");
39    std::string error;
40    scoped_refptr<const Extension> extension = Extension::Create(
41        base::FilePath(), location, values, Extension::NO_FLAGS, &error);
42    CHECK(extension.get()) << error;
43    return extension;
44  }
45
46  base::MessageLoop message_loop_;
47  content::TestBrowserThread ui_thread_;
48  content::TestBrowserThread file_thread_;
49
50  TestExtensionPrefs prefs_;
51  scoped_ptr<ExtensionManagement> settings_;
52
53  StandardManagementPolicyProvider provider_;
54};
55
56// Tests the behavior of the ManagementPolicy provider methods for an
57// extension required by policy.
58TEST_F(StandardManagementPolicyProviderTest, RequiredExtension) {
59  scoped_refptr<const Extension> extension =
60      CreateExtension(Manifest::EXTERNAL_POLICY_DOWNLOAD, true);
61
62  base::string16 error16;
63  EXPECT_TRUE(provider_.UserMayLoad(extension.get(), &error16));
64  EXPECT_EQ(base::string16(), error16);
65
66  // We won't check the exact wording of the error, but it should say
67  // something.
68  EXPECT_FALSE(provider_.UserMayModifySettings(extension.get(), &error16));
69  EXPECT_NE(base::string16(), error16);
70  EXPECT_TRUE(provider_.MustRemainEnabled(extension.get(), &error16));
71  EXPECT_NE(base::string16(), error16);
72}
73
74// Tests the behavior of the ManagementPolicy provider methods for an
75// extension required by policy.
76TEST_F(StandardManagementPolicyProviderTest, NotRequiredExtension) {
77  scoped_refptr<const Extension> extension =
78      CreateExtension(Manifest::INTERNAL, false);
79
80  base::string16 error16;
81  EXPECT_TRUE(provider_.UserMayLoad(extension.get(), &error16));
82  EXPECT_EQ(base::string16(), error16);
83  EXPECT_TRUE(provider_.UserMayModifySettings(extension.get(), &error16));
84  EXPECT_EQ(base::string16(), error16);
85  EXPECT_FALSE(provider_.MustRemainEnabled(extension.get(), &error16));
86  EXPECT_EQ(base::string16(), error16);
87}
88
89}  // namespace extensions
90