blob: 0a2c9a99c70c3f8955b63fb69840bda8edefa61d [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.
#include <stdint.h>
#include <memory>
#include <utility>
#include "ash/session/test_session_controller_client.h"
#include "base/macros.h"
#include "base/test/scoped_command_line.h"
#include "services/service_manager/public/cpp/test/test_connector_factory.h"
class PrefService;
namespace aura {
class Window;
namespace test {
class EnvWindowTreeClientSetter;
namespace chromeos {
namespace system {
class ScopedFakeStatisticsProvider;
} // namespace system
} // namespace chromeos
namespace display {
class Display;
namespace service_manager {
class Connector;
namespace ui {
class ScopedAnimationDurationScaleMode;
namespace views {
class MusClient;
namespace wm {
class WMState;
namespace ash {
class AppListTestHelper;
class AshTestViewsDelegate;
class TestKeyboardControllerObserver;
class TestShellDelegate;
// A helper class that does common initialization required for Ash. Creates a
// root window and an ash::Shell instance with a test delegate.
class AshTestHelper {
// Creates the ash::Shell and performs associated initialization. Set
// |start_session| to true if the user should log in before the test is run.
// Set |provide_local_state| to true to inject local-state PrefService into
// the Shell before the test is run.
void SetUp(bool start_session, bool provide_local_state = true);
// Destroys the ash::Shell and performs associated cleanup.
void TearDown();
// Call this only if this code is being run outside of ash, for example, in
// browser tests that use AshTestBase. This disables CHECKs that are
// applicable only when used inside ash.
// TODO: remove this and ban usage of AshTestHelper outside of ash.
void SetRunningOutsideAsh();
// Returns a root Window. Usually this is the active root Window, but that
// method can return NULL sometimes, and in those cases, we fall back on the
// primary root Window.
aura::Window* CurrentContext();
PrefService* GetLocalStatePrefService();
TestShellDelegate* test_shell_delegate() { return test_shell_delegate_; }
void set_test_shell_delegate(TestShellDelegate* test_shell_delegate) {
test_shell_delegate_ = test_shell_delegate;
AshTestViewsDelegate* test_views_delegate() {
return test_views_delegate_.get();
display::Display GetSecondaryDisplay();
TestSessionControllerClient* test_session_controller_client() {
return session_controller_client_.get();
void set_test_session_controller_client(
std::unique_ptr<TestSessionControllerClient> session_controller_client) {
session_controller_client_ = std::move(session_controller_client);
AppListTestHelper* app_list_test_helper() {
return app_list_test_helper_.get();
TestKeyboardControllerObserver* test_keyboard_controller_observer() {
return test_keyboard_controller_observer_.get();
void reset_commandline() { command_line_.reset(); }
// Creates a MusClient. aura::Env's *must* be set to Mode::MUS. Easiest way
// to ensure that is by subclassing SingleProcessMashTestBase.
void CreateMusClient();
// Gets a Connector that talks directly to the WindowService.
service_manager::Connector* GetWindowServiceConnector();
// Forces creation of the WindowService. The WindowService is normally created
// on demand, this force the creation.
void CreateWindowService();
// Called when running in ash to create Shell.
void CreateShell();
TestShellDelegate* test_shell_delegate_ = nullptr; // Owned by ash::Shell.
std::unique_ptr<ui::ScopedAnimationDurationScaleMode> zero_duration_mode_;
std::unique_ptr<::wm::WMState> wm_state_;
std::unique_ptr<AshTestViewsDelegate> test_views_delegate_;
// Flags for whether various services were initialized here.
bool dbus_thread_manager_initialized_ = false;
bool bluez_dbus_manager_initialized_ = false;
bool power_policy_controller_initialized_ = false;
std::unique_ptr<TestSessionControllerClient> session_controller_client_;
std::unique_ptr<base::test::ScopedCommandLine> command_line_;
std::unique_ptr<AppListTestHelper> app_list_test_helper_;
service_manager::TestConnectorFactory test_connector_factory_;
std::unique_ptr<service_manager::Connector> window_service_connector_;
// |window_tree_client_setter_| and |mus_client_| are created by
// CreateMusClient(). See it for details.
std::unique_ptr<views::MusClient> mus_client_;
} // namespace ash