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 11namespace { 12 13void PostTaskAndWaitHelper(base::WaitableEvent* event, 14 const base::Closure& task) { 15 task.Run(); 16 event->Signal(); 17} 18 19} // namespace 20 21namespace base { 22 23TestIOThread::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 35TestIOThread::~TestIOThread() { 36 Stop(); 37} 38 39void 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 46void 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 52void TestIOThread::PostTask(const tracked_objects::Location& from_here, 53 const base::Closure& task) { 54 task_runner()->PostTask(from_here, task); 55} 56 57void 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