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 66
