| OLD | NEW |
| (Empty) |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "base/test/test_io_thread.h" | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "base/callback.h" | |
| 9 #include "base/synchronization/waitable_event.h" | |
| 10 | |
| 11 namespace { | |
| 12 | |
| 13 void PostTaskAndWaitHelper(base::WaitableEvent* event, | |
| 14 const base::Closure& task) { | |
| 15 task.Run(); | |
| 16 event->Signal(); | |
| 17 } | |
| 18 | |
| 19 } // namespace | |
| 20 | |
| 21 namespace base { | |
| 22 | |
| 23 TestIOThread::TestIOThread(Mode mode) | |
| 24 : io_thread_("test_io_thread"), io_thread_started_(false) { | |
| 25 switch (mode) { | |
| 26 case kAutoStart: | |
| 27 Start(); | |
| 28 return; | |
| 29 case kManualStart: | |
| 30 return; | |
| 31 } | |
| 32 CHECK(false) << "Invalid mode"; | |
| 33 } | |
| 34 | |
| 35 TestIOThread::~TestIOThread() { | |
| 36 Stop(); | |
| 37 } | |
| 38 | |
| 39 void TestIOThread::Start() { | |
| 40 CHECK(!io_thread_started_); | |
| 41 io_thread_started_ = true; | |
| 42 CHECK(io_thread_.StartWithOptions( | |
| 43 base::Thread::Options(base::MessageLoop::TYPE_IO, 0))); | |
| 44 } | |
| 45 | |
| 46 void TestIOThread::Stop() { | |
| 47 // Note: It's okay to call |Stop()| even if the thread isn't running. | |
| 48 io_thread_.Stop(); | |
| 49 io_thread_started_ = false; | |
| 50 } | |
| 51 | |
| 52 void TestIOThread::PostTask(const tracked_objects::Location& from_here, | |
| 53 const base::Closure& task) { | |
| 54 task_runner()->PostTask(from_here, task); | |
| 55 } | |
| 56 | |
| 57 void TestIOThread::PostTaskAndWait(const tracked_objects::Location& from_here, | |
| 58 const base::Closure& task) { | |
| 59 base::WaitableEvent event(false, false); | |
| 60 task_runner()->PostTask(from_here, | |
| 61 base::Bind(&PostTaskAndWaitHelper, &event, task)); | |
| 62 event.Wait(); | |
| 63 } | |
| 64 | |
| 65 } // namespace base | |
| OLD | NEW |