Add policies to control power management on the Chrome OS login screen

chrome/browser/policy/configuration_policy_handler_unittest.cc

 // 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 "base/memory/scoped_ptr.h"
 #include "base/prefs/pref_value_map.h"
 #include "chrome/browser/extensions/external_policy_loader.h"
 #include "chrome/browser/policy/configuration_policy_handler.h"
 #include "chrome/browser/policy/policy_error_map.h"
 #include "chrome/browser/policy/policy_map.h"
 #include "chrome/common/pref_names.h"
 #include "policy/policy_constants.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace policy {
 
 namespace {
 
 StringToIntEnumListPolicyHandler::MappingEntry kTestTypeMap[] = {
   { "one", 1 },
   { "two", 2 },
 };
 
+int kTestAllowedSet[] = { 0, 2 };
+
 const char kTestPref[] = "unit_test.test_pref";
 
 }  // namespace
 
 TEST(StringToIntEnumListPolicyHandlerTest, CheckPolicySettings) {
   base::ListValue list;
   PolicyMap policy_map;
   PolicyErrorMap errors;
   StringToIntEnumListPolicyHandler handler(
       key::kExtensionAllowedTypes, prefs::kExtensionAllowedTypes,
@@ skipping 437 matching lines @@
 
   policy_map.Set(key::kDiskCacheSize, POLICY_LEVEL_MANDATORY,
                  POLICY_SCOPE_USER, base::Value::CreateIntegerValue(10));
   prefs.Clear();
   handler.ApplyPolicySettings(policy_map, &prefs);
   expected.reset(base::Value::CreateDoubleValue(0.1));
   EXPECT_TRUE(prefs.GetValue(kTestPref, &value));
   EXPECT_TRUE(base::Value::Equals(expected.get(), value));
 }
 
+TEST(IntSetPolicyHandlerTest, CheckPolicySettings) {
+  PolicyMap policy_map;
+  PolicyErrorMap errors;
+
+  // This tests needs to modify an int policy. The exact policy used and its
+  // semantics outside the test are irrelevant.
+  IntSetPolicyHandler handler(
+      key::kDiskCacheSize, kTestPref,
+      kTestAllowedSet, kTestAllowedSet + arraysize(kTestAllowedSet));
+
+  // Check that values lying in the accepted set are not rejected.
+  policy_map.Set(key::kDiskCacheSize, POLICY_LEVEL_MANDATORY,
+                 POLICY_SCOPE_USER, base::Value::CreateIntegerValue(0));
+  errors.Clear();
+  EXPECT_TRUE(handler.CheckPolicySettings(policy_map, &errors));
+  EXPECT_TRUE(errors.empty());
+
+  policy_map.Set(key::kDiskCacheSize, POLICY_LEVEL_MANDATORY,
+                 POLICY_SCOPE_USER, base::Value::CreateIntegerValue(2));
+  errors.Clear();
+  EXPECT_TRUE(handler.CheckPolicySettings(policy_map, &errors));
+  EXPECT_TRUE(errors.empty());
+
+  // Check that values not found in the accepted set are rejected and yield an
+  // error message.
+  policy_map.Set(key::kDiskCacheSize, POLICY_LEVEL_MANDATORY,
+                 POLICY_SCOPE_USER, base::Value::CreateIntegerValue(1));
+  errors.Clear();
+  EXPECT_FALSE(handler.CheckPolicySettings(policy_map, &errors));
+  EXPECT_FALSE(errors.empty());
+
+  policy_map.Set(key::kDiskCacheSize, POLICY_LEVEL_MANDATORY,
+                 POLICY_SCOPE_USER, base::Value::CreateIntegerValue(3));
+  errors.Clear();
+  EXPECT_FALSE(handler.CheckPolicySettings(policy_map, &errors));
+  EXPECT_FALSE(errors.empty());
+
+  // Check that an entirely invalid value is rejected and yields an error
+  // message.
+  policy_map.Set(key::kDiskCacheSize, POLICY_LEVEL_MANDATORY,
+                 POLICY_SCOPE_USER, base::Value::CreateStringValue("invalid"));
+  errors.Clear();
+  EXPECT_FALSE(handler.CheckPolicySettings(policy_map, &errors));
+  EXPECT_FALSE(errors.empty());
+}
+
+TEST(IntSetPolicyHandlerTest, ApplyPolicySettings) {
+  PolicyMap policy_map;
+  PrefValueMap prefs;
+  scoped_ptr<base::Value> expected;
+  const base::Value* value;
+
+  // This tests needs to modify an int policy. The exact policy used and its
+  // semantics outside the test are irrelevant.
+  IntSetPolicyHandler handler(
+      key::kDiskCacheSize, kTestPref,
+      kTestAllowedSet, kTestAllowedSet + arraysize(kTestAllowedSet));
+
+  // Check that values lying in the accepted set are written to the pref.
+  policy_map.Set(key::kDiskCacheSize, POLICY_LEVEL_MANDATORY,
+                 POLICY_SCOPE_USER, base::Value::CreateIntegerValue(0));
+  prefs.Clear();
+  handler.ApplyPolicySettings(policy_map, &prefs);
+  expected.reset(base::Value::CreateIntegerValue(0));
+  EXPECT_TRUE(prefs.GetValue(kTestPref, &value));
+  EXPECT_TRUE(base::Value::Equals(expected.get(), value));
+
+  policy_map.Set(key::kDiskCacheSize, POLICY_LEVEL_MANDATORY,
+                 POLICY_SCOPE_USER, base::Value::CreateIntegerValue(2));
+  prefs.Clear();
+  handler.ApplyPolicySettings(policy_map, &prefs);
+  expected.reset(base::Value::CreateIntegerValue(2));
+  EXPECT_TRUE(prefs.GetValue(kTestPref, &value));
+  EXPECT_TRUE(base::Value::Equals(expected.get(), value));
+}
+
 TEST(ExtensionListPolicyHandlerTest, CheckPolicySettings) {
   base::ListValue list;
   PolicyMap policy_map;
   PolicyErrorMap errors;
   ExtensionListPolicyHandler handler(key::kExtensionInstallBlacklist,
                                      prefs::kExtensionInstallDenyList,
                                      true);
 
   policy_map.Set(key::kExtensionInstallBlacklist, POLICY_LEVEL_MANDATORY,
                  POLICY_SCOPE_USER, list.DeepCopy());
@@ skipping 189 matching lines @@
 
   list.Append(Value::CreateStringValue("https://corp.monkey.net/*"));
   policy_map.Set(key::kExtensionInstallSources, POLICY_LEVEL_MANDATORY,
                  POLICY_SCOPE_USER, list.DeepCopy());
   handler.ApplyPolicySettings(policy_map, &prefs);
   ASSERT_TRUE(prefs.GetValue(prefs::kExtensionAllowedInstallSites, &value));
   EXPECT_TRUE(base::Value::Equals(&list, value));
 }
 
 }  // namespace policy