| Index: base/task_scheduler/delayed_task_manager_unittest.cc
|
| diff --git a/base/task_scheduler/delayed_task_manager_unittest.cc b/base/task_scheduler/delayed_task_manager_unittest.cc
|
| index 9f3fa7577fcdde9da3d8e9caf1235288b0021fa7..bb5ec565a48a4b05be47da2d53090b0645a08144 100644
|
| --- a/base/task_scheduler/delayed_task_manager_unittest.cc
|
| +++ b/base/task_scheduler/delayed_task_manager_unittest.cc
|
| @@ -11,7 +11,7 @@
|
| #include "base/bind_helpers.h"
|
| #include "base/memory/ref_counted.h"
|
| #include "base/task_scheduler/scheduler_lock.h"
|
| -#include "base/task_scheduler/scheduler_task_executor.h"
|
| +#include "base/task_scheduler/scheduler_thread_pool_interface.h"
|
| #include "base/task_scheduler/sequence.h"
|
| #include "base/task_scheduler/task.h"
|
| #include "base/time/time.h"
|
| @@ -42,14 +42,14 @@ class TestDelayedTaskManager : public DelayedTaskManager {
|
| DISALLOW_COPY_AND_ASSIGN(TestDelayedTaskManager);
|
| };
|
|
|
| -class MockSchedulerTaskExecutor : public SchedulerTaskExecutor {
|
| +class MockSchedulerThreadPool : public SchedulerThreadPoolInterface {
|
| public:
|
| - void PostTaskWithSequence(std::unique_ptr<Task> task,
|
| - scoped_refptr<Sequence> sequence) override {
|
| - PostTaskWithSequenceMock(task.get(), sequence.get());
|
| + void PostTaskWithSequenceNow(std::unique_ptr<Task> task,
|
| + scoped_refptr<Sequence> sequence) override {
|
| + PostTaskWithSequenceNowMock(task.get(), sequence.get());
|
| }
|
|
|
| - MOCK_METHOD2(PostTaskWithSequenceMock, void(const Task*, const Sequence*));
|
| + MOCK_METHOD2(PostTaskWithSequenceNowMock, void(const Task*, const Sequence*));
|
| };
|
|
|
| // Verify that GetDelayedRunTime() returns a null TimeTicks when there are
|
| @@ -69,16 +69,16 @@ TEST(TaskSchedulerDelayedTaskManagerTest, PostReadyTaskBeforeDelayedRunTime) {
|
| manager.Now() + TimeDelta::FromSeconds(1)));
|
| const Task* task_raw = task.get();
|
| scoped_refptr<Sequence> sequence(new Sequence);
|
| - testing::StrictMock<MockSchedulerTaskExecutor> executor;
|
| + testing::StrictMock<MockSchedulerThreadPool> thread_pool;
|
|
|
| // Add |task| to the DelayedTaskManager.
|
| EXPECT_CALL(manager, OnDelayedRunTimeUpdated());
|
| - manager.AddDelayedTask(std::move(task), sequence, &executor);
|
| + manager.AddDelayedTask(std::move(task), sequence, &thread_pool);
|
| testing::Mock::VerifyAndClear(&manager);
|
| EXPECT_EQ(task_raw->delayed_run_time, manager.GetDelayedRunTime());
|
|
|
| // Ask the DelayedTaskManager to post tasks that are ripe for execution. Don't
|
| - // expect any call to the mock method of |executor|.
|
| + // expect any call to the mock method of |thread_pool|.
|
| manager.PostReadyTasks();
|
|
|
| // The delayed run time shouldn't have changed.
|
| @@ -95,11 +95,11 @@ TEST(TaskSchedulerDelayedTaskManagerTest, PostReadyTasksAtDelayedRunTime) {
|
| manager.Now() + TimeDelta::FromSeconds(1)));
|
| const Task* task_raw = task.get();
|
| scoped_refptr<Sequence> sequence(new Sequence);
|
| - testing::StrictMock<MockSchedulerTaskExecutor> executor;
|
| + testing::StrictMock<MockSchedulerThreadPool> thread_pool;
|
|
|
| // Add |task| to the DelayedTaskManager.
|
| EXPECT_CALL(manager, OnDelayedRunTimeUpdated());
|
| - manager.AddDelayedTask(std::move(task), sequence, &executor);
|
| + manager.AddDelayedTask(std::move(task), sequence, &thread_pool);
|
| testing::Mock::VerifyAndClear(&manager);
|
| EXPECT_EQ(task_raw->delayed_run_time, manager.GetDelayedRunTime());
|
|
|
| @@ -107,7 +107,8 @@ TEST(TaskSchedulerDelayedTaskManagerTest, PostReadyTasksAtDelayedRunTime) {
|
| manager.SetCurrentTime(task_raw->delayed_run_time);
|
|
|
| // Ask the DelayedTaskManager to post tasks that are ripe for execution.
|
| - EXPECT_CALL(executor, PostTaskWithSequenceMock(task_raw, sequence.get()));
|
| + EXPECT_CALL(thread_pool,
|
| + PostTaskWithSequenceNowMock(task_raw, sequence.get()));
|
| manager.PostReadyTasks();
|
| testing::Mock::VerifyAndClear(&manager);
|
| EXPECT_EQ(TimeTicks(), manager.GetDelayedRunTime());
|
| @@ -123,11 +124,11 @@ TEST(TaskSchedulerDelayedTaskManagerTest, PostReadyTasksAfterDelayedRunTime) {
|
| manager.Now() + TimeDelta::FromSeconds(1)));
|
| const Task* task_raw = task.get();
|
| scoped_refptr<Sequence> sequence(new Sequence);
|
| - testing::StrictMock<MockSchedulerTaskExecutor> executor;
|
| + testing::StrictMock<MockSchedulerThreadPool> thread_pool;
|
|
|
| // Add |task| to the DelayedTaskManager.
|
| EXPECT_CALL(manager, OnDelayedRunTimeUpdated());
|
| - manager.AddDelayedTask(std::move(task), sequence, &executor);
|
| + manager.AddDelayedTask(std::move(task), sequence, &thread_pool);
|
| testing::Mock::VerifyAndClear(&manager);
|
| EXPECT_EQ(task_raw->delayed_run_time, manager.GetDelayedRunTime());
|
|
|
| @@ -136,7 +137,8 @@ TEST(TaskSchedulerDelayedTaskManagerTest, PostReadyTasksAfterDelayedRunTime) {
|
| TimeDelta::FromSeconds(10));
|
|
|
| // Ask the DelayedTaskManager to post tasks that are ripe for execution.
|
| - EXPECT_CALL(executor, PostTaskWithSequenceMock(task_raw, sequence.get()));
|
| + EXPECT_CALL(thread_pool,
|
| + PostTaskWithSequenceNowMock(task_raw, sequence.get()));
|
| manager.PostReadyTasks();
|
| testing::Mock::VerifyAndClear(&manager);
|
| EXPECT_EQ(TimeTicks(), manager.GetDelayedRunTime());
|
| @@ -148,7 +150,7 @@ TEST(TaskSchedulerDelayedTaskManagerTest, AddAndPostReadyTasks) {
|
| testing::StrictMock<TestDelayedTaskManager> manager;
|
|
|
| scoped_refptr<Sequence> sequence(new Sequence);
|
| - testing::StrictMock<MockSchedulerTaskExecutor> executor;
|
| + testing::StrictMock<MockSchedulerThreadPool> thread_pool;
|
|
|
| std::unique_ptr<Task> task_a(
|
| new Task(FROM_HERE, Bind(&DoNothing), TaskTraits(),
|
| @@ -168,20 +170,20 @@ TEST(TaskSchedulerDelayedTaskManagerTest, AddAndPostReadyTasks) {
|
| // Add |task_a| to the DelayedTaskManager. The delayed run time should be
|
| // updated to |task_a|'s delayed run time.
|
| EXPECT_CALL(manager, OnDelayedRunTimeUpdated());
|
| - manager.AddDelayedTask(std::move(task_a), sequence, &executor);
|
| + manager.AddDelayedTask(std::move(task_a), sequence, &thread_pool);
|
| testing::Mock::VerifyAndClear(&manager);
|
| EXPECT_EQ(task_a_raw->delayed_run_time, manager.GetDelayedRunTime());
|
|
|
| // Add |task_b| to the DelayedTaskManager. The delayed run time shouldn't
|
| // change.
|
| - manager.AddDelayedTask(std::move(task_b), sequence, &executor);
|
| + manager.AddDelayedTask(std::move(task_b), sequence, &thread_pool);
|
| testing::Mock::VerifyAndClear(&manager);
|
| EXPECT_EQ(task_a_raw->delayed_run_time, manager.GetDelayedRunTime());
|
|
|
| // Add |task_c| to the DelayedTaskManager. The delayed run time should be
|
| // updated to |task_c|'s delayed run time.
|
| EXPECT_CALL(manager, OnDelayedRunTimeUpdated());
|
| - manager.AddDelayedTask(std::move(task_c), sequence, &executor);
|
| + manager.AddDelayedTask(std::move(task_c), sequence, &thread_pool);
|
| testing::Mock::VerifyAndClear(&manager);
|
| EXPECT_EQ(task_c_raw->delayed_run_time, manager.GetDelayedRunTime());
|
|
|
| @@ -191,9 +193,10 @@ TEST(TaskSchedulerDelayedTaskManagerTest, AddAndPostReadyTasks) {
|
| // Ask the DelayedTaskManager to post tasks that are ripe for execution.
|
| // |task_c_raw| should be posted and the delayed run time should become
|
| // |task_a_raw|'s delayed run time.
|
| - EXPECT_CALL(executor, PostTaskWithSequenceMock(task_c_raw, sequence.get()));
|
| + EXPECT_CALL(thread_pool,
|
| + PostTaskWithSequenceNowMock(task_c_raw, sequence.get()));
|
| manager.PostReadyTasks();
|
| - testing::Mock::VerifyAndClear(&executor);
|
| + testing::Mock::VerifyAndClear(&thread_pool);
|
| EXPECT_EQ(task_a_raw->delayed_run_time, manager.GetDelayedRunTime());
|
|
|
| // Fast-forward time to |task_a_raw|'s delayed run time.
|
| @@ -202,10 +205,12 @@ TEST(TaskSchedulerDelayedTaskManagerTest, AddAndPostReadyTasks) {
|
| // Ask the DelayedTaskManager to post tasks that are ripe for execution.
|
| // |task_a_raw| and |task_b_raw| should be posted and the delayed run time
|
| // should become a null TimeTicks.
|
| - EXPECT_CALL(executor, PostTaskWithSequenceMock(task_a_raw, sequence.get()));
|
| - EXPECT_CALL(executor, PostTaskWithSequenceMock(task_b_raw, sequence.get()));
|
| + EXPECT_CALL(thread_pool,
|
| + PostTaskWithSequenceNowMock(task_a_raw, sequence.get()));
|
| + EXPECT_CALL(thread_pool,
|
| + PostTaskWithSequenceNowMock(task_b_raw, sequence.get()));
|
| manager.PostReadyTasks();
|
| - testing::Mock::VerifyAndClear(&executor);
|
| + testing::Mock::VerifyAndClear(&thread_pool);
|
| EXPECT_EQ(TimeTicks(), manager.GetDelayedRunTime());
|
| }
|
|
|
|
|