| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 "base/task_scheduler/scheduler_worker_stack.h" | 5 #include "base/task_scheduler/scheduler_worker_stack.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "base/memory/ptr_util.h" | 8 #include "base/memory/ptr_util.h" |
| 9 #include "base/memory/ref_counted.h" | 9 #include "base/memory/ref_counted.h" |
| 10 #include "base/task_scheduler/scheduler_worker.h" | 10 #include "base/task_scheduler/scheduler_worker.h" |
| (...skipping 14 matching lines...) Expand all Loading... |
| 25 void OnMainEntry(SchedulerWorker* worker) override {} | 25 void OnMainEntry(SchedulerWorker* worker) override {} |
| 26 scoped_refptr<Sequence> GetWork(SchedulerWorker* worker) override { | 26 scoped_refptr<Sequence> GetWork(SchedulerWorker* worker) override { |
| 27 return nullptr; | 27 return nullptr; |
| 28 } | 28 } |
| 29 void ReEnqueueSequence(scoped_refptr<Sequence> sequence) override { | 29 void ReEnqueueSequence(scoped_refptr<Sequence> sequence) override { |
| 30 ADD_FAILURE() << "This delegate not expect to have sequences to reenqueue."; | 30 ADD_FAILURE() << "This delegate not expect to have sequences to reenqueue."; |
| 31 } | 31 } |
| 32 TimeDelta GetSleepTimeout() override { | 32 TimeDelta GetSleepTimeout() override { |
| 33 return TimeDelta::Max(); | 33 return TimeDelta::Max(); |
| 34 } | 34 } |
| 35 bool CanDetach(SchedulerWorker* worker) override { |
| 36 return false; |
| 37 } |
| 35 }; | 38 }; |
| 36 | 39 |
| 37 class TaskSchedulerWorkerStackTest : public testing::Test { | 40 class TaskSchedulerWorkerStackTest : public testing::Test { |
| 38 protected: | 41 protected: |
| 39 void SetUp() override { | 42 void SetUp() override { |
| 40 worker_a_ = SchedulerWorker::Create( | 43 worker_a_ = SchedulerWorker::Create( |
| 41 ThreadPriority::NORMAL, | 44 ThreadPriority::NORMAL, |
| 42 WrapUnique(new MockSchedulerWorkerDelegate), &task_tracker_); | 45 WrapUnique(new MockSchedulerWorkerDelegate), &task_tracker_, |
| 46 SchedulerWorker::InitialState::ALIVE); |
| 43 ASSERT_TRUE(worker_a_); | 47 ASSERT_TRUE(worker_a_); |
| 44 worker_b_ = SchedulerWorker::Create( | 48 worker_b_ = SchedulerWorker::Create( |
| 45 ThreadPriority::NORMAL, | 49 ThreadPriority::NORMAL, |
| 46 WrapUnique(new MockSchedulerWorkerDelegate), &task_tracker_); | 50 WrapUnique(new MockSchedulerWorkerDelegate), &task_tracker_, |
| 51 SchedulerWorker::InitialState::ALIVE); |
| 47 ASSERT_TRUE(worker_b_); | 52 ASSERT_TRUE(worker_b_); |
| 48 worker_c_ = SchedulerWorker::Create( | 53 worker_c_ = SchedulerWorker::Create( |
| 49 ThreadPriority::NORMAL, | 54 ThreadPriority::NORMAL, |
| 50 WrapUnique(new MockSchedulerWorkerDelegate), &task_tracker_); | 55 WrapUnique(new MockSchedulerWorkerDelegate), &task_tracker_, |
| 56 SchedulerWorker::InitialState::ALIVE); |
| 51 ASSERT_TRUE(worker_c_); | 57 ASSERT_TRUE(worker_c_); |
| 52 } | 58 } |
| 53 | 59 |
| 54 void TearDown() override { | 60 void TearDown() override { |
| 55 worker_a_->JoinForTesting(); | 61 worker_a_->JoinForTesting(); |
| 56 worker_b_->JoinForTesting(); | 62 worker_b_->JoinForTesting(); |
| 57 worker_c_->JoinForTesting(); | 63 worker_c_->JoinForTesting(); |
| 58 } | 64 } |
| 59 | 65 |
| 60 std::unique_ptr<SchedulerWorker> worker_a_; | 66 std::unique_ptr<SchedulerWorker> worker_a_; |
| (...skipping 97 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 158 | 164 |
| 159 // Verify that Push() DCHECKs when a value is inserted twice. | 165 // Verify that Push() DCHECKs when a value is inserted twice. |
| 160 TEST_F(TaskSchedulerWorkerStackTest, PushTwice) { | 166 TEST_F(TaskSchedulerWorkerStackTest, PushTwice) { |
| 161 SchedulerWorkerStack stack; | 167 SchedulerWorkerStack stack; |
| 162 stack.Push(worker_a_.get()); | 168 stack.Push(worker_a_.get()); |
| 163 EXPECT_DCHECK_DEATH({ stack.Push(worker_a_.get()); }, ""); | 169 EXPECT_DCHECK_DEATH({ stack.Push(worker_a_.get()); }, ""); |
| 164 } | 170 } |
| 165 | 171 |
| 166 } // namespace internal | 172 } // namespace internal |
| 167 } // namespace base | 173 } // namespace base |
| OLD | NEW |