Handle policy refresh internally in ConfigurationPolicyPrefStore.

chrome/browser/policy/mock_configuration_policy_provider.cc

Index: chrome/browser/policy/mock_configuration_policy_provider.cc
diff --git a/chrome/browser/policy/mock_configuration_policy_provider.cc b/chrome/browser/policy/mock_configuration_policy_provider.cc
index 9c2db8deb97c52b3da40e733c6d980908db7e336..a2566b050f53bce640515ff573f2faa7115f356c 100644
--- a/chrome/browser/policy/mock_configuration_policy_provider.cc
+++ b/chrome/browser/policy/mock_configuration_policy_provider.cc
@@ -11,7 +11,8 @@ namespace policy {
 
 MockConfigurationPolicyProvider::MockConfigurationPolicyProvider()
     : ConfigurationPolicyProvider(
-        ConfigurationPolicyPrefStore::GetChromePolicyDefinitionList()) {
+          ConfigurationPolicyPrefStore::GetChromePolicyDefinitionList()),
+      initialization_complete_(false) {
 }
 
 MockConfigurationPolicyProvider::~MockConfigurationPolicyProvider() {
@@ -24,6 +25,20 @@ void MockConfigurationPolicyProvider::AddPolicy(ConfigurationPolicyType policy,
   delete value;
 }
 
+void MockConfigurationPolicyProvider::RemovePolicy(
+    ConfigurationPolicyType policy) {
+  const PolicyMap::iterator entry = policy_map_.find(policy);
+  if (entry != policy_map_.end()) {
+    delete entry->second;
+    policy_map_.erase(entry);
+  }
+}
+
+void MockConfigurationPolicyProvider::SetInitializationComplete(
+    bool initialization_complete) {
+  initialization_complete_ = initialization_complete;
+}
+
 bool MockConfigurationPolicyProvider::Provide(
     ConfigurationPolicyStoreInterface* store) {
   for (PolicyMap::const_iterator current = policy_map_.begin();
@@ -33,4 +48,8 @@ bool MockConfigurationPolicyProvider::Provide(
   return true;
 }
 
+bool MockConfigurationPolicyProvider::IsInitializationComplete() const {
+  return initialization_complete_;
+}
+
 }