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