blob: c9f6c7d56b0acdb60d36f56a42d1851eca36ac97 [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 "ash/test/ash_test_helper.h"
#include "ash/accelerators/accelerator_controller.h"
#include "ash/ash_switches.h"
#include "ash/shell.h"
#include "ash/shell_init_params.h"
#include "ash/test/ash_test_views_delegate.h"
#include "ash/test/display_manager_test_api.h"
#include "ash/test/shell_test_api.h"
#include "ash/test/test_screenshot_delegate.h"
#include "ash/test/test_session_state_delegate.h"
#include "ash/test/test_shell_delegate.h"
#include "ash/test/test_system_tray_delegate.h"
#include "base/run_loop.h"
#include "ui/aura/env.h"
#include "ui/aura/input_state_lookup.h"
#include "ui/aura/test/env_test_helper.h"
#include "ui/aura/test/event_generator_delegate_aura.h"
#include "ui/base/ime/input_method_initializer.h"
#include "ui/compositor/scoped_animation_duration_scale_mode.h"
#include "ui/compositor/test/context_factories_for_test.h"
#include "ui/message_center/message_center.h"
#include "ui/wm/core/capture_controller.h"
#if defined(OS_CHROMEOS)
#include "chromeos/audio/cras_audio_handler.h"
#include "chromeos/dbus/dbus_thread_manager.h"
#include "ui/keyboard/keyboard.h"
#endif
#if defined(OS_WIN)
#include "base/win/windows_version.h"
#endif
#if defined(USE_X11)
#include "ui/aura/window_tree_host_x11.h"
#endif
namespace ash {
namespace test {
AshTestHelper::AshTestHelper(base::MessageLoopForUI* message_loop)
: message_loop_(message_loop),
test_shell_delegate_(NULL),
test_screenshot_delegate_(NULL) {
CHECK(message_loop_);
#if defined(OS_CHROMEOS)
dbus_thread_manager_initialized_ = false;
#endif
#if defined(USE_X11)
aura::test::SetUseOverrideRedirectWindowByDefault(true);
#endif
aura::test::InitializeAuraEventGeneratorDelegate();
}
AshTestHelper::~AshTestHelper() {
}
void AshTestHelper::SetUp(bool start_session) {
views_delegate_.reset(new AshTestViewsDelegate);
// Disable animations during tests.
zero_duration_mode_.reset(new ui::ScopedAnimationDurationScaleMode(
ui::ScopedAnimationDurationScaleMode::ZERO_DURATION));
ui::InitializeInputMethodForTesting();
bool enable_pixel_output = false;
ui::ContextFactory* context_factory =
ui::InitializeContextFactoryForTests(enable_pixel_output);
// Creates Shell and hook with Desktop.
if (!test_shell_delegate_)
test_shell_delegate_ = new TestShellDelegate;
// Creates MessageCenter since g_browser_process is not created in AshTestBase
// tests.
message_center::MessageCenter::Initialize();
#if defined(OS_CHROMEOS)
// Create DBusThreadManager for testing.
if (!chromeos::DBusThreadManager::IsInitialized()) {
chromeos::DBusThreadManager::Initialize();
dbus_thread_manager_initialized_ = true;
}
// Create CrasAudioHandler for testing since g_browser_process is not
// created in AshTestBase tests.
chromeos::CrasAudioHandler::InitializeForTesting();
#endif
ShellInitParams init_params;
init_params.delegate = test_shell_delegate_;
init_params.context_factory = context_factory;
ash::Shell::CreateInstance(init_params);
aura::test::EnvTestHelper(aura::Env::GetInstance()).SetInputStateLookup(
scoped_ptr<aura::InputStateLookup>());
Shell* shell = Shell::GetInstance();
if (start_session) {
test_shell_delegate_->test_session_state_delegate()->
SetActiveUserSessionStarted(true);
test_shell_delegate_->test_session_state_delegate()->
SetHasActiveUser(true);
}
test::DisplayManagerTestApi(shell->display_manager()).
DisableChangeDisplayUponHostResize();
ShellTestApi(shell).DisableDisplayConfiguratorAnimation();
test_screenshot_delegate_ = new TestScreenshotDelegate();
shell->accelerator_controller()->SetScreenshotDelegate(
scoped_ptr<ScreenshotDelegate>(test_screenshot_delegate_));
}
void AshTestHelper::TearDown() {
// Tear down the shell.
Shell::DeleteInstance();
test_screenshot_delegate_ = NULL;
// Remove global message center state.
message_center::MessageCenter::Shutdown();
#if defined(OS_CHROMEOS)
chromeos::CrasAudioHandler::Shutdown();
if (dbus_thread_manager_initialized_) {
chromeos::DBusThreadManager::Shutdown();
dbus_thread_manager_initialized_ = false;
}
keyboard::ResetKeyboardForTesting();
#endif
aura::Env::DeleteInstance();
ui::TerminateContextFactoryForTests();
// Need to reset the initial login status.
TestSystemTrayDelegate::SetInitialLoginStatus(user::LOGGED_IN_USER);
ui::ShutdownInputMethodForTesting();
zero_duration_mode_.reset();
CHECK(!wm::ScopedCaptureClient::IsActive());
views_delegate_.reset();
}
void AshTestHelper::RunAllPendingInMessageLoop() {
DCHECK(base::MessageLoopForUI::current() == message_loop_);
aura::Env::CreateInstance(true);
base::RunLoop run_loop;
run_loop.RunUntilIdle();
}
aura::Window* AshTestHelper::CurrentContext() {
aura::Window* root_window = Shell::GetTargetRootWindow();
if (!root_window)
root_window = Shell::GetPrimaryRootWindow();
DCHECK(root_window);
return root_window;
}
// static
bool AshTestHelper::SupportsMultipleDisplays() {
#if defined(OS_WIN)
return false;
#else
return true;
#endif
}
// static
bool AshTestHelper::SupportsHostWindowResize() {
#if defined(OS_WIN)
return false;
#else
return true;
#endif
}
} // namespace test
} // namespace ash