| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CHROME_BROWSER_POLICY_MOCK_CONFIGURATION_POLICY_PROVIDER_H_ | 5 #ifndef CHROME_BROWSER_POLICY_MOCK_CONFIGURATION_POLICY_PROVIDER_H_ |
| 6 #define CHROME_BROWSER_POLICY_MOCK_CONFIGURATION_POLICY_PROVIDER_H_ | 6 #define CHROME_BROWSER_POLICY_MOCK_CONFIGURATION_POLICY_PROVIDER_H_ |
| 7 #pragma once | 7 #pragma once |
| 8 | 8 |
| 9 #include <map> | 9 #include <map> |
| 10 #include <utility> | 10 #include <utility> |
| 11 | 11 |
| 12 #include "base/stl_util-inl.h" | 12 #include "base/stl_util-inl.h" |
| 13 #include "chrome/browser/policy/configuration_policy_provider.h" | 13 #include "chrome/browser/policy/configuration_policy_provider.h" |
| 14 | 14 |
| 15 namespace policy { | 15 namespace policy { |
| 16 | 16 |
| 17 // Mock ConfigurationPolicyProvider implementation that supplies canned | 17 // Mock ConfigurationPolicyProvider implementation that supplies canned |
| 18 // values for polices. | 18 // values for polices. |
| 19 class MockConfigurationPolicyProvider : public ConfigurationPolicyProvider { | 19 class MockConfigurationPolicyProvider : public ConfigurationPolicyProvider { |
| 20 public: | 20 public: |
| 21 MockConfigurationPolicyProvider() | 21 MockConfigurationPolicyProvider() |
| 22 : ConfigurationPolicyProvider( | 22 : ConfigurationPolicyProvider( |
| 23 ConfigurationPolicyPrefStore::GetChromePolicyValueMap()) { | 23 ConfigurationPolicyPrefStore::GetChromePolicyDefinitionList()) { |
| 24 } | 24 } |
| 25 ~MockConfigurationPolicyProvider() { | 25 ~MockConfigurationPolicyProvider() { |
| 26 STLDeleteValues(&policy_map_); | 26 STLDeleteValues(&policy_map_); |
| 27 } | 27 } |
| 28 | 28 |
| 29 typedef std::map<ConfigurationPolicyStore::PolicyType, Value*> PolicyMap; | 29 typedef std::map<ConfigurationPolicyStore::PolicyType, Value*> PolicyMap; |
| 30 | 30 |
| 31 void AddPolicy(ConfigurationPolicyStore::PolicyType policy, Value* value) { | 31 void AddPolicy(ConfigurationPolicyStore::PolicyType policy, Value* value) { |
| 32 std::swap(policy_map_[policy], value); | 32 std::swap(policy_map_[policy], value); |
| 33 delete value; | 33 delete value; |
| 34 } | 34 } |
| 35 | 35 |
| 36 // ConfigurationPolicyProvider method overrides. | 36 // ConfigurationPolicyProvider method overrides. |
| 37 virtual bool Provide(ConfigurationPolicyStore* store) { | 37 virtual bool Provide(ConfigurationPolicyStore* store) { |
| 38 for (PolicyMap::const_iterator current = policy_map_.begin(); | 38 for (PolicyMap::const_iterator current = policy_map_.begin(); |
| 39 current != policy_map_.end(); ++current) { | 39 current != policy_map_.end(); ++current) { |
| 40 store->Apply(current->first, current->second->DeepCopy()); | 40 store->Apply(current->first, current->second->DeepCopy()); |
| 41 } | 41 } |
| 42 return true; | 42 return true; |
| 43 } | 43 } |
| 44 | 44 |
| 45 private: | 45 private: |
| 46 PolicyMap policy_map_; | 46 PolicyMap policy_map_; |
| 47 }; | 47 }; |
| 48 | 48 |
| 49 } // namespace policy | 49 } // namespace policy |
| 50 | 50 |
| 51 #endif // CHROME_BROWSER_POLICY_MOCK_CONFIGURATION_POLICY_PROVIDER_H_ | 51 #endif // CHROME_BROWSER_POLICY_MOCK_CONFIGURATION_POLICY_PROVIDER_H_ |
| OLD | NEW |