| // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "ash/focus_cycler.h" |
| |
| #include <memory> |
| |
| #include "ash/shelf/shelf.h" |
| #include "ash/shelf/shelf_widget.h" |
| #include "ash/shell.h" |
| #include "ash/system/status_area_widget.h" |
| #include "ash/system/status_area_widget_delegate.h" |
| #include "ash/system/tray/system_tray.h" |
| #include "ash/test/ash_test_base.h" |
| #include "ash/wm/window_util.h" |
| #include "ui/aura/test/test_windows.h" |
| #include "ui/aura/window.h" |
| #include "ui/aura/window_event_dispatcher.h" |
| #include "ui/events/test/event_generator.h" |
| #include "ui/views/accessible_pane_view.h" |
| #include "ui/views/controls/button/menu_button.h" |
| #include "ui/views/widget/widget.h" |
| |
| namespace ash { |
| |
| using aura::Window; |
| |
| namespace { |
| |
| StatusAreaWidgetDelegate* GetStatusAreaWidgetDelegate(views::Widget* widget) { |
| return static_cast<StatusAreaWidgetDelegate*>(widget->GetContentsView()); |
| } |
| |
| class PanedWidgetDelegate : public views::WidgetDelegate { |
| public: |
| PanedWidgetDelegate(views::Widget* widget) : widget_(widget) {} |
| |
| void SetAccessiblePanes(const std::vector<views::View*>& panes) { |
| accessible_panes_ = panes; |
| } |
| |
| // views::WidgetDelegate. |
| void GetAccessiblePanes(std::vector<views::View*>* panes) override { |
| std::copy(accessible_panes_.begin(), accessible_panes_.end(), |
| std::back_inserter(*panes)); |
| } |
| views::Widget* GetWidget() override { return widget_; }; |
| const views::Widget* GetWidget() const override { return widget_; } |
| |
| private: |
| views::Widget* widget_; |
| std::vector<views::View*> accessible_panes_; |
| }; |
| |
| } // namespace |
| |
| class FocusCyclerTest : public AshTestBase { |
| public: |
| FocusCyclerTest() = default; |
| |
| void SetUp() override { |
| AshTestBase::SetUp(); |
| |
| focus_cycler_.reset(new FocusCycler()); |
| } |
| |
| void TearDown() override { |
| GetStatusAreaWidgetDelegate(GetPrimarySystemTray()->GetWidget()) |
| ->SetFocusCyclerForTesting(nullptr); |
| |
| shelf_widget()->SetFocusCycler(nullptr); |
| |
| focus_cycler_.reset(); |
| |
| AshTestBase::TearDown(); |
| } |
| |
| protected: |
| // Setup the system tray focus cycler. |
| void SetUpTrayFocusCycle() { |
| views::Widget* system_tray_widget = GetPrimarySystemTray()->GetWidget(); |
| ASSERT_TRUE(system_tray_widget); |
| focus_cycler_->AddWidget(system_tray_widget); |
| GetStatusAreaWidgetDelegate(system_tray_widget) |
| ->SetFocusCyclerForTesting(focus_cycler()); |
| } |
| |
| FocusCycler* focus_cycler() { return focus_cycler_.get(); } |
| |
| ShelfWidget* shelf_widget() { return GetPrimaryShelf()->shelf_widget(); } |
| |
| void InstallFocusCycleOnShelf() { |
| // Add the shelf. |
| shelf_widget()->SetFocusCycler(focus_cycler()); |
| } |
| |
| private: |
| std::unique_ptr<FocusCycler> focus_cycler_; |
| |
| DISALLOW_COPY_AND_ASSIGN(FocusCyclerTest); |
| }; |
| |
| TEST_F(FocusCyclerTest, CycleFocusBrowserOnly) { |
| // Create a single test window. |
| std::unique_ptr<Window> window0(CreateTestWindowInShellWithId(0)); |
| wm::ActivateWindow(window0.get()); |
| EXPECT_TRUE(wm::IsActiveWindow(window0.get())); |
| |
| // Cycle the window |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(wm::IsActiveWindow(window0.get())); |
| } |
| |
| TEST_F(FocusCyclerTest, CycleFocusForward) { |
| SetUpTrayFocusCycle(); |
| |
| InstallFocusCycleOnShelf(); |
| |
| // Create a single test window. |
| std::unique_ptr<Window> window0(CreateTestWindowInShellWithId(0)); |
| wm::ActivateWindow(window0.get()); |
| EXPECT_TRUE(wm::IsActiveWindow(window0.get())); |
| |
| // Cycle focus to the status area. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(GetPrimarySystemTray()->GetWidget()->IsActive()); |
| |
| // Cycle focus to the shelf. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(shelf_widget()->IsActive()); |
| |
| // Cycle focus to the browser. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(wm::IsActiveWindow(window0.get())); |
| } |
| |
| TEST_F(FocusCyclerTest, CycleFocusBackward) { |
| SetUpTrayFocusCycle(); |
| |
| InstallFocusCycleOnShelf(); |
| |
| // Create a single test window. |
| std::unique_ptr<Window> window0(CreateTestWindowInShellWithId(0)); |
| wm::ActivateWindow(window0.get()); |
| EXPECT_TRUE(wm::IsActiveWindow(window0.get())); |
| |
| // Cycle focus to the shelf. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(shelf_widget()->IsActive()); |
| |
| // Cycle focus to the status area. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(GetPrimarySystemTray()->GetWidget()->IsActive()); |
| |
| // Cycle focus to the browser. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(wm::IsActiveWindow(window0.get())); |
| } |
| |
| TEST_F(FocusCyclerTest, CycleFocusForwardBackward) { |
| SetUpTrayFocusCycle(); |
| |
| InstallFocusCycleOnShelf(); |
| |
| // Create a single test window. |
| std::unique_ptr<Window> window0(CreateTestWindowInShellWithId(0)); |
| wm::ActivateWindow(window0.get()); |
| EXPECT_TRUE(wm::IsActiveWindow(window0.get())); |
| |
| // Cycle focus to the shelf. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(shelf_widget()->IsActive()); |
| |
| // Cycle focus to the status area. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(GetPrimarySystemTray()->GetWidget()->IsActive()); |
| |
| // Cycle focus to the browser. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(wm::IsActiveWindow(window0.get())); |
| |
| // Cycle focus to the status area. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(GetPrimarySystemTray()->GetWidget()->IsActive()); |
| |
| // Cycle focus to the shelf. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(shelf_widget()->IsActive()); |
| |
| // Cycle focus to the browser. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(wm::IsActiveWindow(window0.get())); |
| } |
| |
| TEST_F(FocusCyclerTest, CycleFocusNoBrowser) { |
| SetUpTrayFocusCycle(); |
| |
| InstallFocusCycleOnShelf(); |
| |
| // Add the shelf and focus it. |
| focus_cycler()->FocusWidget(shelf_widget()); |
| |
| // Cycle focus to the status area. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(GetPrimarySystemTray()->GetWidget()->IsActive()); |
| |
| // Cycle focus to the shelf. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(shelf_widget()->IsActive()); |
| |
| // Cycle focus to the status area. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(GetPrimarySystemTray()->GetWidget()->IsActive()); |
| |
| // Cycle focus to the shelf. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(shelf_widget()->IsActive()); |
| |
| // Cycle focus to the status area. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(GetPrimarySystemTray()->GetWidget()->IsActive()); |
| } |
| |
| // Tests that focus cycles from the active browser to the status area and back. |
| TEST_F(FocusCyclerTest, Shelf_CycleFocusForward) { |
| SetUpTrayFocusCycle(); |
| InstallFocusCycleOnShelf(); |
| shelf_widget()->Hide(); |
| |
| // Create two test windows. |
| std::unique_ptr<Window> window0(CreateTestWindowInShellWithId(0)); |
| std::unique_ptr<Window> window1(CreateTestWindowInShellWithId(1)); |
| wm::ActivateWindow(window1.get()); |
| wm::ActivateWindow(window0.get()); |
| EXPECT_TRUE(wm::IsActiveWindow(window0.get())); |
| |
| // Cycle focus to the status area. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(GetPrimarySystemTray()->GetWidget()->IsActive()); |
| |
| // Cycle focus to the browser. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(wm::IsActiveWindow(window0.get())); |
| |
| // Cycle focus to the status area. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(GetPrimarySystemTray()->GetWidget()->IsActive()); |
| } |
| |
| TEST_F(FocusCyclerTest, Shelf_CycleFocusBackwardInvisible) { |
| SetUpTrayFocusCycle(); |
| InstallFocusCycleOnShelf(); |
| shelf_widget()->Hide(); |
| |
| // Create a single test window. |
| std::unique_ptr<Window> window0(CreateTestWindowInShellWithId(0)); |
| wm::ActivateWindow(window0.get()); |
| EXPECT_TRUE(wm::IsActiveWindow(window0.get())); |
| |
| // Cycle focus to the status area. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(GetPrimarySystemTray()->GetWidget()->IsActive()); |
| |
| // Cycle focus to the browser. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(wm::IsActiveWindow(window0.get())); |
| } |
| |
| TEST_F(FocusCyclerTest, CycleFocusThroughWindowWithPanes) { |
| SetUpTrayFocusCycle(); |
| |
| InstallFocusCycleOnShelf(); |
| |
| std::unique_ptr<PanedWidgetDelegate> test_widget_delegate; |
| std::unique_ptr<views::Widget> browser_widget(new views::Widget); |
| test_widget_delegate.reset(new PanedWidgetDelegate(browser_widget.get())); |
| views::Widget::InitParams widget_params( |
| views::Widget::InitParams::TYPE_WINDOW); |
| widget_params.context = CurrentContext(); |
| widget_params.delegate = test_widget_delegate.get(); |
| widget_params.ownership = |
| views::Widget::InitParams::WIDGET_OWNS_NATIVE_WIDGET; |
| browser_widget->Init(widget_params); |
| browser_widget->Show(); |
| |
| aura::Window* browser_window = browser_widget->GetNativeView(); |
| |
| views::View* root_view = browser_widget->GetRootView(); |
| |
| views::AccessiblePaneView* pane1 = new views::AccessiblePaneView(); |
| root_view->AddChildView(pane1); |
| |
| views::View* view1 = new views::View; |
| view1->SetFocusBehavior(views::View::FocusBehavior::ALWAYS); |
| pane1->AddChildView(view1); |
| |
| views::View* view2 = new views::View; |
| view2->SetFocusBehavior(views::View::FocusBehavior::ALWAYS); |
| pane1->AddChildView(view2); |
| |
| views::AccessiblePaneView* pane2 = new views::AccessiblePaneView(); |
| root_view->AddChildView(pane2); |
| |
| views::View* view3 = new views::View; |
| view3->SetFocusBehavior(views::View::FocusBehavior::ALWAYS); |
| pane2->AddChildView(view3); |
| |
| views::View* view4 = new views::View; |
| view4->SetFocusBehavior(views::View::FocusBehavior::ALWAYS); |
| pane2->AddChildView(view4); |
| |
| std::vector<views::View*> panes; |
| panes.push_back(pane1); |
| panes.push_back(pane2); |
| |
| test_widget_delegate->SetAccessiblePanes(panes); |
| |
| views::FocusManager* focus_manager = browser_widget->GetFocusManager(); |
| |
| // Cycle focus to the status area. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(GetPrimarySystemTray()->GetWidget()->IsActive()); |
| |
| // Cycle focus to the shelf. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(shelf_widget()->IsActive()); |
| |
| // Cycle focus to the first pane in the browser. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(wm::IsActiveWindow(browser_window)); |
| EXPECT_EQ(focus_manager->GetFocusedView(), view1); |
| |
| // Cycle focus to the second pane in the browser. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(wm::IsActiveWindow(browser_window)); |
| EXPECT_EQ(focus_manager->GetFocusedView(), view3); |
| |
| // Cycle focus back to the status area. |
| focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(GetPrimarySystemTray()->GetWidget()->IsActive()); |
| |
| // Reverse direction - back to the second pane in the browser. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(wm::IsActiveWindow(browser_window)); |
| EXPECT_EQ(focus_manager->GetFocusedView(), view3); |
| |
| // Back to the first pane in the browser. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(wm::IsActiveWindow(browser_window)); |
| EXPECT_EQ(focus_manager->GetFocusedView(), view1); |
| |
| // Back to the shelf. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(shelf_widget()->IsActive()); |
| |
| // Back to the status area. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(GetPrimarySystemTray()->GetWidget()->IsActive()); |
| |
| // Pressing "Escape" while on the status area should |
| // deactivate it, and activate the browser window. |
| ui::test::EventGenerator& event_generator = GetEventGenerator(); |
| event_generator.PressKey(ui::VKEY_ESCAPE, 0); |
| EXPECT_TRUE(wm::IsActiveWindow(browser_window)); |
| EXPECT_EQ(focus_manager->GetFocusedView(), view1); |
| |
| // Similarly, pressing "Escape" while on the shelf. |
| // should do the same thing. |
| focus_cycler()->RotateFocus(FocusCycler::BACKWARD); |
| EXPECT_TRUE(shelf_widget()->IsActive()); |
| event_generator.PressKey(ui::VKEY_ESCAPE, 0); |
| EXPECT_TRUE(wm::IsActiveWindow(browser_window)); |
| EXPECT_EQ(focus_manager->GetFocusedView(), view1); |
| } |
| |
| // Test that when the shelf widget & status area widget are removed, they should |
| // also be removed from focus cycler. |
| TEST_F(FocusCyclerTest, RemoveWidgetOnDisplayRemoved) { |
| // Two displays are added, so two shelf widgets and two status area widgets |
| // are added to focus cycler. |
| UpdateDisplay("800x800, 500x500"); |
| // Remove one display. Its shelf widget and status area widget should also be |
| // removed from focus cycler. |
| UpdateDisplay("800x800"); |
| |
| // Create a single test window. |
| std::unique_ptr<Window> window(CreateTestWindowInShellWithId(0)); |
| wm::ActivateWindow(window.get()); |
| EXPECT_TRUE(wm::IsActiveWindow(window.get())); |
| |
| // Cycle focus to the status area. |
| Shell::Get()->focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_FALSE(wm::IsActiveWindow(window.get())); |
| |
| // Cycle focus to the shelf. |
| Shell::Get()->focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| |
| // Cycle focus should go back to the browser. |
| Shell::Get()->focus_cycler()->RotateFocus(FocusCycler::FORWARD); |
| EXPECT_TRUE(wm::IsActiveWindow(window.get())); |
| } |
| |
| } // namespace ash |