ash/test/ash_test_helper.cc

   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.
   4
   5 #include "ash/test/ash_test_helper.h"
   6
   7 #include "ash/accelerators/accelerator_controller.h"
   8 #include "ash/ash_switches.h"
   9 #include "ash/shell.h"
  10 #include "ash/shell_init_params.h"
  11 #include "ash/test/ash_test_views_delegate.h"
  12 #include "ash/test/display_manager_test_api.h"
  13 #include "ash/test/shell_test_api.h"
  14 #include "ash/test/test_screenshot_delegate.h"
  15 #include "ash/test/test_session_state_delegate.h"
  16 #include "ash/test/test_shell_delegate.h"
  17 #include "ash/test/test_system_tray_delegate.h"
  18 #include "base/run_loop.h"
  19 #include "ui/aura/env.h"
  20 #include "ui/aura/input_state_lookup.h"
  21 #include "ui/aura/test/env_test_helper.h"
  22 #include "ui/aura/test/event_generator_delegate_aura.h"
  23 #include "ui/base/ime/input_method_initializer.h"
  24 #include "ui/compositor/scoped_animation_duration_scale_mode.h"
  25 #include "ui/compositor/test/context_factories_for_test.h"
  26 #include "ui/message_center/message_center.h"
  27 #include "ui/wm/core/capture_controller.h"
  28
  29 #if defined(OS_CHROMEOS)
  30 #include "chromeos/audio/cras_audio_handler.h"
  31 #include "chromeos/dbus/dbus_thread_manager.h"
  32 #include "ui/keyboard/keyboard.h"
  33 #endif
  34
  35 #if defined(OS_WIN)
  36 #include "base/win/windows_version.h"
  37 #endif
  38
  39 #if defined(USE_X11)
  40 #include "ui/aura/window_tree_host_x11.h"
  41 #endif
  42
  43 namespace ash {
  44 namespace test {
  45
  46 AshTestHelper::AshTestHelper(base::MessageLoopForUI* message_loop)
  47     : message_loop_(message_loop),
  48       test_shell_delegate_(NULL),
  49       test_screenshot_delegate_(NULL) {
  50   CHECK(message_loop_);
  51 #if defined(OS_CHROMEOS)
  52   dbus_thread_manager_initialized_ = false;
  53 #endif
  54 #if defined(USE_X11)
  55   aura::test::SetUseOverrideRedirectWindowByDefault(true);
  56 #endif
  57   aura::test::InitializeAuraEventGeneratorDelegate();
  58 }
  59
  60 AshTestHelper::~AshTestHelper() {
  61 }
  62
  63 void AshTestHelper::SetUp(bool start_session) {
  64   views_delegate_.reset(new AshTestViewsDelegate);
  65
  66   // Disable animations during tests.
  67   zero_duration_mode_.reset(new ui::ScopedAnimationDurationScaleMode(
  68       ui::ScopedAnimationDurationScaleMode::ZERO_DURATION));
  69   ui::InitializeInputMethodForTesting();
  70
  71   bool enable_pixel_output = false;
  72   ui::ContextFactory* context_factory =
  73       ui::InitializeContextFactoryForTests(enable_pixel_output);
  74
  75   // Creates Shell and hook with Desktop.
  76   if (!test_shell_delegate_)
  77     test_shell_delegate_ = new TestShellDelegate;
  78
  79   // Creates MessageCenter since g_browser_process is not created in AshTestBase
  80   // tests.
  81   message_center::MessageCenter::Initialize();
  82
  83 #if defined(OS_CHROMEOS)
  84   // Create DBusThreadManager for testing.
  85   if (!chromeos::DBusThreadManager::IsInitialized()) {
  86     chromeos::DBusThreadManager::Initialize();
  87     dbus_thread_manager_initialized_ = true;
  88   }
  89   // Create CrasAudioHandler for testing since g_browser_process is not
  90   // created in AshTestBase tests.
  91   chromeos::CrasAudioHandler::InitializeForTesting();
  92 #endif
  93   ShellInitParams init_params;
  94   init_params.delegate = test_shell_delegate_;
  95   init_params.context_factory = context_factory;
  96   ash::Shell::CreateInstance(init_params);
  97   aura::test::EnvTestHelper(aura::Env::GetInstance()).SetInputStateLookup(
  98       scoped_ptr<aura::InputStateLookup>());
  99
 100   Shell* shell = Shell::GetInstance();
 101   if (start_session) {
 102     GetTestSessionStateDelegate()->SetActiveUserSessionStarted(true);
 103     GetTestSessionStateDelegate()->SetHasActiveUser(true);
 104   }
 105
 106   test::DisplayManagerTestApi(shell->display_manager()).
 107       DisableChangeDisplayUponHostResize();
 108   ShellTestApi(shell).DisableDisplayConfiguratorAnimation();
 109
 110   test_screenshot_delegate_ = new TestScreenshotDelegate();
 111   shell->accelerator_controller()->SetScreenshotDelegate(
 112       scoped_ptr<ScreenshotDelegate>(test_screenshot_delegate_));
 113 }
 114
 115 void AshTestHelper::TearDown() {
 116   // Tear down the shell.
 117   Shell::DeleteInstance();
 118   test_screenshot_delegate_ = NULL;
 119
 120   // Remove global message center state.
 121   message_center::MessageCenter::Shutdown();
 122
 123 #if defined(OS_CHROMEOS)
 124   chromeos::CrasAudioHandler::Shutdown();
 125   if (dbus_thread_manager_initialized_) {
 126     chromeos::DBusThreadManager::Shutdown();
 127     dbus_thread_manager_initialized_ = false;
 128   }
 129   keyboard::ResetKeyboardForTesting();
 130 #endif
 131
 132   aura::Env::DeleteInstance();
 133   ui::TerminateContextFactoryForTests();
 134
 135   // Need to reset the initial login status.
 136   TestSystemTrayDelegate::SetInitialLoginStatus(user::LOGGED_IN_USER);
 137
 138   ui::ShutdownInputMethodForTesting();
 139   zero_duration_mode_.reset();
 140
 141   CHECK(!wm::ScopedCaptureClient::IsActive());
 142
 143   views_delegate_.reset();
 144 }
 145
 146 void AshTestHelper::RunAllPendingInMessageLoop() {
 147   DCHECK(base::MessageLoopForUI::current() == message_loop_);
 148   aura::Env::CreateInstance(true);
 149   base::RunLoop run_loop;
 150   run_loop.RunUntilIdle();
 151 }
 152
 153 // static
 154 TestSessionStateDelegate* AshTestHelper::GetTestSessionStateDelegate() {
 155   CHECK(Shell::HasInstance());
 156   return static_cast<TestSessionStateDelegate*>(
 157       Shell::GetInstance()->session_state_delegate());
 158 }
 159
 160 aura::Window* AshTestHelper::CurrentContext() {
 161   aura::Window* root_window = Shell::GetTargetRootWindow();
 162   if (!root_window)
 163     root_window = Shell::GetPrimaryRootWindow();
 164   DCHECK(root_window);
 165   return root_window;
 166 }
 167
 168 // static
 169 bool AshTestHelper::SupportsMultipleDisplays() {
 170 #if defined(OS_WIN)
 171   return false;
 172 #else
 173   return true;
 174 #endif
 175 }
 176
 177 // static
 178 bool AshTestHelper::SupportsHostWindowResize() {
 179 #if defined(OS_WIN)
 180   return false;
 181 #else
 182   return true;
 183 #endif
 184 }
 185
 186 }  // namespace test
 187 }  // namespace ash