| // Copyright 2014 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/profile_policy_connector.h" |
| |
| #include "base/message_loop/message_loop.h" |
| #include "base/run_loop.h" |
| #include "base/values.h" |
| #include "chrome/test/base/testing_browser_process.h" |
| #include "components/autofill/core/common/autofill_pref_names.h" |
| #include "components/policy/core/browser/browser_policy_connector.h" |
| #include "components/policy/core/common/cloud/cloud_policy_constants.h" |
| #include "components/policy/core/common/cloud/cloud_policy_manager.h" |
| #include "components/policy/core/common/cloud/mock_cloud_policy_store.h" |
| #include "components/policy/core/common/mock_configuration_policy_provider.h" |
| #include "components/policy/core/common/policy_bundle.h" |
| #include "components/policy/core/common/policy_map.h" |
| #include "components/policy/core/common/policy_service.h" |
| #include "components/policy/core/common/schema_registry.h" |
| #include "policy/policy_constants.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| using testing::Return; |
| using testing::_; |
| |
| namespace policy { |
| |
| class ProfilePolicyConnectorTest : public testing::Test { |
| protected: |
| ProfilePolicyConnectorTest() {} |
| virtual ~ProfilePolicyConnectorTest() {} |
| |
| virtual void SetUp() OVERRIDE { |
| // This must be set up before the TestingBrowserProcess is created. |
| BrowserPolicyConnector::SetPolicyProviderForTesting(&mock_provider_); |
| |
| EXPECT_CALL(mock_provider_, IsInitializationComplete(_)) |
| .WillRepeatedly(Return(true)); |
| |
| cloud_policy_store_.NotifyStoreLoaded(); |
| cloud_policy_manager_.reset( |
| new CloudPolicyManager(PolicyNamespaceKey("", ""), |
| &cloud_policy_store_, |
| loop_.message_loop_proxy(), |
| loop_.message_loop_proxy(), |
| loop_.message_loop_proxy())); |
| } |
| |
| virtual void TearDown() { |
| TestingBrowserProcess::GetGlobal()->SetBrowserPolicyConnector(NULL); |
| cloud_policy_manager_->Shutdown(); |
| } |
| |
| base::MessageLoop loop_; |
| SchemaRegistry schema_registry_; |
| MockConfigurationPolicyProvider mock_provider_; |
| MockCloudPolicyStore cloud_policy_store_; |
| scoped_ptr<CloudPolicyManager> cloud_policy_manager_; |
| }; |
| |
| TEST_F(ProfilePolicyConnectorTest, IsPolicyFromCloudPolicy) { |
| ProfilePolicyConnector connector; |
| connector.Init(false, |
| #if defined(OS_CHROMEOS) |
| NULL, |
| #endif |
| &schema_registry_, |
| cloud_policy_manager_.get()); |
| |
| // No policy is set initially. |
| EXPECT_FALSE( |
| connector.IsPolicyFromCloudPolicy(autofill::prefs::kAutofillEnabled)); |
| PolicyNamespace chrome_ns(POLICY_DOMAIN_CHROME, ""); |
| EXPECT_FALSE(connector.policy_service()->GetPolicies(chrome_ns).GetValue( |
| key::kAutoFillEnabled)); |
| |
| // Set the policy at the cloud provider. |
| cloud_policy_store_.policy_map_.Set(key::kAutoFillEnabled, |
| POLICY_LEVEL_MANDATORY, |
| POLICY_SCOPE_USER, |
| new base::FundamentalValue(false), |
| NULL); |
| cloud_policy_store_.NotifyStoreLoaded(); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(connector.IsPolicyFromCloudPolicy(key::kAutoFillEnabled)); |
| const base::Value* value = |
| connector.policy_service()->GetPolicies(chrome_ns).GetValue( |
| key::kAutoFillEnabled); |
| ASSERT_TRUE(value); |
| EXPECT_TRUE(base::FundamentalValue(false).Equals(value)); |
| |
| // Now test with a higher-priority provider also setting the policy. |
| PolicyMap map; |
| map.Set(key::kAutoFillEnabled, |
| POLICY_LEVEL_MANDATORY, |
| POLICY_SCOPE_USER, |
| new base::FundamentalValue(true), |
| NULL); |
| mock_provider_.UpdateChromePolicy(map); |
| EXPECT_FALSE(connector.IsPolicyFromCloudPolicy(key::kAutoFillEnabled)); |
| value = connector.policy_service()->GetPolicies(chrome_ns).GetValue( |
| key::kAutoFillEnabled); |
| ASSERT_TRUE(value); |
| EXPECT_TRUE(base::FundamentalValue(true).Equals(value)); |
| |
| // Cleanup. |
| connector.Shutdown(); |
| } |
| |
| } // namespace policy |