1 // Copyright (c) 2010 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 "chrome/browser/policy/asynchronous_policy_loader.h" 6 #include "chrome/browser/policy/asynchronous_policy_provider.h" 7 #include "chrome/browser/policy/asynchronous_policy_test_base.h" 8 #include "chrome/browser/policy/configuration_policy_pref_store.h" 9 #include "chrome/browser/policy/mock_configuration_policy_store.h" 10 #include "policy/policy_constants.h" 11 #include "testing/gmock/include/gmock/gmock.h" 12 #include "testing/gtest/include/gtest/gtest.h" 13 14 using ::testing::_; 15 using ::testing::InSequence; 16 using ::testing::Return; 17 18 namespace policy { 19 20 // Creating the provider should provide initial policy. 21 TEST_F(AsynchronousPolicyTestBase, Provide) { 22 InSequence s; 23 DictionaryValue* policies = new DictionaryValue(); 24 policies->SetBoolean(policy::key::kSyncDisabled, true); 25 EXPECT_CALL(*delegate_, Load()).WillOnce(Return(policies)); 26 EXPECT_CALL(*store_, Apply(policy::kPolicySyncDisabled, _)).Times(1); 27 AsynchronousPolicyProvider provider( 28 ConfigurationPolicyPrefStore::GetChromePolicyDefinitionList(), 29 new AsynchronousPolicyLoader(delegate_.release(), 10)); 30 provider.Provide(store_.get()); 31 } 32 33 34 // Trigger a refresh manually and ensure that policy gets reloaded. 35 TEST_F(AsynchronousPolicyTestBase, ProvideAfterRefresh) { 36 InSequence s; 37 DictionaryValue* original_policies = new DictionaryValue(); 38 original_policies->SetBoolean(policy::key::kSyncDisabled, true); 39 EXPECT_CALL(*delegate_, Load()).WillOnce(Return(original_policies)); 40 DictionaryValue* refresh_policies = new DictionaryValue(); 41 refresh_policies->SetBoolean( 42 policy::key::kJavascriptEnabled, 43 true); 44 EXPECT_CALL(*delegate_, Load()).WillOnce(Return(refresh_policies)); 45 AsynchronousPolicyLoader* loader = 46 new AsynchronousPolicyLoader(delegate_.release(), 10); 47 AsynchronousPolicyProvider provider( 48 ConfigurationPolicyPrefStore::GetChromePolicyDefinitionList(), 49 loader); 50 loop_.RunAllPending(); 51 loader->Reload(); 52 loop_.RunAllPending(); 53 EXPECT_CALL(*store_, Apply(policy::kPolicyJavascriptEnabled, _)).Times(1); 54 provider.Provide(store_.get()); 55 } 56 57 } // namespace policy 58