Index: chrome/browser/policy/policy_service_impl_unittest.cc |
diff --git a/chrome/browser/policy/policy_service_impl_unittest.cc b/chrome/browser/policy/policy_service_impl_unittest.cc |
deleted file mode 100644 |
index 8bf2b8b71b6a6718530648c1ab6e2a3ba553be8a..0000000000000000000000000000000000000000 |
--- a/chrome/browser/policy/policy_service_impl_unittest.cc |
+++ /dev/null |
@@ -1,651 +0,0 @@ |
-// Copyright (c) 2012 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include "chrome/browser/policy/policy_service_impl.h" |
- |
-#include "base/bind.h" |
-#include "base/bind_helpers.h" |
-#include "base/callback.h" |
-#include "base/memory/scoped_ptr.h" |
-#include "base/run_loop.h" |
-#include "base/values.h" |
-#include "chrome/browser/policy/mock_policy_service.h" |
-#include "components/policy/core/common/external_data_fetcher.h" |
-#include "components/policy/core/common/mock_configuration_policy_provider.h" |
-#include "testing/gmock/include/gmock/gmock.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-using ::testing::AnyNumber; |
-using ::testing::Mock; |
-using ::testing::Return; |
-using ::testing::_; |
- |
-namespace policy { |
- |
-namespace { |
- |
-const char kExtension[] = "extension-id"; |
-const char kSameLevelPolicy[] = "policy-same-level-and-scope"; |
-const char kDiffLevelPolicy[] = "chrome-diff-level-and-scope"; |
- |
-void SetPolicyMapValue(const std::string& key, |
- const std::string& value, |
- PolicyBundle* bundle) { |
- bundle->Get(PolicyNamespace(POLICY_DOMAIN_CHROME, std::string())) |
- .Set(key, |
- POLICY_LEVEL_MANDATORY, |
- POLICY_SCOPE_USER, |
- new base::StringValue(value), |
- NULL); |
-} |
- |
-// Helper to compare the arguments to an EXPECT_CALL of OnPolicyUpdated() with |
-// their expected values. |
-MATCHER_P(PolicyEquals, expected, "") { |
- return arg.Equals(*expected); |
-} |
- |
-// Helper to compare the arguments to an EXPECT_CALL of OnPolicyValueUpdated() |
-// with their expected values. |
-MATCHER_P(ValueEquals, expected, "") { |
- return base::Value::Equals(arg, expected); |
-} |
- |
-// Helper that fills |bundle| with test policies. |
-void AddTestPolicies(PolicyBundle* bundle, |
- const char* value, |
- PolicyLevel level, |
- PolicyScope scope) { |
- PolicyMap* policy_map = |
- &bundle->Get(PolicyNamespace(POLICY_DOMAIN_CHROME, std::string())); |
- policy_map->Set(kSameLevelPolicy, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateStringValue(value), NULL); |
- policy_map->Set(kDiffLevelPolicy, level, scope, |
- base::Value::CreateStringValue(value), NULL); |
- policy_map = |
- &bundle->Get(PolicyNamespace(POLICY_DOMAIN_EXTENSIONS, kExtension)); |
- policy_map->Set(kSameLevelPolicy, POLICY_LEVEL_MANDATORY, |
- POLICY_SCOPE_USER, base::Value::CreateStringValue(value), |
- NULL); |
- policy_map->Set(kDiffLevelPolicy, level, scope, |
- base::Value::CreateStringValue(value), NULL); |
-} |
- |
-// Observer class that changes the policy in the passed provider when the |
-// callback is invoked. |
-class ChangePolicyObserver : public PolicyService::Observer { |
- public: |
- explicit ChangePolicyObserver(MockConfigurationPolicyProvider* provider) |
- : provider_(provider), |
- observer_invoked_(false) {} |
- |
- virtual void OnPolicyUpdated(const PolicyNamespace&, |
- const PolicyMap& previous, |
- const PolicyMap& current) OVERRIDE { |
- PolicyMap new_policy; |
- new_policy.Set("foo", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(14), NULL); |
- provider_->UpdateChromePolicy(new_policy); |
- observer_invoked_ = true; |
- } |
- |
- bool observer_invoked() const { return observer_invoked_; } |
- |
- private: |
- MockConfigurationPolicyProvider* provider_; |
- bool observer_invoked_; |
-}; |
- |
-} // namespace |
- |
-class PolicyServiceTest : public testing::Test { |
- public: |
- PolicyServiceTest() {} |
- virtual void SetUp() OVERRIDE { |
- EXPECT_CALL(provider0_, IsInitializationComplete(_)) |
- .WillRepeatedly(Return(true)); |
- EXPECT_CALL(provider1_, IsInitializationComplete(_)) |
- .WillRepeatedly(Return(true)); |
- EXPECT_CALL(provider2_, IsInitializationComplete(_)) |
- .WillRepeatedly(Return(true)); |
- |
- provider0_.Init(); |
- provider1_.Init(); |
- provider2_.Init(); |
- |
- policy0_.Set("pre", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(13), NULL); |
- provider0_.UpdateChromePolicy(policy0_); |
- |
- PolicyServiceImpl::Providers providers; |
- providers.push_back(&provider0_); |
- providers.push_back(&provider1_); |
- providers.push_back(&provider2_); |
- policy_service_.reset(new PolicyServiceImpl( |
- providers, PolicyServiceImpl::PreprocessCallback())); |
- } |
- |
- virtual void TearDown() OVERRIDE { |
- provider0_.Shutdown(); |
- provider1_.Shutdown(); |
- provider2_.Shutdown(); |
- } |
- |
- MOCK_METHOD2(OnPolicyValueUpdated, void(const base::Value*, |
- const base::Value*)); |
- |
- MOCK_METHOD0(OnPolicyRefresh, void()); |
- |
- // Returns true if the policies for namespace |ns| match |expected|. |
- bool VerifyPolicies(const PolicyNamespace& ns, |
- const PolicyMap& expected) { |
- return policy_service_->GetPolicies(ns).Equals(expected); |
- } |
- |
- void RunUntilIdle() { |
- base::RunLoop loop; |
- loop.RunUntilIdle(); |
- } |
- |
- protected: |
- base::MessageLoop loop_; |
- MockConfigurationPolicyProvider provider0_; |
- MockConfigurationPolicyProvider provider1_; |
- MockConfigurationPolicyProvider provider2_; |
- PolicyMap policy0_; |
- PolicyMap policy1_; |
- PolicyMap policy2_; |
- scoped_ptr<PolicyServiceImpl> policy_service_; |
- |
- private: |
- DISALLOW_COPY_AND_ASSIGN(PolicyServiceTest); |
-}; |
- |
-TEST_F(PolicyServiceTest, LoadsPoliciesBeforeProvidersRefresh) { |
- PolicyMap expected; |
- expected.Set("pre", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(13), NULL); |
- EXPECT_TRUE(VerifyPolicies( |
- PolicyNamespace(POLICY_DOMAIN_CHROME, std::string()), expected)); |
-} |
- |
-TEST_F(PolicyServiceTest, NotifyObservers) { |
- MockPolicyServiceObserver observer; |
- policy_service_->AddObserver(POLICY_DOMAIN_CHROME, &observer); |
- |
- PolicyMap expectedPrevious; |
- expectedPrevious.Set("pre", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(13), NULL); |
- |
- PolicyMap expectedCurrent; |
- expectedCurrent.CopyFrom(expectedPrevious); |
- expectedCurrent.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(123), NULL); |
- policy0_.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(123), NULL); |
- EXPECT_CALL(observer, OnPolicyUpdated(PolicyNamespace(POLICY_DOMAIN_CHROME, |
- std::string()), |
- PolicyEquals(&expectedPrevious), |
- PolicyEquals(&expectedCurrent))); |
- provider0_.UpdateChromePolicy(policy0_); |
- Mock::VerifyAndClearExpectations(&observer); |
- |
- // No changes. |
- EXPECT_CALL(observer, OnPolicyUpdated(_, _, _)).Times(0); |
- provider0_.UpdateChromePolicy(policy0_); |
- Mock::VerifyAndClearExpectations(&observer); |
- EXPECT_TRUE(VerifyPolicies( |
- PolicyNamespace(POLICY_DOMAIN_CHROME, std::string()), expectedCurrent)); |
- |
- // New policy. |
- expectedPrevious.CopyFrom(expectedCurrent); |
- expectedCurrent.Set("bbb", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(456), NULL); |
- policy0_.Set("bbb", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(456), NULL); |
- EXPECT_CALL(observer, OnPolicyUpdated(PolicyNamespace(POLICY_DOMAIN_CHROME, |
- std::string()), |
- PolicyEquals(&expectedPrevious), |
- PolicyEquals(&expectedCurrent))); |
- provider0_.UpdateChromePolicy(policy0_); |
- Mock::VerifyAndClearExpectations(&observer); |
- |
- // Removed policy. |
- expectedPrevious.CopyFrom(expectedCurrent); |
- expectedCurrent.Erase("bbb"); |
- policy0_.Erase("bbb"); |
- EXPECT_CALL(observer, OnPolicyUpdated(PolicyNamespace(POLICY_DOMAIN_CHROME, |
- std::string()), |
- PolicyEquals(&expectedPrevious), |
- PolicyEquals(&expectedCurrent))); |
- provider0_.UpdateChromePolicy(policy0_); |
- Mock::VerifyAndClearExpectations(&observer); |
- |
- // Changed policy. |
- expectedPrevious.CopyFrom(expectedCurrent); |
- expectedCurrent.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(789), NULL); |
- policy0_.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(789), NULL); |
- |
- EXPECT_CALL(observer, OnPolicyUpdated(PolicyNamespace(POLICY_DOMAIN_CHROME, |
- std::string()), |
- PolicyEquals(&expectedPrevious), |
- PolicyEquals(&expectedCurrent))); |
- provider0_.UpdateChromePolicy(policy0_); |
- Mock::VerifyAndClearExpectations(&observer); |
- |
- // No changes again. |
- EXPECT_CALL(observer, OnPolicyUpdated(_, _, _)).Times(0); |
- provider0_.UpdateChromePolicy(policy0_); |
- Mock::VerifyAndClearExpectations(&observer); |
- EXPECT_TRUE(VerifyPolicies( |
- PolicyNamespace(POLICY_DOMAIN_CHROME, std::string()), expectedCurrent)); |
- |
- policy_service_->RemoveObserver(POLICY_DOMAIN_CHROME, &observer); |
-} |
- |
-TEST_F(PolicyServiceTest, NotifyObserversInMultipleNamespaces) { |
- const std::string kExtension0("extension-0"); |
- const std::string kExtension1("extension-1"); |
- const std::string kExtension2("extension-2"); |
- MockPolicyServiceObserver chrome_observer; |
- MockPolicyServiceObserver extension_observer; |
- policy_service_->AddObserver(POLICY_DOMAIN_CHROME, &chrome_observer); |
- policy_service_->AddObserver(POLICY_DOMAIN_EXTENSIONS, &extension_observer); |
- |
- PolicyMap previous_policy_map; |
- previous_policy_map.Set("pre", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(13), NULL); |
- PolicyMap policy_map; |
- policy_map.CopyFrom(previous_policy_map); |
- policy_map.Set("policy", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateStringValue("value"), NULL); |
- |
- scoped_ptr<PolicyBundle> bundle(new PolicyBundle()); |
- // The initial setup includes a policy for chrome that is now changing. |
- bundle->Get(PolicyNamespace(POLICY_DOMAIN_CHROME, std::string())) |
- .CopyFrom(policy_map); |
- bundle->Get(PolicyNamespace(POLICY_DOMAIN_EXTENSIONS, kExtension0)) |
- .CopyFrom(policy_map); |
- bundle->Get(PolicyNamespace(POLICY_DOMAIN_EXTENSIONS, kExtension1)) |
- .CopyFrom(policy_map); |
- |
- const PolicyMap kEmptyPolicyMap; |
- EXPECT_CALL( |
- chrome_observer, |
- OnPolicyUpdated(PolicyNamespace(POLICY_DOMAIN_CHROME, std::string()), |
- PolicyEquals(&previous_policy_map), |
- PolicyEquals(&policy_map))); |
- EXPECT_CALL( |
- extension_observer, |
- OnPolicyUpdated(PolicyNamespace(POLICY_DOMAIN_EXTENSIONS, kExtension0), |
- PolicyEquals(&kEmptyPolicyMap), |
- PolicyEquals(&policy_map))); |
- EXPECT_CALL( |
- extension_observer, |
- OnPolicyUpdated(PolicyNamespace(POLICY_DOMAIN_EXTENSIONS, kExtension1), |
- PolicyEquals(&kEmptyPolicyMap), |
- PolicyEquals(&policy_map))); |
- provider0_.UpdatePolicy(bundle.Pass()); |
- RunUntilIdle(); |
- Mock::VerifyAndClearExpectations(&chrome_observer); |
- Mock::VerifyAndClearExpectations(&extension_observer); |
- |
- // Chrome policy stays the same, kExtension0 is gone, kExtension1 changes, |
- // and kExtension2 is new. |
- previous_policy_map.CopyFrom(policy_map); |
- bundle.reset(new PolicyBundle()); |
- bundle->Get(PolicyNamespace(POLICY_DOMAIN_CHROME, std::string())) |
- .CopyFrom(policy_map); |
- policy_map.Set("policy", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateStringValue("another value"), NULL); |
- bundle->Get(PolicyNamespace(POLICY_DOMAIN_EXTENSIONS, kExtension1)) |
- .CopyFrom(policy_map); |
- bundle->Get(PolicyNamespace(POLICY_DOMAIN_EXTENSIONS, kExtension2)) |
- .CopyFrom(policy_map); |
- |
- EXPECT_CALL(chrome_observer, OnPolicyUpdated(_, _, _)).Times(0); |
- EXPECT_CALL( |
- extension_observer, |
- OnPolicyUpdated(PolicyNamespace(POLICY_DOMAIN_EXTENSIONS, kExtension0), |
- PolicyEquals(&previous_policy_map), |
- PolicyEquals(&kEmptyPolicyMap))); |
- EXPECT_CALL( |
- extension_observer, |
- OnPolicyUpdated(PolicyNamespace(POLICY_DOMAIN_EXTENSIONS, kExtension1), |
- PolicyEquals(&previous_policy_map), |
- PolicyEquals(&policy_map))); |
- EXPECT_CALL( |
- extension_observer, |
- OnPolicyUpdated(PolicyNamespace(POLICY_DOMAIN_EXTENSIONS, kExtension2), |
- PolicyEquals(&kEmptyPolicyMap), |
- PolicyEquals(&policy_map))); |
- provider0_.UpdatePolicy(bundle.Pass()); |
- RunUntilIdle(); |
- Mock::VerifyAndClearExpectations(&chrome_observer); |
- Mock::VerifyAndClearExpectations(&extension_observer); |
- |
- policy_service_->RemoveObserver(POLICY_DOMAIN_CHROME, &chrome_observer); |
- policy_service_->RemoveObserver(POLICY_DOMAIN_EXTENSIONS, |
- &extension_observer); |
-} |
- |
-TEST_F(PolicyServiceTest, ObserverChangesPolicy) { |
- ChangePolicyObserver observer(&provider0_); |
- policy_service_->AddObserver(POLICY_DOMAIN_CHROME, &observer); |
- policy0_.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(123), NULL); |
- policy0_.Set("bbb", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(1234), NULL); |
- // Should not crash. |
- provider0_.UpdateChromePolicy(policy0_); |
- policy_service_->RemoveObserver(POLICY_DOMAIN_CHROME, &observer); |
- EXPECT_TRUE(observer.observer_invoked()); |
-} |
- |
-TEST_F(PolicyServiceTest, Priorities) { |
- PolicyMap expected; |
- expected.Set("pre", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(13), NULL); |
- expected.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(0), NULL); |
- policy0_.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(0), NULL); |
- policy1_.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(1), NULL); |
- policy2_.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(2), NULL); |
- provider0_.UpdateChromePolicy(policy0_); |
- provider1_.UpdateChromePolicy(policy1_); |
- provider2_.UpdateChromePolicy(policy2_); |
- EXPECT_TRUE(VerifyPolicies( |
- PolicyNamespace(POLICY_DOMAIN_CHROME, std::string()), expected)); |
- |
- expected.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(1), NULL); |
- policy0_.Erase("aaa"); |
- provider0_.UpdateChromePolicy(policy0_); |
- EXPECT_TRUE(VerifyPolicies( |
- PolicyNamespace(POLICY_DOMAIN_CHROME, std::string()), expected)); |
- |
- expected.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(2), NULL); |
- policy1_.Set("aaa", POLICY_LEVEL_RECOMMENDED, POLICY_SCOPE_USER, |
- base::Value::CreateIntegerValue(1), NULL); |
- provider1_.UpdateChromePolicy(policy1_); |
- EXPECT_TRUE(VerifyPolicies( |
- PolicyNamespace(POLICY_DOMAIN_CHROME, std::string()), expected)); |
-} |
- |
-TEST_F(PolicyServiceTest, PolicyChangeRegistrar) { |
- scoped_ptr<PolicyChangeRegistrar> registrar(new PolicyChangeRegistrar( |
- policy_service_.get(), |
- PolicyNamespace(POLICY_DOMAIN_CHROME, std::string()))); |
- |
- // Starting to observe existing policies doesn't trigger a notification. |
- EXPECT_CALL(*this, OnPolicyValueUpdated(_, _)).Times(0); |
- registrar->Observe("pre", base::Bind( |
- &PolicyServiceTest::OnPolicyValueUpdated, |
- base::Unretained(this))); |
- registrar->Observe("aaa", base::Bind( |
- &PolicyServiceTest::OnPolicyValueUpdated, |
- base::Unretained(this))); |
- RunUntilIdle(); |
- Mock::VerifyAndClearExpectations(this); |
- |
- // Changing it now triggers a notification. |
- base::FundamentalValue kValue0(0); |
- EXPECT_CALL(*this, OnPolicyValueUpdated(NULL, ValueEquals(&kValue0))); |
- policy0_.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- kValue0.DeepCopy(), NULL); |
- provider0_.UpdateChromePolicy(policy0_); |
- Mock::VerifyAndClearExpectations(this); |
- |
- // Changing other values doesn't trigger a notification. |
- EXPECT_CALL(*this, OnPolicyValueUpdated(_, _)).Times(0); |
- policy0_.Set("bbb", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- kValue0.DeepCopy(), NULL); |
- provider0_.UpdateChromePolicy(policy0_); |
- Mock::VerifyAndClearExpectations(this); |
- |
- // Modifying the value triggers a notification. |
- base::FundamentalValue kValue1(1); |
- EXPECT_CALL(*this, OnPolicyValueUpdated(ValueEquals(&kValue0), |
- ValueEquals(&kValue1))); |
- policy0_.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- kValue1.DeepCopy(), NULL); |
- provider0_.UpdateChromePolicy(policy0_); |
- Mock::VerifyAndClearExpectations(this); |
- |
- // Removing the value triggers a notification. |
- EXPECT_CALL(*this, OnPolicyValueUpdated(ValueEquals(&kValue1), NULL)); |
- policy0_.Erase("aaa"); |
- provider0_.UpdateChromePolicy(policy0_); |
- Mock::VerifyAndClearExpectations(this); |
- |
- // No more notifications after destroying the registrar. |
- EXPECT_CALL(*this, OnPolicyValueUpdated(_, _)).Times(0); |
- registrar.reset(); |
- policy0_.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- kValue1.DeepCopy(), NULL); |
- policy0_.Set("pre", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- kValue1.DeepCopy(), NULL); |
- provider0_.UpdateChromePolicy(policy0_); |
- Mock::VerifyAndClearExpectations(this); |
-} |
- |
-TEST_F(PolicyServiceTest, RefreshPolicies) { |
- EXPECT_CALL(provider0_, RefreshPolicies()).Times(AnyNumber()); |
- EXPECT_CALL(provider1_, RefreshPolicies()).Times(AnyNumber()); |
- EXPECT_CALL(provider2_, RefreshPolicies()).Times(AnyNumber()); |
- |
- EXPECT_CALL(*this, OnPolicyRefresh()).Times(0); |
- policy_service_->RefreshPolicies(base::Bind( |
- &PolicyServiceTest::OnPolicyRefresh, |
- base::Unretained(this))); |
- // Let any queued observer tasks run. |
- RunUntilIdle(); |
- Mock::VerifyAndClearExpectations(this); |
- |
- EXPECT_CALL(*this, OnPolicyRefresh()).Times(0); |
- base::FundamentalValue kValue0(0); |
- policy0_.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- kValue0.DeepCopy(), NULL); |
- provider0_.UpdateChromePolicy(policy0_); |
- Mock::VerifyAndClearExpectations(this); |
- |
- EXPECT_CALL(*this, OnPolicyRefresh()).Times(0); |
- base::FundamentalValue kValue1(1); |
- policy1_.Set("aaa", POLICY_LEVEL_RECOMMENDED, POLICY_SCOPE_USER, |
- kValue1.DeepCopy(), NULL); |
- provider1_.UpdateChromePolicy(policy1_); |
- Mock::VerifyAndClearExpectations(this); |
- |
- // A provider can refresh more than once after a RefreshPolicies call, but |
- // OnPolicyRefresh should be triggered only after all providers are |
- // refreshed. |
- EXPECT_CALL(*this, OnPolicyRefresh()).Times(0); |
- policy1_.Set("bbb", POLICY_LEVEL_RECOMMENDED, POLICY_SCOPE_USER, |
- kValue1.DeepCopy(), NULL); |
- provider1_.UpdateChromePolicy(policy1_); |
- Mock::VerifyAndClearExpectations(this); |
- |
- // If another RefreshPolicies() call happens while waiting for a previous |
- // one to complete, then all providers must refresh again. |
- EXPECT_CALL(*this, OnPolicyRefresh()).Times(0); |
- policy_service_->RefreshPolicies(base::Bind( |
- &PolicyServiceTest::OnPolicyRefresh, |
- base::Unretained(this))); |
- RunUntilIdle(); |
- Mock::VerifyAndClearExpectations(this); |
- |
- EXPECT_CALL(*this, OnPolicyRefresh()).Times(0); |
- policy2_.Set("bbb", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- kValue0.DeepCopy(), NULL); |
- provider2_.UpdateChromePolicy(policy2_); |
- Mock::VerifyAndClearExpectations(this); |
- |
- // Providers 0 and 1 must reload again. |
- EXPECT_CALL(*this, OnPolicyRefresh()).Times(2); |
- base::FundamentalValue kValue2(2); |
- policy0_.Set("aaa", POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- kValue2.DeepCopy(), NULL); |
- provider0_.UpdateChromePolicy(policy0_); |
- provider1_.UpdateChromePolicy(policy1_); |
- Mock::VerifyAndClearExpectations(this); |
- |
- const PolicyMap& policies = policy_service_->GetPolicies( |
- PolicyNamespace(POLICY_DOMAIN_CHROME, std::string())); |
- EXPECT_TRUE(base::Value::Equals(&kValue2, policies.GetValue("aaa"))); |
- EXPECT_TRUE(base::Value::Equals(&kValue0, policies.GetValue("bbb"))); |
-} |
- |
-TEST_F(PolicyServiceTest, NamespaceMerge) { |
- scoped_ptr<PolicyBundle> bundle0(new PolicyBundle()); |
- scoped_ptr<PolicyBundle> bundle1(new PolicyBundle()); |
- scoped_ptr<PolicyBundle> bundle2(new PolicyBundle()); |
- |
- AddTestPolicies(bundle0.get(), "bundle0", |
- POLICY_LEVEL_RECOMMENDED, POLICY_SCOPE_USER); |
- AddTestPolicies(bundle1.get(), "bundle1", |
- POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER); |
- AddTestPolicies(bundle2.get(), "bundle2", |
- POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE); |
- |
- provider0_.UpdatePolicy(bundle0.Pass()); |
- provider1_.UpdatePolicy(bundle1.Pass()); |
- provider2_.UpdatePolicy(bundle2.Pass()); |
- RunUntilIdle(); |
- |
- PolicyMap expected; |
- // For policies of the same level and scope, the first provider takes |
- // precedence, on every namespace. |
- expected.Set(kSameLevelPolicy, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- base::Value::CreateStringValue("bundle0"), NULL); |
- // For policies with different levels and scopes, the highest priority |
- // level/scope combination takes precedence, on every namespace. |
- expected.Set(kDiffLevelPolicy, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
- base::Value::CreateStringValue("bundle2"), NULL); |
- EXPECT_TRUE(policy_service_->GetPolicies( |
- PolicyNamespace(POLICY_DOMAIN_CHROME, std::string())).Equals(expected)); |
- EXPECT_TRUE(policy_service_->GetPolicies( |
- PolicyNamespace(POLICY_DOMAIN_EXTENSIONS, kExtension)).Equals(expected)); |
-} |
- |
-TEST_F(PolicyServiceTest, PolicyPreprocessing) { |
- // Reset the PolicyServiceImpl to one that has the preprocessor. |
- PolicyServiceImpl::Providers providers; |
- providers.push_back(&provider0_); |
- policy_service_.reset(new PolicyServiceImpl( |
- providers, base::Bind(&SetPolicyMapValue, kSameLevelPolicy, "bar"))); |
- |
- // Set the policy value to "foo". |
- scoped_ptr<PolicyBundle> bundle(new PolicyBundle()); |
- PolicyMap& map = |
- bundle->Get(PolicyNamespace(POLICY_DOMAIN_CHROME, std::string())); |
- map.Set(kSameLevelPolicy, |
- POLICY_LEVEL_MANDATORY, |
- POLICY_SCOPE_USER, |
- base::Value::CreateStringValue("foo"), |
- NULL); |
- |
- // Push the update through the provider. |
- provider0_.UpdatePolicy(bundle.Pass()); |
- RunUntilIdle(); |
- |
- // The value should have been changed from "foo" to "bar". |
- const PolicyMap& actual = policy_service_->GetPolicies( |
- PolicyNamespace(POLICY_DOMAIN_CHROME, std::string())); |
- PolicyMap expected; |
- expected.Set(kSameLevelPolicy, |
- POLICY_LEVEL_MANDATORY, |
- POLICY_SCOPE_USER, |
- base::Value::CreateStringValue("bar"), |
- NULL); |
- EXPECT_TRUE(actual.Equals(expected)); |
-} |
- |
-TEST_F(PolicyServiceTest, IsInitializationComplete) { |
- // |provider0| has all domains initialized. |
- Mock::VerifyAndClearExpectations(&provider1_); |
- Mock::VerifyAndClearExpectations(&provider2_); |
- EXPECT_CALL(provider1_, IsInitializationComplete(_)) |
- .WillRepeatedly(Return(false)); |
- EXPECT_CALL(provider2_, IsInitializationComplete(_)) |
- .WillRepeatedly(Return(false)); |
- PolicyServiceImpl::Providers providers; |
- providers.push_back(&provider0_); |
- providers.push_back(&provider1_); |
- providers.push_back(&provider2_); |
- policy_service_.reset(new PolicyServiceImpl( |
- providers, PolicyServiceImpl::PreprocessCallback())); |
- EXPECT_FALSE(policy_service_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); |
- EXPECT_FALSE( |
- policy_service_->IsInitializationComplete(POLICY_DOMAIN_EXTENSIONS)); |
- |
- // |provider2_| still doesn't have POLICY_DOMAIN_CHROME initialized, so |
- // the initialization status of that domain won't change. |
- MockPolicyServiceObserver observer; |
- policy_service_->AddObserver(POLICY_DOMAIN_CHROME, &observer); |
- policy_service_->AddObserver(POLICY_DOMAIN_EXTENSIONS, &observer); |
- EXPECT_CALL(observer, OnPolicyServiceInitialized(_)).Times(0); |
- Mock::VerifyAndClearExpectations(&provider1_); |
- EXPECT_CALL(provider1_, IsInitializationComplete(POLICY_DOMAIN_CHROME)) |
- .WillRepeatedly(Return(true)); |
- EXPECT_CALL(provider1_, IsInitializationComplete(POLICY_DOMAIN_EXTENSIONS)) |
- .WillRepeatedly(Return(false)); |
- const PolicyMap kPolicyMap; |
- provider1_.UpdateChromePolicy(kPolicyMap); |
- Mock::VerifyAndClearExpectations(&observer); |
- EXPECT_FALSE(policy_service_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); |
- EXPECT_FALSE( |
- policy_service_->IsInitializationComplete(POLICY_DOMAIN_EXTENSIONS)); |
- |
- // Same if |provider1_| doesn't have POLICY_DOMAIN_EXTENSIONS initialized. |
- EXPECT_CALL(observer, OnPolicyServiceInitialized(_)).Times(0); |
- Mock::VerifyAndClearExpectations(&provider2_); |
- EXPECT_CALL(provider2_, IsInitializationComplete(POLICY_DOMAIN_CHROME)) |
- .WillRepeatedly(Return(false)); |
- EXPECT_CALL(provider2_, IsInitializationComplete(POLICY_DOMAIN_EXTENSIONS)) |
- .WillRepeatedly(Return(true)); |
- provider2_.UpdateChromePolicy(kPolicyMap); |
- Mock::VerifyAndClearExpectations(&observer); |
- EXPECT_FALSE(policy_service_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); |
- EXPECT_FALSE( |
- policy_service_->IsInitializationComplete(POLICY_DOMAIN_EXTENSIONS)); |
- |
- // Now initialize POLICY_DOMAIN_CHROME on all the providers. |
- EXPECT_CALL(observer, OnPolicyServiceInitialized(POLICY_DOMAIN_CHROME)); |
- Mock::VerifyAndClearExpectations(&provider2_); |
- EXPECT_CALL(provider2_, IsInitializationComplete(POLICY_DOMAIN_CHROME)) |
- .WillRepeatedly(Return(true)); |
- EXPECT_CALL(provider2_, IsInitializationComplete(POLICY_DOMAIN_EXTENSIONS)) |
- .WillRepeatedly(Return(true)); |
- provider2_.UpdateChromePolicy(kPolicyMap); |
- Mock::VerifyAndClearExpectations(&observer); |
- EXPECT_TRUE(policy_service_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); |
- // Other domains are still not initialized. |
- EXPECT_FALSE( |
- policy_service_->IsInitializationComplete(POLICY_DOMAIN_EXTENSIONS)); |
- |
- // Initialize the remaining domain. |
- EXPECT_CALL(observer, OnPolicyServiceInitialized(POLICY_DOMAIN_EXTENSIONS)); |
- Mock::VerifyAndClearExpectations(&provider1_); |
- EXPECT_CALL(provider1_, IsInitializationComplete(POLICY_DOMAIN_CHROME)) |
- .WillRepeatedly(Return(true)); |
- EXPECT_CALL(provider1_, IsInitializationComplete(POLICY_DOMAIN_EXTENSIONS)) |
- .WillRepeatedly(Return(true)); |
- provider1_.UpdateChromePolicy(kPolicyMap); |
- Mock::VerifyAndClearExpectations(&observer); |
- EXPECT_TRUE(policy_service_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); |
- EXPECT_TRUE( |
- policy_service_->IsInitializationComplete(POLICY_DOMAIN_EXTENSIONS)); |
- |
- // Cleanup. |
- policy_service_->RemoveObserver(POLICY_DOMAIN_CHROME, &observer); |
- policy_service_->RemoveObserver(POLICY_DOMAIN_EXTENSIONS, &observer); |
-} |
- |
-} // namespace policy |