OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 BASE_TEST_THREAD_TEST_HELPER_H_ | 5 #ifndef BASE_TEST_THREAD_TEST_HELPER_H_ |
6 #define BASE_TEST_THREAD_TEST_HELPER_H_ | 6 #define BASE_TEST_THREAD_TEST_HELPER_H_ |
7 | 7 |
8 #include "base/compiler_specific.h" | 8 #include "base/compiler_specific.h" |
9 #include "base/memory/ref_counted.h" | 9 #include "base/memory/ref_counted.h" |
10 #include "base/message_loop/message_loop_proxy.h" | |
11 #include "base/synchronization/waitable_event.h" | 10 #include "base/synchronization/waitable_event.h" |
12 | 11 |
13 namespace base { | 12 namespace base { |
14 | 13 |
| 14 class SequencedTaskRunner; |
| 15 |
15 // Helper class that executes code on a given thread while blocking on the | 16 // Helper class that executes code on a given thread while blocking on the |
16 // invoking thread. To use, derive from this class and overwrite RunTest. An | 17 // invoking thread. To use, derive from this class and overwrite RunTest. An |
17 // alternative use of this class is to use it directly. It will then block | 18 // alternative use of this class is to use it directly. It will then block |
18 // until all pending tasks on a given thread have been executed. | 19 // until all pending tasks on a given thread have been executed. |
19 class ThreadTestHelper : public RefCountedThreadSafe<ThreadTestHelper> { | 20 class ThreadTestHelper : public RefCountedThreadSafe<ThreadTestHelper> { |
20 public: | 21 public: |
21 explicit ThreadTestHelper( | 22 explicit ThreadTestHelper( |
22 const scoped_refptr<MessageLoopProxy>& target_thread); | 23 const scoped_refptr<SequencedTaskRunner>& target_task_runner); |
23 | 24 |
24 // True if RunTest() was successfully executed on the target thread. | 25 // True if RunTest() was successfully executed on the target thread. |
25 bool Run() WARN_UNUSED_RESULT; | 26 bool Run() WARN_UNUSED_RESULT; |
26 | 27 |
27 virtual void RunTest(); | 28 virtual void RunTest(); |
28 | 29 |
29 protected: | 30 protected: |
30 friend class RefCountedThreadSafe<ThreadTestHelper>; | 31 friend class RefCountedThreadSafe<ThreadTestHelper>; |
31 | 32 |
32 virtual ~ThreadTestHelper(); | 33 virtual ~ThreadTestHelper(); |
33 | 34 |
34 // Use this method to store the result of RunTest(). | 35 // Use this method to store the result of RunTest(). |
35 void set_test_result(bool test_result) { test_result_ = test_result; } | 36 void set_test_result(bool test_result) { test_result_ = test_result; } |
36 | 37 |
37 private: | 38 private: |
38 void RunInThread(); | 39 void RunInThread(); |
39 | 40 |
40 bool test_result_; | 41 bool test_result_; |
41 scoped_refptr<MessageLoopProxy> target_thread_; | 42 scoped_refptr<SequencedTaskRunner> target_task_runner_; |
42 WaitableEvent done_event_; | 43 WaitableEvent done_event_; |
43 | 44 |
44 DISALLOW_COPY_AND_ASSIGN(ThreadTestHelper); | 45 DISALLOW_COPY_AND_ASSIGN(ThreadTestHelper); |
45 }; | 46 }; |
46 | 47 |
47 } // namespace base | 48 } // namespace base |
48 | 49 |
49 #endif // BASE_TEST_THREAD_TEST_HELPER_H_ | 50 #endif // BASE_TEST_THREAD_TEST_HELPER_H_ |
OLD | NEW |