| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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_TEST_MESSAGE_LOOP_H_ | 5 #ifndef BASE_TEST_TEST_MESSAGE_LOOP_H_ |
| 6 #define BASE_TEST_TEST_MESSAGE_LOOP_H_ | 6 #define BASE_TEST_TEST_MESSAGE_LOOP_H_ |
| 7 | 7 |
| 8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
| 9 | 9 |
| 10 namespace base { | 10 namespace base { |
| 11 | 11 |
| 12 // TestMessageLoop is a convenience class for unittests that need to create a | 12 // TestMessageLoop is a convenience class for unittests that need to create a |
| 13 // message loop without a real thread backing it. For most tests, | 13 // message loop without a real thread backing it. For most tests, |
| 14 // it is sufficient to just instantiate TestMessageLoop as a member variable. | 14 // it is sufficient to just instantiate TestMessageLoop as a member variable. |
| 15 // | 15 // |
| 16 // TestMessageLoop will attempt to drain the underlying MessageLoop on | 16 // TestMessageLoop will attempt to drain the underlying MessageLoop on |
| 17 // destruction for clean teardown of tests. | 17 // destruction for clean teardown of tests. |
| 18 class TestMessageLoop { | 18 class TestMessageLoop { |
| 19 public: | 19 public: |
| 20 TestMessageLoop(); | 20 TestMessageLoop(); |
| 21 explicit TestMessageLoop(MessageLoop::Type type); | 21 explicit TestMessageLoop(MessageLoop::Type type); |
| 22 ~TestMessageLoop(); | 22 ~TestMessageLoop(); |
| 23 | 23 |
| 24 const scoped_refptr<SingleThreadTaskRunner>& task_runner() { |
| 25 return loop_.task_runner(); |
| 26 } |
| 27 |
| 24 private: | 28 private: |
| 25 MessageLoop loop_; | 29 MessageLoop loop_; |
| 26 }; | 30 }; |
| 27 | 31 |
| 28 } // namespace base | 32 } // namespace base |
| 29 | 33 |
| 30 #endif // BASE_TEST_TEST_MESSAGE_LOOP_H_ | 34 #endif // BASE_TEST_TEST_MESSAGE_LOOP_H_ |
| OLD | NEW |