| // 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. |
| |
| #ifndef BASE_TASK_SCHEDULER_TEST_UTILS_H_ |
| #define BASE_TASK_SCHEDULER_TEST_UTILS_H_ |
| |
| #include "base/memory/ref_counted.h" |
| #include "base/task_runner.h" |
| #include "base/task_scheduler/scheduler_worker_observer.h" |
| #include "base/task_scheduler/sequence.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| |
| namespace base { |
| namespace internal { |
| |
| class SchedulerWorkerPool; |
| struct Task; |
| |
| namespace test { |
| |
| class MockSchedulerWorkerObserver : public SchedulerWorkerObserver { |
| public: |
| MockSchedulerWorkerObserver(); |
| ~MockSchedulerWorkerObserver(); |
| |
| MOCK_METHOD0(OnSchedulerWorkerMainEntry, void()); |
| MOCK_METHOD0(OnSchedulerWorkerMainExit, void()); |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(MockSchedulerWorkerObserver); |
| }; |
| |
| // An enumeration of possible task scheduler TaskRunner types. Used to |
| // parametrize relevant task_scheduler tests. |
| enum class ExecutionMode { PARALLEL, SEQUENCED, SINGLE_THREADED }; |
| |
| // Creates a Sequence and pushes |task| to it. Returns that sequence. |
| scoped_refptr<Sequence> CreateSequenceWithTask(Task task); |
| |
| // Creates a TaskRunner that posts tasks to |worker_pool| with the |
| // |execution_mode| execution mode and the WithBaseSyncPrimitives() trait. |
| // Caveat: this does not support ExecutionMode::SINGLE_THREADED. |
| scoped_refptr<TaskRunner> CreateTaskRunnerWithExecutionMode( |
| SchedulerWorkerPool* worker_pool, |
| test::ExecutionMode execution_mode); |
| |
| } // namespace test |
| } // namespace internal |
| } // namespace base |
| |
| #endif // BASE_TASK_SCHEDULER_TEST_UTILS_H_ |