1 files changed, 34 insertions, 0 deletions
diff --git a/chrome/browser/mock_configuration_policy_store.h b/chrome/browser/mock_configuration_policy_store.h
new file mode 100644
index 0000000..b1f3696
--- /dev/null
+++ b/chrome/browser/mock_configuration_policy_store.h
@@ -0,0 +1,34 @@
+// Copyright (c) 2010 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.
+
+#ifndef CHROME_BROWSER_MOCK_CONFIGURATION_POLICY_STORE_H_
+#define CHROME_BROWSER_MOCK_CONFIGURATION_POLICY_STORE_H_
+
+#include <map>
+
+#include "base/stl_util-inl.h"
+#include "chrome/browser/configuration_policy_store.h"
+
+// Mock ConfigurationPolicyStore implementation that records values for policy
+// settings as they get set.
+class MockConfigurationPolicyStore : public ConfigurationPolicyStore {
+ public:
+  MockConfigurationPolicyStore() {}
+  ~MockConfigurationPolicyStore() {
+    STLDeleteValues(&policy_map_);
+  }
+
+  typedef std::map<ConfigurationPolicyStore::PolicyType, Value*> PolicyMap;
+  const PolicyMap& policy_map() { return policy_map_; }
+
+  // ConfigurationPolicyStore implementation.
+  virtual void Apply(PolicyType policy, Value* value) {
+    policy_map_[policy] = value;
+  }
+
+ private:
+  PolicyMap policy_map_;
+};
+
+#endif  // CHROME_BROWSER_MOCK_CONFIGURATION_POLICY_STORE_H_