123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104 |
- // Copyright (c) 2012 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/test/test_simple_task_runner.h"
- #include <utility>
- #include "base/check.h"
- #include "base/memory/ptr_util.h"
- #include "base/threading/thread_task_runner_handle.h"
- #include "third_party/abseil-cpp/absl/types/optional.h"
- namespace base {
- TestSimpleTaskRunner::TestSimpleTaskRunner() = default;
- TestSimpleTaskRunner::~TestSimpleTaskRunner() = default;
- bool TestSimpleTaskRunner::PostDelayedTask(const Location& from_here,
- OnceClosure task,
- TimeDelta delay) {
- AutoLock auto_lock(lock_);
- pending_tasks_.push_back(TestPendingTask(from_here, std::move(task),
- TimeTicks(), delay,
- TestPendingTask::NESTABLE));
- return true;
- }
- bool TestSimpleTaskRunner::PostNonNestableDelayedTask(const Location& from_here,
- OnceClosure task,
- TimeDelta delay) {
- AutoLock auto_lock(lock_);
- pending_tasks_.push_back(TestPendingTask(from_here, std::move(task),
- TimeTicks(), delay,
- TestPendingTask::NON_NESTABLE));
- return true;
- }
- // TODO(gab): Use SequenceToken here to differentiate between tasks running in
- // the scope of this TestSimpleTaskRunner and other task runners sharing this
- // thread. http://crbug.com/631186
- bool TestSimpleTaskRunner::RunsTasksInCurrentSequence() const {
- return thread_ref_ == PlatformThread::CurrentRef();
- }
- base::circular_deque<TestPendingTask> TestSimpleTaskRunner::TakePendingTasks() {
- AutoLock auto_lock(lock_);
- return std::move(pending_tasks_);
- }
- size_t TestSimpleTaskRunner::NumPendingTasks() const {
- AutoLock auto_lock(lock_);
- return pending_tasks_.size();
- }
- bool TestSimpleTaskRunner::HasPendingTask() const {
- AutoLock auto_lock(lock_);
- return !pending_tasks_.empty();
- }
- base::TimeDelta TestSimpleTaskRunner::NextPendingTaskDelay() const {
- AutoLock auto_lock(lock_);
- return pending_tasks_.front().GetTimeToRun() - base::TimeTicks();
- }
- base::TimeDelta TestSimpleTaskRunner::FinalPendingTaskDelay() const {
- AutoLock auto_lock(lock_);
- return pending_tasks_.back().GetTimeToRun() - base::TimeTicks();
- }
- void TestSimpleTaskRunner::ClearPendingTasks() {
- AutoLock auto_lock(lock_);
- pending_tasks_.clear();
- }
- void TestSimpleTaskRunner::RunPendingTasks() {
- DCHECK(RunsTasksInCurrentSequence());
- // Swap with a local variable to avoid re-entrancy problems.
- base::circular_deque<TestPendingTask> tasks_to_run;
- {
- AutoLock auto_lock(lock_);
- tasks_to_run.swap(pending_tasks_);
- }
- // Multiple test task runners can share the same thread for determinism in
- // unit tests. Make sure this TestSimpleTaskRunner's tasks run in its scope.
- absl::optional<ThreadTaskRunnerHandleOverrideForTesting> ttrh_override;
- if (!ThreadTaskRunnerHandle::IsSet() ||
- ThreadTaskRunnerHandle::Get() != this) {
- ttrh_override.emplace(this);
- }
- for (auto& task : tasks_to_run)
- std::move(task.task).Run();
- }
- void TestSimpleTaskRunner::RunUntilIdle() {
- while (HasPendingTask())
- RunPendingTasks();
- }
- } // namespace base
|