Index: base/task_scheduler/sequence_unittest.cc |
diff --git a/base/task_scheduler/sequence_unittest.cc b/base/task_scheduler/sequence_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..fc5f32893d5f12c7c823854a96800a8842836d61 |
--- /dev/null |
+++ b/base/task_scheduler/sequence_unittest.cc |
@@ -0,0 +1,141 @@ |
+// 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/sequence.h" |
+ |
+#include "base/task_scheduler/test_util.h" |
+#include "base/time/time.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace base { |
+namespace internal { |
+ |
+class TaskSchedulerSequenceTest : public testing::Test { |
+ public: |
+ TaskSchedulerSequenceTest() |
+ : task_a_(FROM_HERE, |
+ Closure(), |
+ TaskTraits().WithPriority(TaskPriority::BACKGROUND), |
+ TimeTicks::FromInternalValue(1)), |
+ task_b_(FROM_HERE, |
+ Closure(), |
+ TaskTraits().WithPriority(TaskPriority::USER_VISIBLE), |
+ TimeTicks::FromInternalValue(2)), |
+ task_c_(FROM_HERE, |
+ Closure(), |
+ TaskTraits().WithPriority(TaskPriority::USER_BLOCKING), |
+ TimeTicks::FromInternalValue(3)), |
+ task_d_(FROM_HERE, |
+ Closure(), |
+ TaskTraits().WithPriority(TaskPriority::USER_BLOCKING), |
+ TimeTicks::FromInternalValue(4)) {} |
+ |
+ protected: |
+ Task task_a_; |
fdoray
2016/02/18 01:46:11
const Task task_a_;
fdoray
2016/02/18 14:56:13
Done.
|
+ Task task_b_; |
+ Task task_c_; |
+ Task task_d_; |
+}; |
gab
2016/02/18 03:00:46
private:
DISALLOW_COPY_AND_ASSIGN(TaskSchedulerS
fdoray
2016/02/18 14:56:13
Done.
|
+ |
+TEST_F(TaskSchedulerSequenceTest, PushPopPeek) { |
+ scoped_refptr<Sequence> sequence(new Sequence); |
+ size_t prev_num_tasks; |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_a_)), &prev_num_tasks); |
+ EXPECT_EQ(0u, prev_num_tasks); |
+ EXPECT_EQ(task_a_.post_time, sequence->PeekTask()->post_time); |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_b_)), &prev_num_tasks); |
+ EXPECT_EQ(1u, prev_num_tasks); |
+ EXPECT_EQ(task_a_.post_time, sequence->PeekTask()->post_time); |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_c_)), &prev_num_tasks); |
+ EXPECT_EQ(2u, prev_num_tasks); |
+ EXPECT_EQ(task_a_.post_time, sequence->PeekTask()->post_time); |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_d_)), &prev_num_tasks); |
+ EXPECT_EQ(3u, prev_num_tasks); |
+ EXPECT_EQ(task_a_.post_time, sequence->PeekTask()->post_time); |
+ |
+ size_t new_num_tasks; |
+ sequence->PopTask(&new_num_tasks); |
+ EXPECT_EQ(3u, new_num_tasks); |
+ EXPECT_EQ(task_b_.post_time, sequence->PeekTask()->post_time); |
+ |
+ sequence->PopTask(&new_num_tasks); |
+ EXPECT_EQ(2u, new_num_tasks); |
+ EXPECT_EQ(task_c_.post_time, sequence->PeekTask()->post_time); |
+ |
+ sequence->PopTask(&new_num_tasks); |
+ EXPECT_EQ(1u, new_num_tasks); |
+ EXPECT_EQ(task_d_.post_time, sequence->PeekTask()->post_time); |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_a_)), &prev_num_tasks); |
+ EXPECT_EQ(1u, prev_num_tasks); |
+ EXPECT_EQ(task_d_.post_time, sequence->PeekTask()->post_time); |
+ |
+ sequence->PopTask(&new_num_tasks); |
+ EXPECT_EQ(1u, new_num_tasks); |
+ EXPECT_EQ(task_a_.post_time, sequence->PeekTask()->post_time); |
+ |
+ sequence->PopTask(&new_num_tasks); |
+ EXPECT_EQ(0u, new_num_tasks); |
+ EXPECT_EQ(nullptr, sequence->PeekTask()); |
+} |
+ |
+TEST_F(TaskSchedulerSequenceTest, GetSequenceSortKey) { |
fdoray
2016/02/18 01:46:11
*GetSortKey
fdoray
2016/02/18 14:56:13
Done.
|
+ scoped_refptr<Sequence> sequence(new Sequence); |
+ size_t prev_num_tasks; |
+ size_t new_num_tasks; |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_a_)), &prev_num_tasks); |
+ SequenceSortKey sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::BACKGROUND, sort_key.priority_); |
+ EXPECT_EQ(task_a_.post_time, sort_key.next_task_post_time_); |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_b_)), &prev_num_tasks); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::USER_VISIBLE, sort_key.priority_); |
+ EXPECT_EQ(task_a_.post_time, sort_key.next_task_post_time_); |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_c_)), &prev_num_tasks); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::USER_BLOCKING, sort_key.priority_); |
+ EXPECT_EQ(task_a_.post_time, sort_key.next_task_post_time_); |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_d_)), &prev_num_tasks); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::USER_BLOCKING, sort_key.priority_); |
+ EXPECT_EQ(task_a_.post_time, sort_key.next_task_post_time_); |
+ |
+ sequence->PopTask(&new_num_tasks); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::USER_BLOCKING, sort_key.priority_); |
+ EXPECT_EQ(task_b_.post_time, sort_key.next_task_post_time_); |
+ |
+ sequence->PopTask(&new_num_tasks); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::USER_BLOCKING, sort_key.priority_); |
+ EXPECT_EQ(task_c_.post_time, sort_key.next_task_post_time_); |
+ |
+ sequence->PopTask(&new_num_tasks); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::USER_BLOCKING, sort_key.priority_); |
+ EXPECT_EQ(task_d_.post_time, sort_key.next_task_post_time_); |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_a_)), &prev_num_tasks); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::USER_BLOCKING, sort_key.priority_); |
+ EXPECT_EQ(task_d_.post_time, sort_key.next_task_post_time_); |
+ |
+ sequence->PopTask(&new_num_tasks); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::BACKGROUND, sort_key.priority_); |
+ EXPECT_EQ(task_a_.post_time, sort_key.next_task_post_time_); |
+ |
+ sequence->PopTask(&new_num_tasks); |
+} |
+ |
+} // namespace internal |
+} // namespace base |