Introduce ThreadTaskRunnerHandle::OverrideForTesting and TestMockTimeTaskRunner::ScopedContext.

components/sync/driver/sync_stopped_reporter_unittest.cc

 // Copyright 2015 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 "components/sync/driver/sync_stopped_reporter.h"
 
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
-#include "base/test/test_simple_task_runner.h"
+#include "base/test/scoped_mock_time_message_loop_task_runner.h"
 #include "base/threading/non_thread_safe.h"
 #include "components/sync/protocol/sync.pb.h"
 #include "net/http/http_status_code.h"
 #include "net/url_request/test_url_fetcher_factory.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace syncer {
 
 const char kTestURL[] = "http://chromium.org/test";
@@ skipping 125 matching lines @@
   {
     SyncStoppedReporter ssr(test_url(), user_agent(), request_context(),
                             callback());
     ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday);
     EXPECT_NE(nullptr, factory.GetFetcherByID(0));
   }
   EXPECT_EQ(nullptr, factory.GetFetcherByID(0));
 }
 
 TEST_F(SyncStoppedReporterTest, Timeout) {
+  // Mock the underlying loop's clock to trigger the timer at will.
+  base::ScopedMockTimeMessageLoopTaskRunner mock_main_runner;
+
   SyncStoppedReporter ssr(test_url(), user_agent(), request_context(),
                           callback());
 
-  // A task runner that can trigger the timeout immediately.
-  scoped_refptr<base::TestSimpleTaskRunner> task_runner(
-      new base::TestSimpleTaskRunner());
-  ssr.SetTimerTaskRunnerForTest(task_runner);
-
   // Begin request.
   ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday);
 
   // Trigger the timeout.
-  ASSERT_TRUE(task_runner->HasPendingTask());
-  task_runner->RunPendingTasks();
-
-  base::RunLoop run_loop;
-  run_loop.RunUntilIdle();
+  ASSERT_TRUE(mock_main_runner->HasPendingTask());
+  mock_main_runner->FastForwardUntilNoTasksRemain();
   EXPECT_EQ(SyncStoppedReporter::RESULT_TIMEOUT, request_result());
 }
 
 TEST_F(SyncStoppedReporterTest, NoCallback) {
   net::TestURLFetcherFactory factory;
   SyncStoppedReporter ssr(GURL(kTestURL), user_agent(), request_context(),
                           SyncStoppedReporter::ResultCallback());
   ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday);
   net::TestURLFetcher* fetcher = factory.GetFetcherByID(0);
   fetcher->set_response_code(net::HTTP_OK);
   ssr.OnURLFetchComplete(fetcher);
 }
 
 TEST_F(SyncStoppedReporterTest, NoCallbackTimeout) {
+  // Mock the underlying loop's clock to trigger the timer at will.
+  base::ScopedMockTimeMessageLoopTaskRunner mock_main_runner;
+
   SyncStoppedReporter ssr(GURL(kTestURL), user_agent(), request_context(),
                           SyncStoppedReporter::ResultCallback());
 
-  // A task runner that can trigger the timeout immediately.
-  scoped_refptr<base::TestSimpleTaskRunner> task_runner(
-      new base::TestSimpleTaskRunner());
-  ssr.SetTimerTaskRunnerForTest(task_runner);
-
   // Begin request.
   ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday);
 
   // Trigger the timeout.
-  ASSERT_TRUE(task_runner->HasPendingTask());
-  task_runner->RunPendingTasks();
+  ASSERT_TRUE(mock_main_runner->HasPendingTask());
+  mock_main_runner->FastForwardUntilNoTasksRemain();
 }
 
 }  // namespace syncer