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