ash/focus_cycler_unittest.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/focus_cycler.h"
   6
   7 #include "ash/launcher/launcher.h"
   8 #include "ash/root_window_controller.h"
   9 #include "ash/shell.h"
  10 #include "ash/shell_window_ids.h"
  11 #include "ash/system/status_area_widget.h"
  12 #include "ash/system/status_area_widget_delegate.h"
  13 #include "ash/system/tray/system_tray.h"
  14 #include "ash/wm/window_util.h"
  15 #include "ash/test/ash_test_base.h"
  16 #include "ash/shell_factory.h"
  17 #include "ui/aura/test/test_windows.h"
  18 #include "ui/aura/window.h"
  19 #include "ui/views/controls/button/menu_button.h"
  20 #include "ui/views/widget/widget.h"
  21
  22 namespace ash {
  23 namespace test {
  24
  25 using aura::test::CreateTestWindowWithId;
  26 using aura::Window;
  27 using internal::FocusCycler;
  28
  29 namespace {
  30
  31 internal::StatusAreaWidgetDelegate* GetStatusAreaWidgetDelegate(
  32     views::Widget* widget) {
  33   return static_cast<internal::StatusAreaWidgetDelegate*>(
  34       widget->GetContentsView());
  35 }
  36
  37 }  // namespace
  38
  39 class FocusCyclerTest : public AshTestBase {
  40  public:
  41   FocusCyclerTest() {}
  42
  43   virtual void SetUp() OVERRIDE {
  44     AshTestBase::SetUp();
  45
  46     focus_cycler_.reset(new FocusCycler());
  47
  48     ASSERT_TRUE(Launcher::ForPrimaryDisplay());
  49   }
  50
  51   virtual void TearDown() OVERRIDE {
  52     if (tray_.get()) {
  53       GetStatusAreaWidgetDelegate(tray_->GetWidget())->
  54           SetFocusCyclerForTesting(NULL);
  55       tray_.reset();
  56     }
  57
  58     Launcher::ForPrimaryDisplay()->SetFocusCycler(NULL);
  59
  60     focus_cycler_.reset();
  61
  62     AshTestBase::TearDown();
  63   }
  64
  65  protected:
  66   // Creates the system tray, returning true on success.
  67   bool CreateTray() {
  68     if (tray_.get())
  69       return false;
  70     aura::Window* parent = Shell::GetPrimaryRootWindowController()->
  71         GetContainer(ash::internal::kShellWindowId_StatusContainer);
  72
  73     internal::StatusAreaWidget* widget = new internal::StatusAreaWidget(parent);
  74     widget->CreateTrayViews(NULL);
  75     widget->Show();
  76     tray_.reset(widget->system_tray());
  77     if (!tray_->GetWidget())
  78       return false;
  79     focus_cycler_->AddWidget(tray()->GetWidget());
  80     GetStatusAreaWidgetDelegate(tray_->GetWidget())->SetFocusCyclerForTesting(
  81         focus_cycler());
  82     return true;
  83   }
  84
  85   FocusCycler* focus_cycler() { return focus_cycler_.get(); }
  86
  87   SystemTray* tray() { return tray_.get(); }
  88
  89   views::Widget* launcher_widget() {
  90     return Launcher::ForPrimaryDisplay()->widget();
  91   }
  92
  93   void InstallFocusCycleOnLauncher() {
  94     // Add the launcher
  95     Launcher* launcher = Launcher::ForPrimaryDisplay();
  96     launcher->SetFocusCycler(focus_cycler());
  97   }
  98
  99  private:
 100   scoped_ptr<FocusCycler> focus_cycler_;
 101   scoped_ptr<SystemTray> tray_;
 102
 103   DISALLOW_COPY_AND_ASSIGN(FocusCyclerTest);
 104 };
 105
 106 TEST_F(FocusCyclerTest, CycleFocusBrowserOnly) {
 107   // Create a single test window.
 108   scoped_ptr<Window> window0(CreateTestWindowWithId(0, NULL));
 109   wm::ActivateWindow(window0.get());
 110   EXPECT_TRUE(wm::IsActiveWindow(window0.get()));
 111
 112   // Cycle the window
 113   focus_cycler()->RotateFocus(FocusCycler::FORWARD);
 114   EXPECT_TRUE(wm::IsActiveWindow(window0.get()));
 115 }
 116
 117 TEST_F(FocusCyclerTest, CycleFocusForward) {
 118   ASSERT_TRUE(CreateTray());
 119
 120   InstallFocusCycleOnLauncher();
 121
 122   // Create a single test window.
 123   scoped_ptr<Window> window0(CreateTestWindowWithId(0, NULL));
 124   wm::ActivateWindow(window0.get());
 125   EXPECT_TRUE(wm::IsActiveWindow(window0.get()));
 126
 127   // Cycle focus to the status area
 128   focus_cycler()->RotateFocus(FocusCycler::FORWARD);
 129   EXPECT_TRUE(tray()->GetWidget()->IsActive());
 130
 131   // Cycle focus to the launcher
 132   focus_cycler()->RotateFocus(FocusCycler::FORWARD);
 133   EXPECT_TRUE(launcher_widget()->IsActive());
 134
 135   // Cycle focus to the browser
 136   focus_cycler()->RotateFocus(FocusCycler::FORWARD);
 137   EXPECT_TRUE(wm::IsActiveWindow(window0.get()));
 138 }
 139
 140 TEST_F(FocusCyclerTest, CycleFocusBackward) {
 141   ASSERT_TRUE(CreateTray());
 142
 143   InstallFocusCycleOnLauncher();
 144
 145   // Create a single test window.
 146   scoped_ptr<Window> window0(CreateTestWindowWithId(0, NULL));
 147   wm::ActivateWindow(window0.get());
 148   EXPECT_TRUE(wm::IsActiveWindow(window0.get()));
 149
 150   // Cycle focus to the launcher
 151   focus_cycler()->RotateFocus(FocusCycler::BACKWARD);
 152   EXPECT_TRUE(launcher_widget()->IsActive());
 153
 154   // Cycle focus to the status area
 155   focus_cycler()->RotateFocus(FocusCycler::BACKWARD);
 156   EXPECT_TRUE(tray()->GetWidget()->IsActive());
 157
 158   // Cycle focus to the browser
 159   focus_cycler()->RotateFocus(FocusCycler::BACKWARD);
 160   EXPECT_TRUE(wm::IsActiveWindow(window0.get()));
 161 }
 162
 163 TEST_F(FocusCyclerTest, CycleFocusForwardBackward) {
 164   ASSERT_TRUE(CreateTray());
 165
 166   InstallFocusCycleOnLauncher();
 167
 168   // Create a single test window.
 169   scoped_ptr<Window> window0(CreateTestWindowWithId(0, NULL));
 170   wm::ActivateWindow(window0.get());
 171   EXPECT_TRUE(wm::IsActiveWindow(window0.get()));
 172
 173   // Cycle focus to the launcher
 174   focus_cycler()->RotateFocus(FocusCycler::BACKWARD);
 175   EXPECT_TRUE(launcher_widget()->IsActive());
 176
 177   // Cycle focus to the status area
 178   focus_cycler()->RotateFocus(FocusCycler::BACKWARD);
 179   EXPECT_TRUE(tray()->GetWidget()->IsActive());
 180
 181   // Cycle focus to the browser
 182   focus_cycler()->RotateFocus(FocusCycler::BACKWARD);
 183   EXPECT_TRUE(wm::IsActiveWindow(window0.get()));
 184
 185   // Cycle focus to the status area
 186   focus_cycler()->RotateFocus(FocusCycler::FORWARD);
 187   EXPECT_TRUE(tray()->GetWidget()->IsActive());
 188
 189   // Cycle focus to the launcher
 190   focus_cycler()->RotateFocus(FocusCycler::FORWARD);
 191   EXPECT_TRUE(launcher_widget()->IsActive());
 192
 193   // Cycle focus to the browser
 194   focus_cycler()->RotateFocus(FocusCycler::FORWARD);
 195   EXPECT_TRUE(wm::IsActiveWindow(window0.get()));
 196 }
 197
 198 TEST_F(FocusCyclerTest, CycleFocusNoBrowser) {
 199   ASSERT_TRUE(CreateTray());
 200
 201   InstallFocusCycleOnLauncher();
 202
 203   // Add the launcher and focus it
 204   focus_cycler()->FocusWidget(launcher_widget());
 205
 206   // Cycle focus to the status area
 207   focus_cycler()->RotateFocus(FocusCycler::FORWARD);
 208   EXPECT_TRUE(tray()->GetWidget()->IsActive());
 209
 210   // Cycle focus to the launcher
 211   focus_cycler()->RotateFocus(FocusCycler::FORWARD);
 212   EXPECT_TRUE(launcher_widget()->IsActive());
 213
 214   // Cycle focus to the status area
 215   focus_cycler()->RotateFocus(FocusCycler::FORWARD);
 216   EXPECT_TRUE(tray()->GetWidget()->IsActive());
 217
 218   // Cycle focus to the launcher
 219   focus_cycler()->RotateFocus(FocusCycler::BACKWARD);
 220   EXPECT_TRUE(launcher_widget()->IsActive());
 221
 222   // Cycle focus to the status area
 223   focus_cycler()->RotateFocus(FocusCycler::BACKWARD);
 224   EXPECT_TRUE(tray()->GetWidget()->IsActive());
 225 }
 226
 227 TEST_F(FocusCyclerTest, Launcher_CycleFocusForward) {
 228   ASSERT_TRUE(CreateTray());
 229   InstallFocusCycleOnLauncher();
 230   launcher_widget()->Hide();
 231
 232   // Create a single test window.
 233   scoped_ptr<Window> window0(CreateTestWindowWithId(0, NULL));
 234   wm::ActivateWindow(window0.get());
 235   EXPECT_TRUE(wm::IsActiveWindow(window0.get()));
 236
 237   // Cycle focus to the status area
 238   focus_cycler()->RotateFocus(FocusCycler::FORWARD);
 239   EXPECT_TRUE(tray()->GetWidget()->IsActive());
 240
 241   // Cycle focus to the browser
 242   focus_cycler()->RotateFocus(FocusCycler::FORWARD);
 243   EXPECT_TRUE(wm::IsActiveWindow(window0.get()));
 244 }
 245
 246 TEST_F(FocusCyclerTest, Launcher_CycleFocusBackwardInvisible) {
 247   ASSERT_TRUE(CreateTray());
 248   InstallFocusCycleOnLauncher();
 249   launcher_widget()->Hide();
 250
 251   // Create a single test window.
 252   scoped_ptr<Window> window0(CreateTestWindowWithId(0, NULL));
 253   wm::ActivateWindow(window0.get());
 254   EXPECT_TRUE(wm::IsActiveWindow(window0.get()));
 255
 256   // Cycle focus to the status area
 257   focus_cycler()->RotateFocus(FocusCycler::BACKWARD);
 258   EXPECT_TRUE(tray()->GetWidget()->IsActive());
 259
 260   // Cycle focus to the browser
 261   focus_cycler()->RotateFocus(FocusCycler::BACKWARD);
 262   EXPECT_TRUE(wm::IsActiveWindow(window0.get()));
 263 }
 264
 265 }  // namespace test
 266 }  // namespace ash