arc: Make Location Service manageable

chrome/browser/policy/default_geolocation_policy_handler_unittest.cc

Index: chrome/browser/policy/default_geolocation_policy_handler_unittest.cc
diff --git a/chrome/browser/policy/default_geolocation_policy_handler_unittest.cc b/chrome/browser/policy/default_geolocation_policy_handler_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..817862597e1dff58145870016bf68119ed53f859
--- /dev/null
+++ b/chrome/browser/policy/default_geolocation_policy_handler_unittest.cc
@@ -0,0 +1,73 @@
+// Copyright 2016 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/default_geolocation_policy_handler.h"
+
+#include "base/memory/ptr_util.h"
+#include "base/values.h"
+#include "chrome/common/pref_names.h"
+#include "components/content_settings/core/common/content_settings.h"
+#include "components/policy/core/browser/configuration_policy_pref_store.h"
+#include "components/policy/core/browser/configuration_policy_pref_store_test.h"
+#include "components/policy/core/common/policy_map.h"
+#include "components/policy/core/common/policy_types.h"
+#include "policy/policy_constants.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace policy {
+
+class DefaultGeolocationPolicyHandlerTest
+    : public ConfigurationPolicyPrefStoreTest {
+  void SetUp() override {
+    handler_list_.AddHandler(base::WrapUnique<ConfigurationPolicyHandler>(
+        new DefaultGeolocationPolicyHandler));
+  }
+};
+
+TEST_F(DefaultGeolocationPolicyHandlerTest, AllowGeolocation) {
+  // DefaultGeolocationSetting of CONTENT_SETTING_ALLOW (AllowGeolocation)
+  // should not translate to the ArcLocationServiceEnabled preference.
+  EXPECT_FALSE(store_->GetValue(prefs::kArcLocationServiceEnabled, nullptr));
+  PolicyMap policy;
+  policy.Set(key::kDefaultGeolocationSetting,
+      POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER,
+      POLICY_SOURCE_CLOUD,
+      base::WrapUnique(new base::FundamentalValue(CONTENT_SETTING_ALLOW)),
+      nullptr);
+  UpdateProviderPolicy(policy);
+  EXPECT_FALSE(store_->GetValue(prefs::kArcLocationServiceEnabled, nullptr));
+}
+
+TEST_F(DefaultGeolocationPolicyHandlerTest, BlockGeolocation) {
+  // DefaultGeolocationSetting of CONTENT_SETTING_BLOCK (BlockGeolocation)
+  // should set the ArcLocationServiceEnabled preference to false.
+  EXPECT_FALSE(store_->GetValue(prefs::kArcLocationServiceEnabled, nullptr));
+  PolicyMap policy;
+  policy.Set(key::kDefaultGeolocationSetting,
+      POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER,
+      POLICY_SOURCE_CLOUD,
+      base::WrapUnique(new base::FundamentalValue(CONTENT_SETTING_BLOCK)),
+      nullptr);
+  UpdateProviderPolicy(policy);
+  const base::Value* value = nullptr;
+  EXPECT_TRUE(store_->GetValue(prefs::kArcLocationServiceEnabled, &value));
+  EXPECT_TRUE(base::FundamentalValue(false).Equals(value));
+}
+
+TEST_F(DefaultGeolocationPolicyHandlerTest, AskGeolocation) {
+  // DefaultGeolocationSetting of CONTENT_SETTING_ASK (AskGeolocation) should
+  // not translate to the ArcLocationServiceEnabled preference.
+  EXPECT_FALSE(store_->GetValue(prefs::kArcLocationServiceEnabled, nullptr));
+  PolicyMap policy;
+  policy.Set(key::kDefaultGeolocationSetting,
+      POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER,
+      POLICY_SOURCE_CLOUD,
+      base::WrapUnique(new base::FundamentalValue(CONTENT_SETTING_ASK)),
+      nullptr);
+  UpdateProviderPolicy(policy);
+  EXPECT_FALSE(store_->GetValue(prefs::kArcLocationServiceEnabled, nullptr));
+}
+
+}  // namespace policy
+