OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 // Common utilities for Quic tests | 5 // Common utilities for Quic tests |
6 | 6 |
7 #ifndef NET_QUIC_TEST_TOOLS_TEST_TASK_RUNNER_H_ | 7 #ifndef NET_QUIC_TEST_TOOLS_TEST_TASK_RUNNER_H_ |
8 #define NET_QUIC_TEST_TOOLS_TEST_TASK_RUNNER_H_ | 8 #define NET_QUIC_TEST_TOOLS_TEST_TASK_RUNNER_H_ |
9 | 9 |
10 #include <vector> | 10 #include <vector> |
11 | 11 |
| 12 #include "base/callback.h" |
12 #include "base/macros.h" | 13 #include "base/macros.h" |
13 #include "base/task_runner.h" | 14 #include "base/task_runner.h" |
14 #include "base/test/test_pending_task.h" | 15 #include "base/test/test_pending_task.h" |
15 | 16 |
16 namespace net { | 17 namespace net { |
17 | 18 |
18 class MockClock; | 19 class MockClock; |
19 | 20 |
20 namespace test { | 21 namespace test { |
21 | 22 |
22 typedef base::TestPendingTask PostedTask; | 23 typedef base::TestPendingTask PostedTask; |
23 | 24 |
24 class TestTaskRunner : public base::TaskRunner { | 25 class TestTaskRunner : public base::TaskRunner { |
25 public: | 26 public: |
26 explicit TestTaskRunner(MockClock* clock); | 27 explicit TestTaskRunner(MockClock* clock); |
27 | 28 |
28 // base::TaskRunner implementation. | 29 // base::TaskRunner implementation. |
29 bool PostDelayedTask(const tracked_objects::Location& from_here, | 30 bool PostDelayedTask(const tracked_objects::Location& from_here, |
30 const base::Closure& task, | 31 base::Closure task, |
31 base::TimeDelta delay) override; | 32 base::TimeDelta delay) override; |
32 bool RunsTasksOnCurrentThread() const override; | 33 bool RunsTasksOnCurrentThread() const override; |
33 | 34 |
34 const std::vector<PostedTask>& GetPostedTasks() const; | 35 const std::vector<PostedTask>& GetPostedTasks() const; |
35 | 36 |
36 void RunNextTask(); | 37 void RunNextTask(); |
37 | 38 |
38 protected: | 39 protected: |
39 ~TestTaskRunner() override; | 40 ~TestTaskRunner() override; |
40 | 41 |
41 private: | 42 private: |
42 std::vector<PostedTask>::iterator FindNextTask(); | 43 std::vector<PostedTask>::iterator FindNextTask(); |
43 | 44 |
44 MockClock* const clock_; | 45 MockClock* const clock_; |
45 std::vector<PostedTask> tasks_; | 46 std::vector<PostedTask> tasks_; |
46 | 47 |
47 DISALLOW_COPY_AND_ASSIGN(TestTaskRunner); | 48 DISALLOW_COPY_AND_ASSIGN(TestTaskRunner); |
48 }; | 49 }; |
49 | 50 |
50 } // namespace test | 51 } // namespace test |
51 | 52 |
52 } // namespace net | 53 } // namespace net |
53 | 54 |
54 #endif // NET_QUIC_TEST_TOOLS_TEST_TASK_RUNNER_H_ | 55 #endif // NET_QUIC_TEST_TOOLS_TEST_TASK_RUNNER_H_ |
OLD | NEW |