| 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 #ifndef CONTENT_RENDERER_SCHEDULER_NESTABLE_TASK_RUNNER_FOR_TEST_H_ | 5 #ifndef CONTENT_RENDERER_SCHEDULER_NESTABLE_TASK_RUNNER_FOR_TEST_H_ |
| 6 #define CONTENT_RENDERER_SCHEDULER_NESTABLE_TASK_RUNNER_FOR_TEST_H_ | 6 #define CONTENT_RENDERER_SCHEDULER_NESTABLE_TASK_RUNNER_FOR_TEST_H_ |
| 7 | 7 |
| 8 #include "components/scheduler/child/nestable_single_thread_task_runner.h" | 8 #include "components/scheduler/child/nestable_single_thread_task_runner.h" |
| 9 | 9 |
| 10 namespace scheduler { | 10 namespace scheduler { |
| 11 | 11 |
| 12 class NestableTaskRunnerForTest : public NestableSingleThreadTaskRunner { | 12 class NestableTaskRunnerForTest : public NestableSingleThreadTaskRunner { |
| 13 public: | 13 public: |
| 14 static scoped_refptr<NestableTaskRunnerForTest> Create( | 14 static scoped_refptr<NestableTaskRunnerForTest> Create( |
| 15 scoped_refptr<base::SingleThreadTaskRunner> task_runner); | 15 scoped_refptr<base::SingleThreadTaskRunner> task_runner); |
| 16 | 16 |
| 17 void SetNested(bool is_nested); | 17 void SetNested(bool is_nested); |
| 18 | 18 |
| 19 // NestableSingleThreadTaskRunner implementation | 19 // NestableSingleThreadTaskRunner implementation |
| 20 bool PostDelayedTask(const tracked_objects::Location& from_here, | 20 bool PostDelayedTask(const tracked_objects::Location& from_here, |
| 21 const base::Closure& task, | 21 const base::Closure& task, |
| 22 base::TimeDelta delay) override; | 22 base::TimeDelta delay) override; |
| 23 bool PostNonNestableDelayedTask(const tracked_objects::Location& from_here, | 23 bool PostNonNestableDelayedTask(const tracked_objects::Location& from_here, |
| 24 const base::Closure& task, | 24 const base::Closure& task, |
| 25 base::TimeDelta delay) override; | 25 base::TimeDelta delay) override; |
| 26 bool RunsTasksOnCurrentThread() const override; | 26 bool RunsTasksOnCurrentThread() const override; |
| 27 bool IsNested() const override; | 27 bool IsNested() const override; |
| 28 void AddTaskObserver( | 28 void AddTaskObserver(base::MessageLoop::TaskObserver* task_observer) override; |
| 29 base::MessageLoop::TaskObserver* task_observer) override; | |
| 30 void RemoveTaskObserver( | 29 void RemoveTaskObserver( |
| 31 base::MessageLoop::TaskObserver* task_observer) override; | 30 base::MessageLoop::TaskObserver* task_observer) override; |
| 32 | 31 |
| 33 protected: | 32 protected: |
| 34 ~NestableTaskRunnerForTest() override; | 33 ~NestableTaskRunnerForTest() override; |
| 35 | 34 |
| 36 private: | 35 private: |
| 37 NestableTaskRunnerForTest( | 36 NestableTaskRunnerForTest( |
| 38 scoped_refptr<base::SingleThreadTaskRunner> task_runner); | 37 scoped_refptr<base::SingleThreadTaskRunner> task_runner); |
| 39 | 38 |
| 39 void WrapTask(const base::PendingTask* wrapped_task); |
| 40 |
| 40 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; | 41 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
| 41 bool is_nested_; | 42 bool is_nested_; |
| 42 | 43 |
| 44 ObserverList<base::MessageLoop::TaskObserver> task_observers_; |
| 45 |
| 46 base::WeakPtr<NestableTaskRunnerForTest> weak_nestable_task_runner_ptr_; |
| 47 base::WeakPtrFactory<NestableTaskRunnerForTest> weak_factory_; |
| 48 |
| 43 DISALLOW_COPY_AND_ASSIGN(NestableTaskRunnerForTest); | 49 DISALLOW_COPY_AND_ASSIGN(NestableTaskRunnerForTest); |
| 44 }; | 50 }; |
| 45 | 51 |
| 46 } // namespace scheduler | 52 } // namespace scheduler |
| 47 | 53 |
| 48 #endif // CONTENT_RENDERER_SCHEDULER_NESTABLE_TASK_RUNNER_FOR_TEST_H_ | 54 #endif // CONTENT_RENDERER_SCHEDULER_NESTABLE_TASK_RUNNER_FOR_TEST_H_ |
| OLD | NEW |