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 #include "base/test/thread_test_helper.h" | 5 #include "base/test/thread_test_helper.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/location.h" | 8 #include "base/location.h" |
| 9 #include "base/threading/thread_restrictions.h" |
9 | 10 |
10 namespace base { | 11 namespace base { |
11 | 12 |
12 ThreadTestHelper::ThreadTestHelper(MessageLoopProxy* target_thread) | 13 ThreadTestHelper::ThreadTestHelper(MessageLoopProxy* target_thread) |
13 : test_result_(false), | 14 : test_result_(false), |
14 target_thread_(target_thread), | 15 target_thread_(target_thread), |
15 done_event_(false, false) { | 16 done_event_(false, false) { |
16 } | 17 } |
17 | 18 |
18 bool ThreadTestHelper::Run() { | 19 bool ThreadTestHelper::Run() { |
19 if (!target_thread_->PostTask( | 20 if (!target_thread_->PostTask( |
20 FROM_HERE, base::Bind(&ThreadTestHelper::RunInThread, this))) { | 21 FROM_HERE, base::Bind(&ThreadTestHelper::RunInThread, this))) { |
21 return false; | 22 return false; |
22 } | 23 } |
| 24 base::ThreadRestrictions::ScopedAllowWait allow_wait; |
23 done_event_.Wait(); | 25 done_event_.Wait(); |
24 return test_result_; | 26 return test_result_; |
25 } | 27 } |
26 | 28 |
27 void ThreadTestHelper::RunTest() { set_test_result(true); } | 29 void ThreadTestHelper::RunTest() { set_test_result(true); } |
28 | 30 |
29 ThreadTestHelper::~ThreadTestHelper() {} | 31 ThreadTestHelper::~ThreadTestHelper() {} |
30 | 32 |
31 void ThreadTestHelper::RunInThread() { | 33 void ThreadTestHelper::RunInThread() { |
32 RunTest(); | 34 RunTest(); |
33 done_event_.Signal(); | 35 done_event_.Signal(); |
34 } | 36 } |
35 | 37 |
36 } // namespace base | 38 } // namespace base |
OLD | NEW |