Index: base/task_scheduler/scheduler_worker_thread_unittest.cc |
diff --git a/base/task_scheduler/scheduler_worker_thread_unittest.cc b/base/task_scheduler/scheduler_worker_thread_unittest.cc |
deleted file mode 100644 |
index 88d9b23c8f2a5199d45245cd67a10df4752a12e5..0000000000000000000000000000000000000000 |
--- a/base/task_scheduler/scheduler_worker_thread_unittest.cc |
+++ /dev/null |
@@ -1,291 +0,0 @@ |
-// 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> |
- |
-#include "base/bind.h" |
-#include "base/bind_helpers.h" |
-#include "base/macros.h" |
-#include "base/memory/ptr_util.h" |
-#include "base/synchronization/condition_variable.h" |
-#include "base/task_scheduler/scheduler_lock.h" |
-#include "base/task_scheduler/sequence.h" |
-#include "base/task_scheduler/task.h" |
-#include "base/task_scheduler/task_tracker.h" |
-#include "base/time/time.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-namespace base { |
-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_(WaitableEvent::ResetPolicy::MANUAL, |
- WaitableEvent::InitialState::NOT_SIGNALED), |
- num_get_work_cv_(lock_.CreateConditionVariable()), |
- worker_thread_set_(WaitableEvent::ResetPolicy::MANUAL, |
- WaitableEvent::InitialState::NOT_SIGNALED) {} |
- |
- 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. |
- void WaitForNumGetWork(size_t num_get_work) { |
- AutoSchedulerLock auto_lock(lock_); |
- while (num_get_work_ < num_get_work) |
- num_get_work_cv_->Wait(); |
- } |
- |
- void SetMaxGetWork(size_t max_get_work) { |
- AutoSchedulerLock auto_lock(lock_); |
- max_get_work_ = max_get_work; |
- } |
- |
- void SetNumSequencesToCreate(size_t num_sequences_to_create) { |
- AutoSchedulerLock auto_lock(lock_); |
- EXPECT_EQ(0U, num_sequences_to_create_); |
- num_sequences_to_create_ = num_sequences_to_create; |
- } |
- |
- size_t NumRunTasks() { |
- AutoSchedulerLock auto_lock(lock_); |
- return num_run_tasks_; |
- } |
- |
- std::vector<scoped_refptr<Sequence>> CreatedSequences() { |
- AutoSchedulerLock auto_lock(lock_); |
- return created_sequences_; |
- } |
- |
- std::vector<scoped_refptr<Sequence>> EnqueuedSequences() { |
- AutoSchedulerLock auto_lock(lock_); |
- return re_enqueued_sequences_; |
- } |
- |
- std::unique_ptr<SchedulerWorkerThread> worker_thread_; |
- |
- private: |
- class TestSchedulerWorkerThreadDelegate |
- : public SchedulerWorkerThread::Delegate { |
- public: |
- TestSchedulerWorkerThreadDelegate(TaskSchedulerWorkerThreadTest* outer) |
- : outer_(outer) {} |
- |
- // SchedulerWorkerThread::Delegate: |
- 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); |
- |
- { |
- AutoSchedulerLock auto_lock(outer_->lock_); |
- |
- // Increment the number of times that this method has been called. |
- ++outer_->num_get_work_; |
- outer_->num_get_work_cv_->Signal(); |
- |
- // Verify that this method isn't called more times than expected. |
- EXPECT_LE(outer_->num_get_work_, outer_->max_get_work_); |
- |
- // Check if a Sequence should be returned. |
- if (outer_->num_sequences_to_create_ == 0) |
- return nullptr; |
- --outer_->num_sequences_to_create_; |
- } |
- |
- // Create a Sequence with TasksPerSequence() Tasks. |
- scoped_refptr<Sequence> sequence(new Sequence); |
- for (size_t i = 0; i < outer_->TasksPerSequence(); ++i) { |
- std::unique_ptr<Task> task(new Task( |
- FROM_HERE, Bind(&TaskSchedulerWorkerThreadTest::RunTaskCallback, |
- Unretained(outer_)), |
- TaskTraits(), TimeDelta())); |
- EXPECT_TRUE(outer_->task_tracker_.WillPostTask(task.get())); |
- sequence->PushTask(std::move(task)); |
- } |
- |
- { |
- // Add the Sequence to the vector of created Sequences. |
- AutoSchedulerLock auto_lock(outer_->lock_); |
- outer_->created_sequences_.push_back(sequence); |
- } |
- |
- return sequence; |
- } |
- |
- // This override verifies that |sequence| contains the expected number of |
- // Tasks and adds it to |enqueued_sequences_|. Unlike a normal |
- // EnqueueSequence implementation, it doesn't reinsert |sequence| into a |
- // queue for further execution. |
- void ReEnqueueSequence(scoped_refptr<Sequence> sequence) override { |
- EXPECT_GT(outer_->TasksPerSequence(), 1U); |
- |
- // Verify that |sequence| contains TasksPerSequence() - 1 Tasks. |
- for (size_t i = 0; i < outer_->TasksPerSequence() - 1; ++i) { |
- EXPECT_TRUE(sequence->PeekTask()); |
- sequence->PopTask(); |
- } |
- EXPECT_FALSE(sequence->PeekTask()); |
- |
- // Add |sequence| to |re_enqueued_sequences_|. |
- AutoSchedulerLock auto_lock(outer_->lock_); |
- outer_->re_enqueued_sequences_.push_back(std::move(sequence)); |
- EXPECT_LE(outer_->re_enqueued_sequences_.size(), |
- outer_->created_sequences_.size()); |
- } |
- |
- TimeDelta GetSleepTimeout() override { |
- return TimeDelta::Max(); |
- } |
- |
- private: |
- TaskSchedulerWorkerThreadTest* outer_; |
- }; |
- |
- void RunTaskCallback() { |
- AutoSchedulerLock auto_lock(lock_); |
- ++num_run_tasks_; |
- EXPECT_LE(num_run_tasks_, created_sequences_.size()); |
- } |
- |
- TaskTracker task_tracker_; |
- |
- // Synchronizes access to all members below. |
- mutable SchedulerLock lock_; |
- |
- // Signaled once OnMainEntry() has been called. |
- WaitableEvent main_entry_called_; |
- |
- // Number of Sequences that should be created by GetWork(). When this |
- // is 0, GetWork() returns nullptr. |
- size_t num_sequences_to_create_ = 0; |
- |
- // Number of times that GetWork() has been called. |
- size_t num_get_work_ = 0; |
- |
- // Maximum number of times that GetWork() can be called. |
- size_t max_get_work_ = 0; |
- |
- // Condition variable signaled when |num_get_work_| is incremented. |
- std::unique_ptr<ConditionVariable> num_get_work_cv_; |
- |
- // 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); |
- |
- // Expect |kNumSequencesPerTest| calls to GetWork() in which it returns a |
- // Sequence and one call in which its returns nullptr. |
- const size_t kExpectedNumGetWork = kNumSequencesPerTest + 1; |
- SetMaxGetWork(kExpectedNumGetWork); |
- |
- // Wake up |worker_thread_| and wait until GetWork() has been invoked the |
- // expected amount of times. |
- worker_thread_->WakeUp(); |
- WaitForNumGetWork(kExpectedNumGetWork); |
- |
- // All tasks should have run. |
- EXPECT_EQ(kNumSequencesPerTest, NumRunTasks()); |
- |
- // If Sequences returned by GetWork() contain more than one Task, they aren't |
- // empty after the worker thread pops Tasks from them and thus should be |
- // returned to EnqueueSequence(). |
- if (TasksPerSequence() > 1) |
- EXPECT_EQ(CreatedSequences(), EnqueuedSequences()); |
- else |
- EXPECT_TRUE(EnqueuedSequences().empty()); |
-} |
- |
-// Verify that when GetWork() alternates between returning a Sequence and |
-// returning nullptr, all Tasks in the returned Sequences run successfully. The |
-// test wakes up the SchedulerWorkerThread once for each Sequence. |
-TEST_P(TaskSchedulerWorkerThreadTest, IntermittentWork) { |
- for (size_t i = 0; i < kNumSequencesPerTest; ++i) { |
- // Set GetWork() to return 1 Sequence before starting to return |
- // nullptr. |
- SetNumSequencesToCreate(1); |
- |
- // Expect |i + 1| calls to GetWork() in which it returns a Sequence and |
- // |i + 1| calls in which it returns nullptr. |
- const size_t expected_num_get_work = 2 * (i + 1); |
- SetMaxGetWork(expected_num_get_work); |
- |
- // Wake up |worker_thread_| and wait until GetWork() has been invoked |
- // the expected amount of times. |
- worker_thread_->WakeUp(); |
- WaitForNumGetWork(expected_num_get_work); |
- |
- // The Task should have run |
- EXPECT_EQ(i + 1, NumRunTasks()); |
- |
- // If Sequences returned by GetWork() contain more than one Task, they |
- // aren't empty after the worker thread pops Tasks from them and thus should |
- // be returned to EnqueueSequence(). |
- if (TasksPerSequence() > 1) |
- EXPECT_EQ(CreatedSequences(), EnqueuedSequences()); |
- else |
- EXPECT_TRUE(EnqueuedSequences().empty()); |
- } |
-} |
- |
-INSTANTIATE_TEST_CASE_P(OneTaskPerSequence, |
- TaskSchedulerWorkerThreadTest, |
- ::testing::Values(1)); |
-INSTANTIATE_TEST_CASE_P(TwoTasksPerSequence, |
- TaskSchedulerWorkerThreadTest, |
- ::testing::Values(2)); |
- |
-} // namespace |
-} // namespace internal |
-} // namespace base |