Make TestSimpleTaskRunner thread-safe.

base/test/test_simple_task_runner.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/test/test_simple_task_runner.h"
 
 #include "base/logging.h"
 
 namespace base {
 
-TestSimpleTaskRunner::TestSimpleTaskRunner() {}
+TestSimpleTaskRunner::TestSimpleTaskRunner() = default;
 
-TestSimpleTaskRunner::~TestSimpleTaskRunner() {
-  DCHECK(thread_checker_.CalledOnValidThread());
-}
+TestSimpleTaskRunner::~TestSimpleTaskRunner() = default;
 
 bool TestSimpleTaskRunner::PostDelayedTask(
     const tracked_objects::Location& from_here,
     const Closure& task,
     TimeDelta delay) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  AutoLock auto_lock(lock_);
   pending_tasks_.push_back(
       TestPendingTask(from_here, task, TimeTicks(), delay,
                       TestPendingTask::NESTABLE));
   return true;
 }
 
 bool TestSimpleTaskRunner::PostNonNestableDelayedTask(
     const tracked_objects::Location& from_here,
     const Closure& task,
     TimeDelta delay) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  AutoLock auto_lock(lock_);
   pending_tasks_.push_back(
       TestPendingTask(from_here, task, TimeTicks(), delay,
                       TestPendingTask::NON_NESTABLE));
   return true;
 }
 
 bool TestSimpleTaskRunner::RunsTasksOnCurrentThread() const {
-  DCHECK(thread_checker_.CalledOnValidThread());
-  return true;
+  return thread_ref_ == PlatformThread::CurrentRef();
 }
 
-const std::deque<TestPendingTask>&
-TestSimpleTaskRunner::GetPendingTasks() const {
-  DCHECK(thread_checker_.CalledOnValidThread());
+std::deque<TestPendingTask> TestSimpleTaskRunner::GetPendingTasks() const {
+  AutoLock auto_lock(lock_);
   return pending_tasks_;
 }
 
+size_t TestSimpleTaskRunner::NumPendingTasks() const {
+  AutoLock auto_lock(lock_);
+  return pending_tasks_.size();
+}
+
 bool TestSimpleTaskRunner::HasPendingTask() const {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  AutoLock auto_lock(lock_);
   return !pending_tasks_.empty();
 }
 
 base::TimeDelta TestSimpleTaskRunner::NextPendingTaskDelay() const {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  AutoLock auto_lock(lock_);
   return pending_tasks_.front().GetTimeToRun() - base::TimeTicks();
 }
 
 void TestSimpleTaskRunner::ClearPendingTasks() {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  AutoLock auto_lock(lock_);
   pending_tasks_.clear();
 }
 
 void TestSimpleTaskRunner::RunPendingTasks() {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(RunsTasksOnCurrentThread());
+
   // Swap with a local variable to avoid re-entrancy problems.
   std::deque<TestPendingTask> tasks_to_run;
-  tasks_to_run.swap(pending_tasks_);
-  for (std::deque<TestPendingTask>::iterator it = tasks_to_run.begin();
-       it != tasks_to_run.end(); ++it) {
-    it->task.Run();
+  {
+    AutoLock auto_lock(lock_);
+    tasks_to_run.swap(pending_tasks_);
   }
+
+  for (const auto& task : tasks_to_run)
+    task.task.Run();
 }
 
 void TestSimpleTaskRunner::RunUntilIdle() {
   while (!pending_tasks_.empty()) {
     RunPendingTasks();
   }
 }
 
 }  // namespace base