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
PostTaskAndWaitHelper(base::WaitableEvent * event,const base::Closure & task)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
TestIOThread(Mode mode)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
~TestIOThread()35 TestIOThread::~TestIOThread() {
36 Stop();
37 }
38
Start()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
Stop()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
PostTask(const tracked_objects::Location & from_here,const base::Closure & task)52 void TestIOThread::PostTask(const tracked_objects::Location& from_here,
53 const base::Closure& task) {
54 task_runner()->PostTask(from_here, task);
55 }
56
PostTaskAndWait(const tracked_objects::Location & from_here,const base::Closure & task)57 void TestIOThread::PostTaskAndWait(const tracked_objects::Location& from_here,
58 const base::Closure& task) {
59 base::WaitableEvent event(WaitableEvent::ResetPolicy::AUTOMATIC,
60 WaitableEvent::InitialState::NOT_SIGNALED);
61 task_runner()->PostTask(from_here,
62 base::Bind(&PostTaskAndWaitHelper, &event, task));
63 event.Wait();
64 }
65
66 } // namespace base
67