OLD | NEW |
1 // Copyright 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/message_loop.h" | 5 #include "base/message_loop.h" |
6 #include "base/string_number_conversions.h" | 6 #include "base/string_number_conversions.h" |
7 #include "base/values.h" | 7 #include "base/values.h" |
8 #include "chrome/browser/extensions/blacklist.h" | 8 #include "chrome/browser/extensions/blacklist.h" |
9 #include "chrome/browser/extensions/extension_prefs.h" | 9 #include "chrome/browser/extensions/extension_prefs.h" |
10 #include "chrome/browser/extensions/standard_management_policy_provider.h" | 10 #include "chrome/browser/extensions/standard_management_policy_provider.h" |
11 #include "chrome/browser/extensions/test_extension_prefs.h" | 11 #include "chrome/browser/extensions/test_extension_prefs.h" |
12 #include "chrome/common/extensions/extension_manifest_constants.h" | 12 #include "chrome/common/extensions/extension_manifest_constants.h" |
13 #include "content/public/test/test_browser_thread.h" | 13 #include "content/public/test/test_browser_thread.h" |
14 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
15 | 15 |
16 namespace extensions { | 16 namespace extensions { |
17 | 17 |
18 class StandardManagementPolicyProviderTest : public testing::Test { | 18 class StandardManagementPolicyProviderTest : public testing::Test { |
19 public: | 19 public: |
20 StandardManagementPolicyProviderTest() | 20 StandardManagementPolicyProviderTest() |
21 : ui_thread_(content::BrowserThread::UI, &message_loop_), | 21 : ui_thread_(content::BrowserThread::UI, &message_loop_), |
22 file_thread_(content::BrowserThread::FILE, &message_loop_), | 22 file_thread_(content::BrowserThread::FILE, &message_loop_), |
23 prefs_(message_loop_.message_loop_proxy()), | 23 prefs_(message_loop_.message_loop_proxy()), |
24 blacklist_(prefs()), | 24 provider_(prefs()) { |
25 provider_(prefs(), &blacklist_) { | |
26 } | 25 } |
27 | 26 |
28 | |
29 protected: | 27 protected: |
30 ExtensionPrefs* prefs() { | 28 ExtensionPrefs* prefs() { |
31 return prefs_.prefs(); | 29 return prefs_.prefs(); |
32 } | 30 } |
33 | 31 |
34 scoped_refptr<const Extension> CreateExtension(Extension::Location location, | 32 scoped_refptr<const Extension> CreateExtension(Extension::Location location, |
35 bool required) { | 33 bool required) { |
36 base::DictionaryValue values; | 34 base::DictionaryValue values; |
37 values.SetString(extension_manifest_keys::kName, "test"); | 35 values.SetString(extension_manifest_keys::kName, "test"); |
38 values.SetString(extension_manifest_keys::kVersion, "0.1"); | 36 values.SetString(extension_manifest_keys::kVersion, "0.1"); |
39 std::string error; | 37 std::string error; |
40 scoped_refptr<const Extension> extension = Extension::Create( | 38 scoped_refptr<const Extension> extension = Extension::Create( |
41 FilePath(), location, values, Extension::NO_FLAGS, &error); | 39 FilePath(), location, values, Extension::NO_FLAGS, &error); |
42 CHECK(extension.get()) << error; | 40 CHECK(extension.get()) << error; |
43 return extension; | 41 return extension; |
44 } | 42 } |
45 | 43 |
46 MessageLoop message_loop_; | 44 MessageLoop message_loop_; |
47 content::TestBrowserThread ui_thread_; | 45 content::TestBrowserThread ui_thread_; |
48 content::TestBrowserThread file_thread_; | 46 content::TestBrowserThread file_thread_; |
49 | 47 |
50 TestExtensionPrefs prefs_; | 48 TestExtensionPrefs prefs_; |
51 | 49 |
52 Blacklist blacklist_; | |
53 | |
54 StandardManagementPolicyProvider provider_; | 50 StandardManagementPolicyProvider provider_; |
55 }; | 51 }; |
56 | 52 |
57 // Tests various areas of blacklist functionality. | |
58 TEST_F(StandardManagementPolicyProviderTest, Blacklist) { | |
59 std::string not_installed_id = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"; | |
60 | |
61 // Install 5 extensions. | |
62 ExtensionList extensions; | |
63 for (int i = 0; i < 5; i++) { | |
64 std::string name = "test" + base::IntToString(i); | |
65 extensions.push_back(prefs_.AddExtension(name)); | |
66 } | |
67 EXPECT_EQ(NULL, | |
68 prefs()->GetInstalledExtensionInfo(not_installed_id).get()); | |
69 | |
70 ExtensionList::const_iterator iter; | |
71 for (iter = extensions.begin(); iter != extensions.end(); ++iter) { | |
72 EXPECT_TRUE(provider_.UserMayLoad(*iter, NULL)); | |
73 } | |
74 // Blacklist one installed and one not-installed extension id. | |
75 std::vector<std::string> blacklisted_ids; | |
76 blacklisted_ids.push_back(extensions[0]->id()); | |
77 blacklisted_ids.push_back(not_installed_id); | |
78 blacklist_.SetFromUpdater(blacklisted_ids, "version0"); | |
79 | |
80 // Make sure the id we expect to be blacklisted is. | |
81 EXPECT_FALSE(provider_.UserMayLoad(extensions[0], NULL)); | |
82 | |
83 // Make sure the other id's are not blacklisted. | |
84 for (iter = extensions.begin() + 1; iter != extensions.end(); ++iter) | |
85 EXPECT_TRUE(provider_.UserMayLoad(*iter, NULL)); | |
86 | |
87 // Make sure GetInstalledExtensionsInfo returns only the non-blacklisted | |
88 // extensions data. | |
89 scoped_ptr<ExtensionPrefs::ExtensionsInfo> info( | |
90 prefs()->GetInstalledExtensionsInfo()); | |
91 EXPECT_EQ(4u, info->size()); | |
92 ExtensionPrefs::ExtensionsInfo::iterator info_iter; | |
93 for (info_iter = info->begin(); info_iter != info->end(); ++info_iter) { | |
94 ExtensionInfo* extension_info = info_iter->get(); | |
95 EXPECT_NE(extensions[0]->id(), extension_info->extension_id); | |
96 } | |
97 } | |
98 | |
99 // Tests the behavior of the ManagementPolicy provider methods for an | 53 // Tests the behavior of the ManagementPolicy provider methods for an |
100 // extension required by policy. | 54 // extension required by policy. |
101 TEST_F(StandardManagementPolicyProviderTest, RequiredExtension) { | 55 TEST_F(StandardManagementPolicyProviderTest, RequiredExtension) { |
102 scoped_refptr<const Extension> extension = | 56 scoped_refptr<const Extension> extension = |
103 CreateExtension(Extension::EXTERNAL_POLICY_DOWNLOAD, true); | 57 CreateExtension(Extension::EXTERNAL_POLICY_DOWNLOAD, true); |
104 | 58 |
105 string16 error16; | 59 string16 error16; |
106 EXPECT_TRUE(provider_.UserMayLoad(extension.get(), &error16)); | 60 EXPECT_TRUE(provider_.UserMayLoad(extension.get(), &error16)); |
107 EXPECT_EQ(string16(), error16); | 61 EXPECT_EQ(string16(), error16); |
108 | 62 |
(...skipping 14 matching lines...) Expand all Loading... |
123 string16 error16; | 77 string16 error16; |
124 EXPECT_TRUE(provider_.UserMayLoad(extension.get(), &error16)); | 78 EXPECT_TRUE(provider_.UserMayLoad(extension.get(), &error16)); |
125 EXPECT_EQ(string16(), error16); | 79 EXPECT_EQ(string16(), error16); |
126 EXPECT_TRUE(provider_.UserMayModifySettings(extension.get(), &error16)); | 80 EXPECT_TRUE(provider_.UserMayModifySettings(extension.get(), &error16)); |
127 EXPECT_EQ(string16(), error16); | 81 EXPECT_EQ(string16(), error16); |
128 EXPECT_FALSE(provider_.MustRemainEnabled(extension.get(), &error16)); | 82 EXPECT_FALSE(provider_.MustRemainEnabled(extension.get(), &error16)); |
129 EXPECT_EQ(string16(), error16); | 83 EXPECT_EQ(string16(), error16); |
130 } | 84 } |
131 | 85 |
132 } // namespace extensions | 86 } // namespace extensions |
OLD | NEW |