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