Migrate Extensions code to Task Scheduler API

chrome/browser/extensions/api/storage/policy_value_store_unittest.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 "chrome/browser/extensions/api/storage/policy_value_store.h"
 
 #include <memory>
 
 #include "base/callback.h"
 #include "base/files/file_path.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/memory/ref_counted.h"
-#include "base/run_loop.h"
 #include "components/policy/core/common/external_data_fetcher.h"
 #include "components/policy/core/common/policy_map.h"
 #include "components/policy/core/common/policy_types.h"
 #include "content/public/test/test_browser_thread_bundle.h"
+#include "content/public/test/test_utils.h"
+#include "extensions/browser/api/storage/backend_task_runner.h"
 #include "extensions/browser/api/storage/settings_observer.h"
 #include "extensions/browser/value_store/leveldb_value_store.h"
 #include "extensions/browser/value_store/value_store_unittest.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::_;
 using testing::Mock;
 
 namespace extensions {
@@ skipping 74 matching lines @@
         base::MakeUnique<LeveldbValueStore>(kDatabaseUMAClientName,
                                             scoped_temp_dir_.GetPath())));
   }
 
   void TearDown() override {
     observers_->RemoveObserver(&observer_);
     store_.reset();
   }
 
  protected:
+  void SetCurrentPolicy(const policy::PolicyMap& policies) {
+    GetBackendTaskRunner()->PostTask(
+        FROM_HERE,
+        base::Bind(&PolicyValueStoreTest::SetCurrentPolicyOnBackendSequence,
+                   base::Unretained(this), base::Passed(policies.DeepCopy())));
+    content::RunAllBlockingPoolTasksUntilIdle();
+  }
+
+  void SetCurrentPolicyOnBackendSequence(
+      std::unique_ptr<policy::PolicyMap> policies) {
+    DCHECK(IsOnBackendSequence());
+    store_->SetCurrentPolicy(*policies);
+  }
+
   base::ScopedTempDir scoped_temp_dir_;
   content::TestBrowserThreadBundle test_browser_thread_bundle_;
   std::unique_ptr<PolicyValueStore> store_;
   MockSettingsObserver observer_;
   scoped_refptr<SettingsObserverList> observers_;
 };
 
 TEST_F(PolicyValueStoreTest, DontProvideRecommendedPolicies) {
   policy::PolicyMap policies;
   base::Value expected(123);
   policies.Set("must", policy::POLICY_LEVEL_MANDATORY,
                policy::POLICY_SCOPE_USER, policy::POLICY_SOURCE_CLOUD,
                expected.CreateDeepCopy(), nullptr);
   policies.Set("may", policy::POLICY_LEVEL_RECOMMENDED,
                policy::POLICY_SCOPE_USER, policy::POLICY_SOURCE_CLOUD,
                base::MakeUnique<base::Value>(456), nullptr);
-  store_->SetCurrentPolicy(policies);
+  SetCurrentPolicy(policies);
+
   ValueStore::ReadResult result = store_->Get();
   ASSERT_TRUE(result->status().ok());
   EXPECT_EQ(1u, result->settings().size());
   base::Value* value = NULL;
   EXPECT_FALSE(result->settings().Get("may", &value));
   EXPECT_TRUE(result->settings().Get("must", &value));
   EXPECT_TRUE(base::Value::Equals(&expected, value));
 }
 
 TEST_F(PolicyValueStoreTest, ReadOnly) {
@@ skipping 21 matching lines @@
     changes.push_back(ValueStoreChange("aaa", nullptr, value.CreateDeepCopy()));
     EXPECT_CALL(observer_,
                 OnSettingsChanged(kTestExtensionId,
                                   settings_namespace::MANAGED,
                                   ValueStoreChange::ToJson(changes)));
   }
 
   policy::PolicyMap policies;
   policies.Set("aaa", policy::POLICY_LEVEL_MANDATORY, policy::POLICY_SCOPE_USER,
                policy::POLICY_SOURCE_CLOUD, value.CreateDeepCopy(), nullptr);
-  store_->SetCurrentPolicy(policies);
-  base::RunLoop().RunUntilIdle();
+  SetCurrentPolicy(policies);
   Mock::VerifyAndClearExpectations(&observer_);
 
   // Notify when new policies are added.
   {
     ValueStoreChangeList changes;
     changes.push_back(ValueStoreChange("bbb", nullptr, value.CreateDeepCopy()));
     EXPECT_CALL(observer_,
                 OnSettingsChanged(kTestExtensionId,
                                   settings_namespace::MANAGED,
                                   ValueStoreChange::ToJson(changes)));
   }
 
   policies.Set("bbb", policy::POLICY_LEVEL_MANDATORY, policy::POLICY_SCOPE_USER,
                policy::POLICY_SOURCE_CLOUD, value.CreateDeepCopy(), nullptr);
-  store_->SetCurrentPolicy(policies);
-  base::RunLoop().RunUntilIdle();
+  SetCurrentPolicy(policies);
   Mock::VerifyAndClearExpectations(&observer_);
 
   // Notify when policies change.
   const base::Value new_value("222");
   {
     ValueStoreChangeList changes;
     changes.push_back(ValueStoreChange("bbb", value.CreateDeepCopy(),
                                        new_value.CreateDeepCopy()));
     EXPECT_CALL(observer_,
                 OnSettingsChanged(kTestExtensionId,
                                   settings_namespace::MANAGED,
                                   ValueStoreChange::ToJson(changes)));
   }
 
   policies.Set("bbb", policy::POLICY_LEVEL_MANDATORY, policy::POLICY_SCOPE_USER,
                policy::POLICY_SOURCE_CLOUD, new_value.CreateDeepCopy(),
                nullptr);
-  store_->SetCurrentPolicy(policies);
-  base::RunLoop().RunUntilIdle();
+  SetCurrentPolicy(policies);
   Mock::VerifyAndClearExpectations(&observer_);
 
   // Notify when policies are removed.
   {
     ValueStoreChangeList changes;
     changes.push_back(
         ValueStoreChange("bbb", new_value.CreateDeepCopy(), nullptr));
     EXPECT_CALL(observer_,
                 OnSettingsChanged(kTestExtensionId,
                                   settings_namespace::MANAGED,
                                   ValueStoreChange::ToJson(changes)));
   }
 
   policies.Erase("bbb");
-  store_->SetCurrentPolicy(policies);
-  base::RunLoop().RunUntilIdle();
+  SetCurrentPolicy(policies);
   Mock::VerifyAndClearExpectations(&observer_);
 
   // Don't notify when there aren't any changes.
   EXPECT_CALL(observer_, OnSettingsChanged(_, _, _)).Times(0);
-  store_->SetCurrentPolicy(policies);
-  base::RunLoop().RunUntilIdle();
+  SetCurrentPolicy(policies);
   Mock::VerifyAndClearExpectations(&observer_);
 }
 
 }  // namespace extensions