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..efb6c6366cbdf8343c8044140799797490487e89 |
--- /dev/null |
+++ b/base/task_scheduler/sequence_unittest.cc |
@@ -0,0 +1,128 @@ |
+// 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/macros.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: |
+ const Task task_a_; |
+ const Task task_b_; |
+ const Task task_c_; |
+ const Task task_d_; |
+ |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(TaskSchedulerSequenceTest); |
+}; |
+ |
+TEST_F(TaskSchedulerSequenceTest, PushPopPeek) { |
+ scoped_refptr<Sequence> sequence(new Sequence); |
+ |
+ EXPECT_TRUE(sequence->PushTask(scoped_ptr<Task>(new Task(task_a_)))); |
+ EXPECT_EQ(task_a_.sequenced_time, sequence->PeekTask()->sequenced_time); |
+ |
+ EXPECT_FALSE(sequence->PushTask(scoped_ptr<Task>(new Task(task_b_)))); |
+ EXPECT_EQ(task_a_.sequenced_time, sequence->PeekTask()->sequenced_time); |
+ |
+ EXPECT_FALSE(sequence->PushTask(scoped_ptr<Task>(new Task(task_c_)))); |
+ EXPECT_EQ(task_a_.sequenced_time, sequence->PeekTask()->sequenced_time); |
+ |
+ EXPECT_FALSE(sequence->PushTask(scoped_ptr<Task>(new Task(task_d_)))); |
+ EXPECT_EQ(task_a_.sequenced_time, sequence->PeekTask()->sequenced_time); |
+ |
+ EXPECT_FALSE(sequence->PopTask()); |
+ EXPECT_EQ(task_b_.sequenced_time, sequence->PeekTask()->sequenced_time); |
+ |
+ EXPECT_FALSE(sequence->PopTask()); |
+ EXPECT_EQ(task_c_.sequenced_time, sequence->PeekTask()->sequenced_time); |
+ |
+ EXPECT_FALSE(sequence->PopTask()); |
+ EXPECT_EQ(task_d_.sequenced_time, sequence->PeekTask()->sequenced_time); |
+ |
+ EXPECT_FALSE(sequence->PushTask(scoped_ptr<Task>(new Task(task_a_)))); |
+ EXPECT_EQ(task_d_.sequenced_time, sequence->PeekTask()->sequenced_time); |
+ |
+ EXPECT_FALSE(sequence->PopTask()); |
+ EXPECT_EQ(task_a_.sequenced_time, sequence->PeekTask()->sequenced_time); |
+ |
+ EXPECT_TRUE(sequence->PopTask()); |
+ EXPECT_EQ(nullptr, sequence->PeekTask()); |
+} |
+ |
+TEST_F(TaskSchedulerSequenceTest, GetSortKey) { |
+ scoped_refptr<Sequence> sequence(new Sequence); |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_a_))); |
+ SequenceSortKey sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::BACKGROUND, sort_key.priority_); |
+ EXPECT_EQ(task_a_.sequenced_time, sort_key.next_task_sequenced_time_); |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_b_))); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::USER_VISIBLE, sort_key.priority_); |
+ EXPECT_EQ(task_a_.sequenced_time, sort_key.next_task_sequenced_time_); |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_c_))); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::USER_BLOCKING, sort_key.priority_); |
+ EXPECT_EQ(task_a_.sequenced_time, sort_key.next_task_sequenced_time_); |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_d_))); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::USER_BLOCKING, sort_key.priority_); |
+ EXPECT_EQ(task_a_.sequenced_time, sort_key.next_task_sequenced_time_); |
+ |
+ sequence->PopTask(); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::USER_BLOCKING, sort_key.priority_); |
+ EXPECT_EQ(task_b_.sequenced_time, sort_key.next_task_sequenced_time_); |
+ |
+ sequence->PopTask(); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::USER_BLOCKING, sort_key.priority_); |
+ EXPECT_EQ(task_c_.sequenced_time, sort_key.next_task_sequenced_time_); |
+ |
+ sequence->PopTask(); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::USER_BLOCKING, sort_key.priority_); |
+ EXPECT_EQ(task_d_.sequenced_time, sort_key.next_task_sequenced_time_); |
+ |
+ sequence->PushTask(scoped_ptr<Task>(new Task(task_a_))); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::USER_BLOCKING, sort_key.priority_); |
+ EXPECT_EQ(task_d_.sequenced_time, sort_key.next_task_sequenced_time_); |
+ |
+ sequence->PopTask(); |
+ sort_key = sequence->GetSortKey(); |
+ EXPECT_EQ(TaskPriority::BACKGROUND, sort_key.priority_); |
+ EXPECT_EQ(task_a_.sequenced_time, sort_key.next_task_sequenced_time_); |
+} |
+ |
+} // namespace internal |
+} // namespace base |