mock_configuration_policy_store.h revision bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293 
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
3// found in the LICENSE file.
4
5#ifndef CHROME_BROWSER_POLICY_MOCK_CONFIGURATION_POLICY_STORE_H_
6#define CHROME_BROWSER_POLICY_MOCK_CONFIGURATION_POLICY_STORE_H_
7#pragma once
8
9#include <map>
10#include <utility>
11
12#include "base/stl_util-inl.h"
13#include "chrome/browser/policy/configuration_policy_store.h"
14
15namespace policy {
16
17// Mock ConfigurationPolicyStore implementation that records values for policy
18// settings as they get set.
19class MockConfigurationPolicyStore : public ConfigurationPolicyStore {
20 public:
21  MockConfigurationPolicyStore() {}
22  ~MockConfigurationPolicyStore() {
23    STLDeleteValues(&policy_map_);
24  }
25
26  typedef std::map<ConfigurationPolicyStore::PolicyType, Value*> PolicyMap;
27  const PolicyMap& policy_map() { return policy_map_; }
28
29  // Get a value for the given policy. Returns NULL if that key doesn't exist.
30  const Value* Get(ConfigurationPolicyStore::PolicyType type) const {
31    PolicyMap::const_iterator entry(policy_map_.find(type));
32    return entry == policy_map_.end() ? NULL : entry->second;
33  }
34
35  // ConfigurationPolicyStore implementation.
36  virtual void Apply(PolicyType policy, Value* value) {
37    std::swap(policy_map_[policy], value);
38    delete value;
39  }
40
41 private:
42  PolicyMap policy_map_;
43};
44
45}  // namespace policy
46
47#endif  // CHROME_BROWSER_POLICY_MOCK_CONFIGURATION_POLICY_STORE_H_
48