OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "components/scheduler/base/task_queue_sets.h" | 5 #include "components/scheduler/base/task_queue_sets.h" |
6 | 6 |
7 #include "components/scheduler/base/task_queue_impl.h" | 7 #include "components/scheduler/base/task_queue_impl.h" |
| 8 #include "components/scheduler/base/virtual_time_domain.h" |
8 #include "testing/gmock/include/gmock/gmock.h" | 9 #include "testing/gmock/include/gmock/gmock.h" |
9 | 10 |
10 namespace scheduler { | 11 namespace scheduler { |
| 12 class TimeDomain; |
| 13 |
11 namespace internal { | 14 namespace internal { |
12 | 15 |
13 class TaskQueueSetsTest : public testing::Test { | 16 class TaskQueueSetsTest : public testing::Test { |
14 public: | 17 public: |
15 void SetUp() override { task_queue_sets_.reset(new TaskQueueSets(kNumSets)); } | 18 void SetUp() override { |
| 19 virtual_time_domain_ = make_scoped_refptr<VirtualTimeDomain>( |
| 20 new VirtualTimeDomain(base::TimeTicks())); |
| 21 task_queue_sets_.reset(new TaskQueueSets(kNumSets)); |
| 22 } |
16 | 23 |
17 protected: | 24 protected: |
18 enum { | 25 enum { |
19 kNumSets = 5 // An arbitary choice. | 26 kNumSets = 5 // An arbitary choice. |
20 }; | 27 }; |
21 | 28 |
22 TaskQueueImpl* NewTaskQueue(const char* queue_name) { | 29 TaskQueueImpl* NewTaskQueue(const char* queue_name) { |
23 scoped_refptr<internal::TaskQueueImpl> queue = | 30 scoped_refptr<internal::TaskQueueImpl> queue = |
24 make_scoped_refptr(new internal::TaskQueueImpl( | 31 make_scoped_refptr(new internal::TaskQueueImpl( |
25 nullptr, TaskQueue::Spec(queue_name), "test", "test")); | 32 nullptr, virtual_time_domain_, TaskQueue::Spec(queue_name), "test", |
| 33 "test")); |
26 task_queues_.push_back(queue); | 34 task_queues_.push_back(queue); |
27 return queue.get(); | 35 return queue.get(); |
28 } | 36 } |
29 | 37 |
30 TaskQueueImpl::Task FakeTaskWithEnqueueOrder(int enqueue_order) { | 38 TaskQueueImpl::Task FakeTaskWithEnqueueOrder(int enqueue_order) { |
31 TaskQueueImpl::Task fake_task(FROM_HERE, base::Closure(), 0, true); | 39 TaskQueueImpl::Task fake_task(FROM_HERE, base::Closure(), 0, true); |
32 fake_task.set_enqueue_order(enqueue_order); | 40 fake_task.set_enqueue_order(enqueue_order); |
33 return fake_task; | 41 return fake_task; |
34 } | 42 } |
35 | 43 |
| 44 scoped_refptr<VirtualTimeDomain> virtual_time_domain_; |
36 std::vector<scoped_refptr<internal::TaskQueueImpl>> task_queues_; | 45 std::vector<scoped_refptr<internal::TaskQueueImpl>> task_queues_; |
37 scoped_ptr<TaskQueueSets> task_queue_sets_; | 46 scoped_ptr<TaskQueueSets> task_queue_sets_; |
38 }; | 47 }; |
39 | 48 |
40 TEST_F(TaskQueueSetsTest, AssignQueueToSet) { | 49 TEST_F(TaskQueueSetsTest, AssignQueueToSet) { |
41 internal::TaskQueueImpl* queue = NewTaskQueue("queue"); | 50 internal::TaskQueueImpl* queue = NewTaskQueue("queue"); |
42 size_t set = TaskQueue::NORMAL_PRIORITY; | 51 size_t set = TaskQueue::NORMAL_PRIORITY; |
43 task_queue_sets_->AssignQueueToSet(queue, set); | 52 task_queue_sets_->AssignQueueToSet(queue, set); |
44 | 53 |
45 EXPECT_EQ(set, queue->get_task_queue_set_index()); | 54 EXPECT_EQ(set, queue->get_task_queue_set_index()); |
(...skipping 184 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
230 task_queue_sets_->AssignQueueToSet(queue, set); | 239 task_queue_sets_->AssignQueueToSet(queue, set); |
231 EXPECT_FALSE(task_queue_sets_->IsSetEmpty(set)); | 240 EXPECT_FALSE(task_queue_sets_->IsSetEmpty(set)); |
232 | 241 |
233 queue->PopTaskFromWorkQueueForTest(); | 242 queue->PopTaskFromWorkQueueForTest(); |
234 task_queue_sets_->OnPopQueue(queue); | 243 task_queue_sets_->OnPopQueue(queue); |
235 EXPECT_TRUE(task_queue_sets_->IsSetEmpty(set)); | 244 EXPECT_TRUE(task_queue_sets_->IsSetEmpty(set)); |
236 } | 245 } |
237 | 246 |
238 } // namespace internal | 247 } // namespace internal |
239 } // namespace scheduler | 248 } // namespace scheduler |
OLD | NEW |