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.
5 #include "ash/test/ash_test_helper.h"
7 #include "ash/accelerators/accelerator_controller.h"
8 #include "ash/ash_switches.h"
10 #include "ash/test/display_manager_test_api.h"
11 #include "ash/test/shell_test_api.h"
12 #include "ash/test/test_screenshot_delegate.h"
13 #include "ash/test/test_session_state_delegate.h"
14 #include "ash/test/test_shell_delegate.h"
15 #include "ash/test/test_system_tray_delegate.h"
16 #include "base/run_loop.h"
17 #include "ui/aura/env.h"
18 #include "ui/aura/input_state_lookup.h"
19 #include "ui/aura/test/env_test_helper.h"
20 #include "ui/base/ime/input_method_initializer.h"
21 #include "ui/compositor/scoped_animation_duration_scale_mode.h"
22 #include "ui/compositor/test/context_factories_for_test.h"
23 #include "ui/message_center/message_center.h"
24 #include "ui/views/corewm/capture_controller.h"
26 #if defined(OS_CHROMEOS)
27 #include "chromeos/audio/cras_audio_handler.h"
28 #include "chromeos/network/network_handler.h"
32 #include "ui/aura/root_window_host_x11.h"
38 AshTestHelper::AshTestHelper(base::MessageLoopForUI* message_loop)
39 : message_loop_(message_loop),
40 test_shell_delegate_(NULL),
41 test_screenshot_delegate_(NULL),
42 tear_down_network_handler_(false) {
45 aura::test::SetUseOverrideRedirectWindowByDefault(true);
49 AshTestHelper::~AshTestHelper() {
52 void AshTestHelper::SetUp(bool start_session) {
53 // Disable animations during tests.
54 zero_duration_mode_.reset(new ui::ScopedAnimationDurationScaleMode(
55 ui::ScopedAnimationDurationScaleMode::ZERO_DURATION));
56 ui::InitializeInputMethodForTesting();
58 bool allow_test_contexts = true;
59 ui::InitializeContextFactoryForTests(allow_test_contexts);
61 // Creates Shell and hook with Desktop.
62 test_shell_delegate_ = new TestShellDelegate;
64 // Creates MessageCenter since g_browser_process is not created in AshTestBase
66 message_center::MessageCenter::Initialize();
68 #if defined(OS_CHROMEOS)
69 // Create CrasAudioHandler for testing since g_browser_process is not
70 // created in AshTestBase tests.
71 chromeos::CrasAudioHandler::InitializeForTesting();
73 // Some tests may not initialize NetworkHandler. Initialize it here if that
75 if (!chromeos::NetworkHandler::IsInitialized()) {
76 tear_down_network_handler_ = true;
77 chromeos::NetworkHandler::Initialize();
80 RunAllPendingInMessageLoop();
82 ash::Shell::CreateInstance(test_shell_delegate_);
83 aura::test::EnvTestHelper(aura::Env::GetInstance()).SetInputStateLookup(
84 scoped_ptr<aura::InputStateLookup>());
86 Shell* shell = Shell::GetInstance();
88 test_shell_delegate_->test_session_state_delegate()->
89 SetActiveUserSessionStarted(true);
90 test_shell_delegate_->test_session_state_delegate()->
91 SetHasActiveUser(true);
94 test::DisplayManagerTestApi(shell->display_manager()).
95 DisableChangeDisplayUponHostResize();
96 ShellTestApi(shell).DisableOutputConfiguratorAnimation();
98 test_screenshot_delegate_ = new TestScreenshotDelegate();
99 shell->accelerator_controller()->SetScreenshotDelegate(
100 scoped_ptr<ScreenshotDelegate>(test_screenshot_delegate_));
103 void AshTestHelper::TearDown() {
104 // Tear down the shell.
105 Shell::DeleteInstance();
106 test_screenshot_delegate_ = NULL;
108 // Remove global message center state.
109 message_center::MessageCenter::Shutdown();
111 #if defined(OS_CHROMEOS)
112 if (tear_down_network_handler_ && chromeos::NetworkHandler::IsInitialized())
113 chromeos::NetworkHandler::Shutdown();
114 chromeos::CrasAudioHandler::Shutdown();
117 aura::Env::DeleteInstance();
118 ui::TerminateContextFactoryForTests();
120 // Need to reset the initial login status.
121 TestSystemTrayDelegate::SetInitialLoginStatus(user::LOGGED_IN_USER);
123 ui::ShutdownInputMethodForTesting();
124 zero_duration_mode_.reset();
126 CHECK(!views::corewm::ScopedCaptureClient::IsActive());
129 void AshTestHelper::RunAllPendingInMessageLoop() {
130 DCHECK(base::MessageLoopForUI::current() == message_loop_);
131 aura::Env::CreateInstance();
132 base::RunLoop run_loop(aura::Env::GetInstance()->GetDispatcher());
133 run_loop.RunUntilIdle();
136 aura::Window* AshTestHelper::CurrentContext() {
137 aura::Window* root_window = Shell::GetTargetRootWindow();
139 root_window = Shell::GetPrimaryRootWindow();