src/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/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"
  25 #include "ui/views/corewm/wm_state.h"
  26
  27 #if defined(OS_CHROMEOS)
  28 #include "chromeos/audio/cras_audio_handler.h"
  29 #include "chromeos/dbus/dbus_thread_manager.h"
  30 #include "ui/keyboard/keyboard.h"
  31 #endif
  32
  33 #if defined(USE_X11)
  34 #include "ui/aura/window_tree_host_x11.h"
  35 #endif
  36
  37 namespace ash {
  38 namespace test {
  39
  40 AshTestHelper::AshTestHelper(base::MessageLoopForUI* message_loop)
  41     : message_loop_(message_loop),
  42       test_shell_delegate_(NULL),
  43       test_screenshot_delegate_(NULL),
  44       dbus_thread_manager_initialized_(false) {
  45   CHECK(message_loop_);
  46 #if defined(USE_X11)
  47   aura::test::SetUseOverrideRedirectWindowByDefault(true);
  48 #endif
  49 }
  50
  51 AshTestHelper::~AshTestHelper() {
  52 }
  53
  54 void AshTestHelper::SetUp(bool start_session) {
  55   wm_state_.reset(new views::corewm::WMState);
  56
  57   // Disable animations during tests.
  58   zero_duration_mode_.reset(new ui::ScopedAnimationDurationScaleMode(
  59       ui::ScopedAnimationDurationScaleMode::ZERO_DURATION));
  60   ui::InitializeInputMethodForTesting();
  61
  62   bool allow_test_contexts = true;
  63   ui::InitializeContextFactoryForTests(allow_test_contexts);
  64
  65   // Creates Shell and hook with Desktop.
  66   test_shell_delegate_ = new TestShellDelegate;
  67
  68   // Creates MessageCenter since g_browser_process is not created in AshTestBase
  69   // tests.
  70   message_center::MessageCenter::Initialize();
  71
  72 #if defined(OS_CHROMEOS)
  73   // Create DBusThreadManager for testing.
  74   if (!chromeos::DBusThreadManager::IsInitialized()) {
  75     chromeos::DBusThreadManager::InitializeWithStub();
  76     dbus_thread_manager_initialized_ = true;
  77   }
  78   // Create CrasAudioHandler for testing since g_browser_process is not
  79   // created in AshTestBase tests.
  80   chromeos::CrasAudioHandler::InitializeForTesting();
  81 #endif
  82   ash::Shell::CreateInstance(test_shell_delegate_);
  83   aura::test::EnvTestHelper(aura::Env::GetInstance()).SetInputStateLookup(
  84       scoped_ptr<aura::InputStateLookup>());
  85
  86   Shell* shell = Shell::GetInstance();
  87   if (start_session) {
  88     test_shell_delegate_->test_session_state_delegate()->
  89         SetActiveUserSessionStarted(true);
  90     test_shell_delegate_->test_session_state_delegate()->
  91         SetHasActiveUser(true);
  92   }
  93
  94   test::DisplayManagerTestApi(shell->display_manager()).
  95       DisableChangeDisplayUponHostResize();
  96   ShellTestApi(shell).DisableOutputConfiguratorAnimation();
  97
  98   test_screenshot_delegate_ = new TestScreenshotDelegate();
  99   shell->accelerator_controller()->SetScreenshotDelegate(
 100       scoped_ptr<ScreenshotDelegate>(test_screenshot_delegate_));
 101 }
 102
 103 void AshTestHelper::TearDown() {
 104   // Tear down the shell.
 105   Shell::DeleteInstance();
 106   test_screenshot_delegate_ = NULL;
 107
 108   // Remove global message center state.
 109   message_center::MessageCenter::Shutdown();
 110
 111 #if defined(OS_CHROMEOS)
 112   chromeos::CrasAudioHandler::Shutdown();
 113   if (dbus_thread_manager_initialized_) {
 114     chromeos::DBusThreadManager::Shutdown();
 115     dbus_thread_manager_initialized_ = false;
 116   }
 117   keyboard::ResetKeyboardForTesting();
 118 #endif
 119
 120   aura::Env::DeleteInstance();
 121   ui::TerminateContextFactoryForTests();
 122
 123   // Need to reset the initial login status.
 124   TestSystemTrayDelegate::SetInitialLoginStatus(user::LOGGED_IN_USER);
 125
 126   ui::ShutdownInputMethodForTesting();
 127   zero_duration_mode_.reset();
 128
 129   CHECK(!views::corewm::ScopedCaptureClient::IsActive());
 130
 131   wm_state_.reset();
 132 }
 133
 134 void AshTestHelper::RunAllPendingInMessageLoop() {
 135   DCHECK(base::MessageLoopForUI::current() == message_loop_);
 136   aura::Env::CreateInstance();
 137   base::RunLoop run_loop;
 138   run_loop.RunUntilIdle();
 139 }
 140
 141 aura::Window* AshTestHelper::CurrentContext() {
 142   aura::Window* root_window = Shell::GetTargetRootWindow();
 143   if (!root_window)
 144     root_window = Shell::GetPrimaryRootWindow();
 145   DCHECK(root_window);
 146   return root_window;
 147 }
 148
 149 }  // namespace test
 150 }  // namespace ash