15d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// Copyright 2014 The Chromium Authors. All rights reserved. 25d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// found in the LICENSE file. 45d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 55d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "chrome/browser/extensions/api/messaging/native_messaging_policy_handler.h" 65d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 75d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "base/prefs/pref_value_map.h" 85d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "chrome/browser/extensions/policy_handlers.h" 95d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "chrome/common/pref_names.h" 105d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "components/policy/core/browser/policy_error_map.h" 115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "components/policy/core/common/policy_map.h" 125d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "policy/policy_constants.h" 135d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 145d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 155d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)namespace extensions { 165d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 175d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)const char kTestPref[] = "unit_test.test_pref"; 185d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 195d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)TEST(NativeMessagingHostListPolicyHandlerTest, CheckPolicySettings) { 205d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::ListValue list; 215d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::PolicyMap policy_map; 225d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) NativeMessagingHostListPolicyHandler handler( 235d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::key::kNativeMessagingBlacklist, kTestPref, true); 245d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 255d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy_map.Set(policy::key::kNativeMessagingBlacklist, 265d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::POLICY_LEVEL_MANDATORY, 275d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::POLICY_SCOPE_USER, 285d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) list.DeepCopy(), 295d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) NULL); 305d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) { 315d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::PolicyErrorMap errors; 325d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_TRUE(handler.CheckPolicySettings(policy_map, &errors)); 335d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_TRUE(errors.empty()); 345d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 355d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 365d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) list.Append(new base::StringValue("test.a.b")); 375d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy_map.Set(policy::key::kNativeMessagingBlacklist, 385d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::POLICY_LEVEL_MANDATORY, 395d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::POLICY_SCOPE_USER, 405d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) list.DeepCopy(), 415d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) NULL); 425d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) { 435d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::PolicyErrorMap errors; 445d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_TRUE(handler.CheckPolicySettings(policy_map, &errors)); 455d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_TRUE(errors.empty()); 465d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 475d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 485d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) list.Append(new base::StringValue("*")); 495d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy_map.Set(policy::key::kNativeMessagingBlacklist, 505d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::POLICY_LEVEL_MANDATORY, 515d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::POLICY_SCOPE_USER, 525d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) list.DeepCopy(), 535d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) NULL); 545d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) { 555d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::PolicyErrorMap errors; 565d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_TRUE(handler.CheckPolicySettings(policy_map, &errors)); 575d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_TRUE(errors.empty()); 585d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 595d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 605d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) list.Append(new base::StringValue("invalid Name")); 615d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy_map.Set(policy::key::kNativeMessagingBlacklist, 625d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::POLICY_LEVEL_MANDATORY, 635d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::POLICY_SCOPE_USER, 645d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) list.DeepCopy(), 655d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) NULL); 665d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) { 675d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::PolicyErrorMap errors; 685d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_TRUE(handler.CheckPolicySettings(policy_map, &errors)); 695d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_FALSE(errors.empty()); 705d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_FALSE( 715d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) errors.GetErrors(policy::key::kNativeMessagingBlacklist).empty()); 725d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 735d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 745d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 755d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)TEST(NativeMessagingHostListPolicyHandlerTest, ApplyPolicySettings) { 765d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::ListValue policy; 775d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::ListValue expected; 785d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::PolicyMap policy_map; 795d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) PrefValueMap prefs; 805d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::Value* value = NULL; 815d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) NativeMessagingHostListPolicyHandler handler( 825d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::key::kNativeMessagingBlacklist, kTestPref, true); 835d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 845d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy.Append(new base::StringValue("com.example.test")); 855d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) expected.Append(new base::StringValue("com.example.test")); 865d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 875d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy_map.Set(policy::key::kNativeMessagingBlacklist, 885d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::POLICY_LEVEL_MANDATORY, 895d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::POLICY_SCOPE_USER, 905d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy.DeepCopy(), 915d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) NULL); 925d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) handler.ApplyPolicySettings(policy_map, &prefs); 935d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_TRUE(prefs.GetValue(kTestPref, &value)); 945d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_TRUE(base::Value::Equals(&expected, value)); 955d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 965d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy.Append(new base::StringValue("*")); 975d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) expected.Append(new base::StringValue("*")); 985d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 995d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy_map.Set(policy::key::kNativeMessagingBlacklist, 1005d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::POLICY_LEVEL_MANDATORY, 1015d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::POLICY_SCOPE_USER, 1025d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy.DeepCopy(), 1035d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) NULL); 1045d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) handler.ApplyPolicySettings(policy_map, &prefs); 1055d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_TRUE(prefs.GetValue(kTestPref, &value)); 1065d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_TRUE(base::Value::Equals(&expected, value)); 1075d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 1085d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy.Append(new base::StringValue("invalid Name")); 1095d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy_map.Set(policy::key::kNativeMessagingBlacklist, 1105d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::POLICY_LEVEL_MANDATORY, 1115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy::POLICY_SCOPE_USER, 1125d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) policy.DeepCopy(), 1135d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) NULL); 1145d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) handler.ApplyPolicySettings(policy_map, &prefs); 1155d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_TRUE(prefs.GetValue(kTestPref, &value)); 1165d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_TRUE(base::Value::Equals(&expected, value)); 1175d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 1185d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 1195d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} // namespace extensions 120