blob: fe7f5fd7edf00eeabe371b9622a205667559212a [file] [log] [blame]
// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
// TestBrowserThreadBundle is a convenience class for creating a set of
// TestBrowserThreads in unit tests. For most tests, it is sufficient to
// just instantiate the TestBrowserThreadBundle as a member variable.
//
// By default, all of the created TestBrowserThreads will be backed by a single
// shared MessageLoop. If a test truly needs separate threads, it can do
// so by passing the appropriate combination of option values during
// the TestBrowserThreadBundle construction.
//
// The TestBrowserThreadBundle will attempt to drain the MessageLoop on
// destruction. Sometimes a test needs to drain currently enqueued tasks
// mid-test. Browser tests should call content::RunAllPendingInMessageLoop().
// Unit tests should use base::RunLoop (e.g., base::RunLoop().RunUntilIdle()).
// TODO(phajdan.jr): Revise this comment after switch to Aura.
//
// The TestBrowserThreadBundle will also flush the blocking pool on destruction.
// We do this to avoid memory leaks, particularly in the case of threads posting
// tasks to the blocking pool via PostTaskAndReply. By ensuring that the tasks
// are run while the originating TestBroswserThreads still exist, we prevent
// leakage of PostTaskAndReplyRelay objects. We also flush the blocking pool
// again at the point where it would normally be shut down, to better simulate
// the normal thread shutdown process.
//
// Some tests using the IO thread expect a MessageLoopForIO. Passing
// IO_MAINLOOP will use a MessageLoopForIO for the main MessageLoop.
// Most of the time, this avoids needing to use a REAL_IO_THREAD.
//
// For some tests it is important to emulate real browser startup. During real
// browser startup some initialization is done (e.g. creation of thread objects)
// between creating the main thread message loop, which is bound to the existing
// main thread, and starting the other threads. Passing DONT_START_THREADS to
// constructor will delay staring these other threads until the test explicitly
// calls Start().
//
// DONT_START_THREADS should only be used when the options specify at least
// one real thread other than the main thread.
#ifndef CONTENT_PUBLIC_TEST_TEST_BROWSER_THREAD_BUNDLE_H_
#define CONTENT_PUBLIC_TEST_TEST_BROWSER_THREAD_BUNDLE_H_
#include <memory>
#include "base/macros.h"
namespace base {
class MessageLoop;
} // namespace base
namespace content {
class TestBrowserThread;
class TestBrowserThreadBundle {
public:
// Used to specify the type of MessageLoop that backs the UI thread, and
// which of the named BrowserThreads should be backed by a real
// threads. The UI thread is always the main thread in a unit test.
enum Options {
DEFAULT = 0x00,
IO_MAINLOOP = 0x01,
REAL_DB_THREAD = 0x02,
REAL_FILE_THREAD = 0x08,
REAL_FILE_USER_BLOCKING_THREAD = 0x10,
REAL_PROCESS_LAUNCHER_THREAD = 0x20,
REAL_CACHE_THREAD = 0x40,
REAL_IO_THREAD = 0x80,
DONT_START_THREADS = 0x100,
};
TestBrowserThreadBundle();
explicit TestBrowserThreadBundle(int options);
// Start the real threads; should only be called from other classes if the
// DONT_START_THREADS option was used when the bundle was created.
void Start();
~TestBrowserThreadBundle();
private:
void Init();
std::unique_ptr<base::MessageLoop> message_loop_;
std::unique_ptr<TestBrowserThread> ui_thread_;
std::unique_ptr<TestBrowserThread> db_thread_;
std::unique_ptr<TestBrowserThread> file_thread_;
std::unique_ptr<TestBrowserThread> file_user_blocking_thread_;
std::unique_ptr<TestBrowserThread> process_launcher_thread_;
std::unique_ptr<TestBrowserThread> cache_thread_;
std::unique_ptr<TestBrowserThread> io_thread_;
int options_;
bool threads_started_;
DISALLOW_COPY_AND_ASSIGN(TestBrowserThreadBundle);
};
} // namespace content
#endif // CONTENT_PUBLIC_TEST_TEST_BROWSER_THREAD_BUNDLE_H_