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