base/test/task_runner_test_template.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/test/task_runner_test_template.h"
   6
   7 namespace base {
   8
   9 namespace internal {
  10
  11 TaskTracker::TaskTracker() : task_runs_(0), task_runs_cv_(&lock_) {}
  12
  13 TaskTracker::~TaskTracker() {}
  14
  15 Closure TaskTracker::WrapTask(const Closure& task, int i) {
  16   return Bind(&TaskTracker::RunTask, this, task, i);
  17 }
  18
  19 void TaskTracker::RunTask(const Closure& task, int i) {
  20   AutoLock lock(lock_);
  21   if (!task.is_null()) {
  22     task.Run();
  23   }
  24   ++task_run_counts_[i];
  25   ++task_runs_;
  26   task_runs_cv_.Signal();
  27 }
  28
  29 std::map<int, int> TaskTracker::GetTaskRunCounts() const {
  30   AutoLock lock(lock_);
  31   return task_run_counts_;
  32 }
  33
  34 void TaskTracker::WaitForCompletedTasks(int count) {
  35   AutoLock lock(lock_);
  36   while (task_runs_ < count)
  37     task_runs_cv_.Wait();
  38 }
  39
  40 void ExpectRunsTasksOnCurrentThread(
  41     bool expected_value,
  42     const scoped_refptr<TaskRunner>& task_runner) {
  43   EXPECT_EQ(expected_value, task_runner->RunsTasksOnCurrentThread());
  44 }
  45
  46 }  // namespace internal
  47
  48 }  // namespace base