TaskScheduler [11] Support ExecutionMode::SINGLE_THREADED.

base/task_scheduler/scheduler_worker_thread_unittest.cc

 // Copyright 2016 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/task_scheduler/scheduler_worker_thread.h"
 
 #include <stddef.h>
 
 #include <memory>
 #include <vector>
@@ skipping 13 matching lines @@
 namespace internal {
 namespace {
 
 const size_t kNumSequencesPerTest = 150;
 
 // The test parameter is the number of Tasks per Sequence returned by GetWork().
 class TaskSchedulerWorkerThreadTest : public testing::TestWithParam<size_t> {
  protected:
   TaskSchedulerWorkerThreadTest()
       : main_entry_called_(true, false),
-        num_get_work_cv_(lock_.CreateConditionVariable()) {}
+        num_get_work_cv_(lock_.CreateConditionVariable()),
+        worker_thread_set_(true, false) {}
 
   void SetUp() override {
     worker_thread_ = SchedulerWorkerThread::Create(
         ThreadPriority::NORMAL,
         WrapUnique(new TestSchedulerWorkerThreadDelegate(this)),
         &task_tracker_);
     ASSERT_TRUE(worker_thread_);
+    worker_thread_set_.Signal();
     main_entry_called_.Wait();
   }
 
   void TearDown() override {
     worker_thread_->JoinForTesting();
   }
 
   size_t TasksPerSequence() const { return GetParam(); }
 
   // Wait until GetWork() has been called |num_get_work| times.
@@ skipping 32 matching lines @@
   std::unique_ptr<SchedulerWorkerThread> worker_thread_;
 
  private:
   class TestSchedulerWorkerThreadDelegate
       : public SchedulerWorkerThread::Delegate {
    public:
     TestSchedulerWorkerThreadDelegate(TaskSchedulerWorkerThreadTest* outer)
         : outer_(outer) {}
 
     // SchedulerWorkerThread::Delegate:
-    void OnMainEntry() override {
+    void OnMainEntry(SchedulerWorkerThread* worker_thread) override {
+      outer_->worker_thread_set_.Wait();
+      EXPECT_EQ(outer_->worker_thread_.get(), worker_thread);
+
       // Without synchronization, OnMainEntry() could be called twice without
       // generating an error.
       AutoSchedulerLock auto_lock(outer_->lock_);
       EXPECT_FALSE(outer_->main_entry_called_.IsSignaled());
       outer_->main_entry_called_.Signal();
     }
 
     scoped_refptr<Sequence> GetWork(
         SchedulerWorkerThread* worker_thread) override {
       EXPECT_EQ(outer_->worker_thread_.get(), worker_thread);
@@ skipping 88 matching lines @@
 
   // Sequences created by GetWork().
   std::vector<scoped_refptr<Sequence>> created_sequences_;
 
   // Sequences passed to EnqueueSequence().
   std::vector<scoped_refptr<Sequence>> re_enqueued_sequences_;
 
   // Number of times that RunTaskCallback() has been called.
   size_t num_run_tasks_ = 0;
 
+  // Signaled after |worker_thread_| is set.
+  WaitableEvent worker_thread_set_;
+
   DISALLOW_COPY_AND_ASSIGN(TaskSchedulerWorkerThreadTest);
 };
 
 // Verify that when GetWork() continuously returns Sequences, all Tasks in these
 // Sequences run successfully. The test wakes up the SchedulerWorkerThread once.
 TEST_P(TaskSchedulerWorkerThreadTest, ContinuousWork) {
   // Set GetWork() to return |kNumSequencesPerTest| Sequences before starting to
   // return nullptr.
   SetNumSequencesToCreate(kNumSequencesPerTest);
 
@@ skipping 54 matching lines @@
 INSTANTIATE_TEST_CASE_P(OneTaskPerSequence,
                         TaskSchedulerWorkerThreadTest,
                         ::testing::Values(1));
 INSTANTIATE_TEST_CASE_P(TwoTasksPerSequence,
                         TaskSchedulerWorkerThreadTest,
                         ::testing::Values(2));
 
 }  // namespace
 }  // namespace internal
 }  // namespace base