Added ProfilePolicyConnector::IsFromCloudPolicy().

chrome/browser/policy/profile_policy_connector_unittest.cc

+// 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