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