Flush SequenceWorkerPool tasks after each unit test.

base/test/sequenced_task_runner_test_template.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/sequenced_task_runner_test_template.h"
 
 #include <ostream>
 
 #include "base/location.h"
 
 namespace base {
 
 namespace internal {
 
 TaskEvent::TaskEvent(int i, Type type)
   : i(i), type(type) {
 }
 
-SequencedTaskTracker::SequencedTaskTracker() : next_post_i_(0) {
+SequencedTaskTracker::SequencedTaskTracker()
+    : next_post_i_(0),
+      task_end_count_(0),
+      task_end_cv_(&lock_) {
 }
 
 void SequencedTaskTracker::PostWrappedNonNestableTask(
     const scoped_refptr<SequencedTaskRunner>& task_runner,
     const Closure& task) {
   AutoLock event_lock(lock_);
   const int post_i = next_post_i_++;
   Closure wrapped_task = Bind(&SequencedTaskTracker::RunTask, this,
                               task, post_i);
   task_runner->PostNonNestableTask(FROM_HERE, wrapped_task);
@@ skipping 44 matching lines @@
 }
 
 void SequencedTaskTracker::TaskStarted(int i) {
   AutoLock lock(lock_);
   events_.push_back(TaskEvent(i, TaskEvent::START));
 }
 
 void SequencedTaskTracker::TaskEnded(int i) {
   AutoLock lock(lock_);
   events_.push_back(TaskEvent(i, TaskEvent::END));
+  ++task_end_count_;
+  task_end_cv_.Signal();
 }
 
 const std::vector<TaskEvent>&
 SequencedTaskTracker::GetTaskEvents() const {
   return events_;
 }
 
+void SequencedTaskTracker::WaitForCompletedTasks(int count) {
+  AutoLock lock(lock_);
+  while (task_end_count_ < count)
+    task_end_cv_.Wait();
+}
+
 SequencedTaskTracker::~SequencedTaskTracker() {
 }
 
 void PrintTo(const TaskEvent& event, std::ostream* os) {
   *os << "(i=" << event.i << ", type=";
   switch (event.type) {
     case TaskEvent::POST: *os << "POST"; break;
     case TaskEvent::START: *os << "START"; break;
     case TaskEvent::END: *os << "END"; break;
   }
@@ skipping 149 matching lines @@
       CheckEventOrdersForEachTask(events, task_count);
   if (!result)
     return result;
 
   return CheckNoTaskRunsOverlap(events);
 }
 
 }  // namespace internal
 
 }  // namespace base