Remove MessageLoop::current()->RunUntilIdle() in components.

components/safe_browsing_db/v4_database_unittest.cc

 // Copyright 2016 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/debug/leak_annotations.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/memory/ptr_util.h"
+#include "base/run_loop.h"
 #include "base/test/test_simple_task_runner.h"
 #include "components/safe_browsing_db/v4_database.h"
 #include "components/safe_browsing_db/v4_store.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "testing/platform_test.h"
 
 namespace safe_browsing {
 
 class FakeV4Store : public V4Store {
  public:
@@ skipping 110 matching lines @@
 TEST_F(SafeBrowsingV4DatabaseTest, TestSetupDatabaseWithNoStores) {
   NewDatabaseReadyCallback callback_db_ready = base::Bind(
       &SafeBrowsingV4DatabaseTest::NewDatabaseReadyWithExpectedStorePathsAndIds,
       base::Unretained(this), expected_store_paths_, expected_identifiers_,
       true);
   V4Database::Create(task_runner_, database_dirname_, list_info_map_,
                      callback_db_ready);
   created_but_not_called_back_ = true;
   task_runner_->RunPendingTasks();
 
-  base::MessageLoop::current()->RunUntilIdle();
+  base::RunLoop().RunUntilIdle();
   EXPECT_EQ(true, created_and_called_back_);
 }
 
 // Test to set up the database with fake stores.
 TEST_F(SafeBrowsingV4DatabaseTest, TestSetupDatabaseWithFakeStores) {
   SetupInfoMapAndExpectedState();
 
   NewDatabaseReadyCallback callback_db_ready = base::Bind(
       &SafeBrowsingV4DatabaseTest::NewDatabaseReadyWithExpectedStorePathsAndIds,
       base::Unretained(this), expected_store_paths_, expected_identifiers_,
       true);
 
   FakeV4StoreFactory* factory = new FakeV4StoreFactory(false);
   ANNOTATE_LEAKING_OBJECT_PTR(factory);
   V4Database::RegisterStoreFactoryForTest(factory);
   V4Database::Create(task_runner_, database_dirname_, list_info_map_,
                      callback_db_ready);
   created_but_not_called_back_ = true;
   task_runner_->RunPendingTasks();
 
-  base::MessageLoop::current()->RunUntilIdle();
+  base::RunLoop().RunUntilIdle();
   EXPECT_EQ(true, created_and_called_back_);
 }
 
 // Test to set up the database with fake stores that fail to reset.
 TEST_F(SafeBrowsingV4DatabaseTest, TestSetupDatabaseWithFakeStoresFailsReset) {
   SetupInfoMapAndExpectedState();
 
   NewDatabaseReadyCallback callback_db_ready = base::Bind(
       &SafeBrowsingV4DatabaseTest::NewDatabaseReadyWithExpectedStorePathsAndIds,
       base::Unretained(this), expected_store_paths_, expected_identifiers_,
       false);
 
   FakeV4StoreFactory* factory = new FakeV4StoreFactory(true);
   ANNOTATE_LEAKING_OBJECT_PTR(factory);
   V4Database::RegisterStoreFactoryForTest(factory);
   V4Database::Create(task_runner_, database_dirname_, list_info_map_,
                      callback_db_ready);
   created_but_not_called_back_ = true;
   task_runner_->RunPendingTasks();
 
-  base::MessageLoop::current()->RunUntilIdle();
+  base::RunLoop().RunUntilIdle();
   EXPECT_EQ(true, created_and_called_back_);
 }
 
 }  // namespace safe_browsing