ash/test/shell_test_api.cc

   1 // Copyright (c) 2012 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/shell_test_api.h"
   6
   7 #include "ash/root_window_controller.h"
   8 #include "ash/session/session_state_delegate.h"
   9 #include "ash/shelf/shelf_delegate.h"
  10 #include "ash/shell.h"
  11
  12 #if defined(OS_CHROMEOS)
  13 #include "ash/display/display_configurator_animation.h"
  14 #include "ui/display/chromeos/display_configurator.h"
  15 #endif
  16
  17 namespace ash {
  18 namespace test {
  19
  20 ShellTestApi::ShellTestApi(Shell* shell) : shell_(shell) {}
  21
  22 RootWindowLayoutManager* ShellTestApi::root_window_layout() {
  23   return shell_->GetPrimaryRootWindowController()->root_window_layout();
  24 }
  25
  26 SystemGestureEventFilter* ShellTestApi::system_gesture_event_filter() {
  27   return shell_->system_gesture_filter_.get();
  28 }
  29
  30 WorkspaceController* ShellTestApi::workspace_controller() {
  31   return shell_->GetPrimaryRootWindowController()->workspace_controller();
  32 }
  33
  34 ScreenPositionController* ShellTestApi::screen_position_controller() {
  35   return shell_->screen_position_controller_.get();
  36 }
  37
  38 AshNativeCursorManager* ShellTestApi::ash_native_cursor_manager() {
  39   return shell_->native_cursor_manager_;
  40 }
  41
  42 ShelfModel* ShellTestApi::shelf_model() {
  43   return shell_->shelf_model_.get();
  44 }
  45
  46 DragDropController* ShellTestApi::drag_drop_controller() {
  47   return shell_->drag_drop_controller_.get();
  48 }
  49
  50 AppListController* ShellTestApi::app_list_controller() {
  51   return shell_->app_list_controller_.get();
  52 }
  53
  54 void ShellTestApi::DisableDisplayConfiguratorAnimation() {
  55 #if defined(OS_CHROMEOS)
  56   if (shell_->display_configurator_animation_) {
  57     shell_->display_configurator_->RemoveObserver(
  58         shell_->display_configurator_animation_.get());
  59     shell_->display_configurator_animation_.reset();
  60   }
  61 #endif  // defined(OS_CHROMEOS)
  62 }
  63
  64 void ShellTestApi::SetShelfDelegate(ShelfDelegate* delegate) {
  65   shell_->shelf_delegate_.reset(delegate);
  66 }
  67
  68 void ShellTestApi::SetSessionStateDelegate(
  69     SessionStateDelegate* session_state_delegate) {
  70   shell_->session_state_delegate_.reset(session_state_delegate);
  71 }
  72
  73 }  // namespace test
  74 }  // namespace ash