| // 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/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/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/views/corewm/capture_controller.h" |
| |
| #if defined(OS_CHROMEOS) |
| #include "chromeos/audio/cras_audio_handler.h" |
| #endif |
| |
| #if defined(USE_X11) |
| #include "ui/aura/root_window_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(USE_X11) |
| aura::test::SetUseOverrideRedirectWindowByDefault(true); |
| #endif |
| } |
| |
| AshTestHelper::~AshTestHelper() { |
| } |
| |
| void AshTestHelper::SetUp(bool start_session) { |
| // Disable animations during tests. |
| zero_duration_mode_.reset(new ui::ScopedAnimationDurationScaleMode( |
| ui::ScopedAnimationDurationScaleMode::ZERO_DURATION)); |
| ui::InitializeInputMethodForTesting(); |
| |
| bool allow_test_contexts = true; |
| ui::InitializeContextFactoryForTests(allow_test_contexts); |
| |
| // Creates Shell and hook with Desktop. |
| 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 CrasAudioHandler for testing since g_browser_process is not |
| // created in AshTestBase tests. |
| chromeos::CrasAudioHandler::InitializeForTesting(); |
| #endif |
| ash::Shell::CreateInstance(test_shell_delegate_); |
| 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).DisableOutputConfiguratorAnimation(); |
| |
| 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(); |
| #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(!views::corewm::ScopedCaptureClient::IsActive()); |
| } |
| |
| void AshTestHelper::RunAllPendingInMessageLoop() { |
| DCHECK(base::MessageLoopForUI::current() == message_loop_); |
| aura::Env::CreateInstance(); |
| base::RunLoop run_loop(aura::Env::GetInstance()->GetDispatcher()); |
| 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; |
| } |
| |
| } // namespace test |
| } // namespace ash |