Index: base/task_runner_test_template.cc |
diff --git a/base/task_runner_test_template.cc b/base/task_runner_test_template.cc |
index 064eac32c1f118e9bd111afbf7484d0dea460fcb..03729ee355b577f0e79185bec12f14205d126d3f 100644 |
--- a/base/task_runner_test_template.cc |
+++ b/base/task_runner_test_template.cc |
@@ -6,12 +6,21 @@ |
namespace base { |
+namespace internal { |
+ |
TaskTracker::TaskTracker() {} |
TaskTracker::~TaskTracker() {} |
-void TaskTracker::RunTask(int i) { |
+Closure TaskTracker::WrapTask(const Closure& task, int i) { |
+ return Bind(&TaskTracker::RunTask, this, task, i); |
+} |
+ |
+void TaskTracker::RunTask(const Closure& task, int i) { |
AutoLock lock(task_run_counts_lock_); |
+ if (!task.is_null()) { |
+ task.Run(); |
+ } |
++task_run_counts_[i]; |
} |
@@ -20,4 +29,12 @@ std::map<int, int> TaskTracker::GetTaskRunCounts() const { |
return task_run_counts_; |
} |
+void ExpectRunsTasksOnCurrentThread( |
+ bool expected_value, |
+ const scoped_refptr<TaskRunner>& task_runner) { |
+ EXPECT_EQ(expected_value, task_runner->RunsTasksOnCurrentThread()); |
+} |
+ |
+} // namespace internal |
+ |
} // namespace base |