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 706bb2b61283742d354be3003e50f4e37c3b7a80..dc9c15772c4b4ebd7dd04e356e6e2fb765c82686 100644 |
--- a/base/task_scheduler/delayed_task_manager_unittest.cc |
+++ b/base/task_scheduler/delayed_task_manager_unittest.cc |
@@ -59,17 +59,22 @@ class MockSchedulerThreadPool : public SchedulerThreadPool { |
} |
bool PostTaskWithSequence(std::unique_ptr<Task> task, |
- scoped_refptr<Sequence> sequence) override { |
+ scoped_refptr<Sequence> sequence, |
+ SchedulerWorkerThread* worker_thread) override { |
NOTREACHED(); |
return true; |
} |
void PostTaskWithSequenceNow(std::unique_ptr<Task> task, |
- scoped_refptr<Sequence> sequence) override { |
- PostTaskWithSequenceNowMock(task.get(), sequence.get()); |
+ scoped_refptr<Sequence> sequence, |
+ SchedulerWorkerThread* worker_thread) override { |
+ PostTaskWithSequenceNowMock(task.get(), sequence.get(), worker_thread); |
} |
- MOCK_METHOD2(PostTaskWithSequenceNowMock, void(const Task*, const Sequence*)); |
+ MOCK_METHOD3(PostTaskWithSequenceNowMock, |
+ void(const Task*, |
+ const Sequence*, |
+ const SchedulerWorkerThread* worker_thread)); |
}; |
} // namespace |
@@ -95,7 +100,7 @@ TEST(TaskSchedulerDelayedTaskManagerTest, PostReadyTaskBeforeDelayedRunTime) { |
// Add |task| to the DelayedTaskManager. |
EXPECT_CALL(manager, OnDelayedRunTimeUpdated()); |
- manager.AddDelayedTask(std::move(task), sequence, &thread_pool); |
+ manager.AddDelayedTask(std::move(task), sequence, nullptr, &thread_pool); |
testing::Mock::VerifyAndClear(&manager); |
EXPECT_EQ(task_raw->delayed_run_time, manager.GetDelayedRunTime()); |
@@ -121,7 +126,7 @@ TEST(TaskSchedulerDelayedTaskManagerTest, PostReadyTasksAtDelayedRunTime) { |
// Add |task| to the DelayedTaskManager. |
EXPECT_CALL(manager, OnDelayedRunTimeUpdated()); |
- manager.AddDelayedTask(std::move(task), sequence, &thread_pool); |
+ manager.AddDelayedTask(std::move(task), sequence, nullptr, &thread_pool); |
testing::Mock::VerifyAndClear(&manager); |
EXPECT_EQ(task_raw->delayed_run_time, manager.GetDelayedRunTime()); |
@@ -130,7 +135,7 @@ TEST(TaskSchedulerDelayedTaskManagerTest, PostReadyTasksAtDelayedRunTime) { |
// Ask the DelayedTaskManager to post tasks that are ripe for execution. |
EXPECT_CALL(thread_pool, |
- PostTaskWithSequenceNowMock(task_raw, sequence.get())); |
+ PostTaskWithSequenceNowMock(task_raw, sequence.get(), nullptr)); |
manager.PostReadyTasks(); |
testing::Mock::VerifyAndClear(&manager); |
EXPECT_EQ(TimeTicks(), manager.GetDelayedRunTime()); |
@@ -150,7 +155,7 @@ TEST(TaskSchedulerDelayedTaskManagerTest, PostReadyTasksAfterDelayedRunTime) { |
// Add |task| to the DelayedTaskManager. |
EXPECT_CALL(manager, OnDelayedRunTimeUpdated()); |
- manager.AddDelayedTask(std::move(task), sequence, &thread_pool); |
+ manager.AddDelayedTask(std::move(task), sequence, nullptr, &thread_pool); |
testing::Mock::VerifyAndClear(&manager); |
EXPECT_EQ(task_raw->delayed_run_time, manager.GetDelayedRunTime()); |
@@ -160,7 +165,7 @@ TEST(TaskSchedulerDelayedTaskManagerTest, PostReadyTasksAfterDelayedRunTime) { |
// Ask the DelayedTaskManager to post tasks that are ripe for execution. |
EXPECT_CALL(thread_pool, |
- PostTaskWithSequenceNowMock(task_raw, sequence.get())); |
+ PostTaskWithSequenceNowMock(task_raw, sequence.get(), nullptr)); |
manager.PostReadyTasks(); |
testing::Mock::VerifyAndClear(&manager); |
EXPECT_EQ(TimeTicks(), manager.GetDelayedRunTime()); |
@@ -192,20 +197,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, &thread_pool); |
+ manager.AddDelayedTask(std::move(task_a), sequence, nullptr, &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, &thread_pool); |
+ manager.AddDelayedTask(std::move(task_b), sequence, nullptr, &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, &thread_pool); |
+ manager.AddDelayedTask(std::move(task_c), sequence, nullptr, &thread_pool); |
testing::Mock::VerifyAndClear(&manager); |
EXPECT_EQ(task_c_raw->delayed_run_time, manager.GetDelayedRunTime()); |
@@ -216,7 +221,7 @@ TEST(TaskSchedulerDelayedTaskManagerTest, AddAndPostReadyTasks) { |
// |task_c_raw| should be posted and the delayed run time should become |
// |task_a_raw|'s delayed run time. |
EXPECT_CALL(thread_pool, |
- PostTaskWithSequenceNowMock(task_c_raw, sequence.get())); |
+ PostTaskWithSequenceNowMock(task_c_raw, sequence.get(), nullptr)); |
manager.PostReadyTasks(); |
testing::Mock::VerifyAndClear(&thread_pool); |
EXPECT_EQ(task_a_raw->delayed_run_time, manager.GetDelayedRunTime()); |
@@ -228,9 +233,9 @@ TEST(TaskSchedulerDelayedTaskManagerTest, AddAndPostReadyTasks) { |
// |task_a_raw| and |task_b_raw| should be posted and the delayed run time |
// should become a null TimeTicks. |
EXPECT_CALL(thread_pool, |
- PostTaskWithSequenceNowMock(task_a_raw, sequence.get())); |
+ PostTaskWithSequenceNowMock(task_a_raw, sequence.get(), nullptr)); |
EXPECT_CALL(thread_pool, |
- PostTaskWithSequenceNowMock(task_b_raw, sequence.get())); |
+ PostTaskWithSequenceNowMock(task_b_raw, sequence.get(), nullptr)); |
manager.PostReadyTasks(); |
testing::Mock::VerifyAndClear(&thread_pool); |
EXPECT_EQ(TimeTicks(), manager.GetDelayedRunTime()); |