Trigger chrome.storage.onChanged events for policy updates on the 'managed' namespace.

chrome/browser/extensions/settings/settings_apitest.cc

 // Copyright (c) 2012 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.
 
 #include "base/bind.h"
 #include "base/json/json_writer.h"
 #include "chrome/browser/extensions/extension_apitest.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_test_message_listener.h"
 #include "chrome/browser/extensions/settings/settings_frontend.h"
@@ skipping 121 matching lines @@
   }
 
   void SendChanges(const syncer::SyncChangeList& change_list) {
     MessageLoop::current()->RunAllPending();
     SendChangesToSyncableService(
         change_list,
         browser()->profile()->GetExtensionService()->settings_frontend()->
               GetBackendForSync(kModelType));
   }
 
+#if defined(ENABLE_CONFIGURATION_POLICY)
+  void SetPolicies(const base::DictionaryValue& policies) {
+    scoped_ptr<policy::PolicyBundle> bundle(new policy::PolicyBundle());
+    policy::PolicyMap& policy_map = bundle->Get(
+        policy::POLICY_DOMAIN_EXTENSIONS, kManagedStorageExtensionId);
+    policy_map.LoadFrom(
+        &policies, policy::POLICY_LEVEL_MANDATORY, policy::POLICY_SCOPE_USER);
+    policy_provider_.UpdatePolicy(bundle.Pass());
+  }
+#endif
+
  private:
   const Extension* MaybeLoadAndReplyWhenSatisfied(
       Namespace settings_namespace,
       const std::string& normal_action,
       const std::string& incognito_action,
       // May be NULL to imply not loading the extension.
       const std::string* extension_dir,
       bool is_final_action) {
     ExtensionTestMessageListener listener("waiting", true);
     ExtensionTestMessageListener listener_incognito("waiting_incognito", true);
@@ skipping 270 matching lines @@
           .Append("two")
           .Append("three"))
       .Set("dict-policy", extensions::DictionaryBuilder()
           .Set("list", extensions::ListBuilder()
               .Append(extensions::DictionaryBuilder()
                   .Set("one", 1)
                   .Set("two", 2))
               .Append(extensions::DictionaryBuilder()
                   .Set("three", 3))))
       .Build();
-
-  scoped_ptr<policy::PolicyBundle> bundle(new policy::PolicyBundle());
-  policy::PolicyMap& policy_map =
-      bundle->Get(policy::POLICY_DOMAIN_EXTENSIONS, kManagedStorageExtensionId);
-  policy_map.LoadFrom(
-      policy.get(), policy::POLICY_LEVEL_MANDATORY, policy::POLICY_SCOPE_USER);
-  policy_provider_.UpdatePolicy(bundle.Pass());
-
+  SetPolicies(*policy);
   // Now run the extension.
   ASSERT_TRUE(RunExtensionTest("settings/managed_storage")) << message_;
 }
-#endif
+
+IN_PROC_BROWSER_TEST_F(ExtensionSettingsApiTest, ManagedStorageEvents) {
+  ResultCatcher catcher;
+
+  // Set policies for the test extension.
+  scoped_ptr<base::DictionaryValue> policy = extensions::DictionaryBuilder()
+      .Set("constant-policy", "aaa")
+      .Set("changes-policy", "bbb")
+      .Set("deleted-policy", "ccc")
+      .Build();
+  SetPolicies(*policy);
+
+  ExtensionTestMessageListener ready_listener("ready", false);
+  // Load the extension to install the event listener.
+  const Extension* extension = LoadExtension(
+      test_data_dir_.AppendASCII("settings/managed_storage_events"));
+  ASSERT_TRUE(extension);
+  // Wait until the extension sends the "ready" message.
+  ASSERT_TRUE(ready_listener.WaitUntilSatisfied());
+
+  // Now change the policies and wait until the extension is done.
+  policy = extensions::DictionaryBuilder()
+      .Set("constant-policy", "aaa")
+      .Set("changes-policy", "ddd")
+      .Set("new-policy", "eee")
+      .Build();
+  SetPolicies(*policy);
+  EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
+}
+#endif  // defined(ENABLE_CONFIGURATION_POLICY)
 
 IN_PROC_BROWSER_TEST_F(ExtensionSettingsApiTest, ManagedStorageDisabled) {
   // Disable the 'managed' namespace. This is redundant when
   // ENABLE_CONFIGURATION_POLICY is not defined.
   SettingsFrontend* frontend =
       browser()->profile()->GetExtensionService()->settings_frontend();
   frontend->DisableStorageForTesting(MANAGED);
   EXPECT_FALSE(frontend->IsStorageEnabled(MANAGED));
-
-  // Set a policy for the extension.
-  scoped_ptr<policy::PolicyBundle> bundle(new policy::PolicyBundle());
-  policy::PolicyMap& policy_map =
-      bundle->Get(policy::POLICY_DOMAIN_EXTENSIONS, kManagedStorageExtensionId);
-  policy_map.Set(
-      "policy", policy::POLICY_LEVEL_MANDATORY, policy::POLICY_SCOPE_USER,
-      base::Value::CreateStringValue("policy_value"));
-  policy_provider_.UpdatePolicy(bundle.Pass());
-
   // Now run the extension.
   ASSERT_TRUE(RunExtensionTest("settings/managed_storage_disabled"))
       << message_;
 }
 
 }  // namespace extensions