blob: 3ab72cce2c23a42a1b04bef383ea07b4341f0cf2 [file] [log] [blame]
// 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/common/focus_cycler.h"
#include <memory>
#include "ash/common/system/status_area_widget_delegate.h"
#include "ash/common/wm_shell.h"
#include "ash/shelf/shelf.h"
#include "ash/shelf/shelf_widget.h"
#include "ash/system/status_area_widget.h"
#include "ash/system/tray/system_tray.h"
#include "ash/test/ash_test_base.h"
#include "ash/test/status_area_widget_test_helper.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 {
namespace test {
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
// TODO(jamescook): Migrate this test to //ash/common after the status area
// widget moves. http://crbug.com/620955
class FocusCyclerTest : public AshTestBase {
public:
FocusCyclerTest() {}
void SetUp() override {
AshTestBase::SetUp();
focus_cycler_.reset(new FocusCycler());
ASSERT_TRUE(Shelf::ForPrimaryDisplay());
}
void TearDown() override {
GetStatusAreaWidgetDelegate(GetPrimarySystemTray()->GetWidget())
->SetFocusCyclerForTesting(nullptr);
shelf_widget()->SetFocusCycler(nullptr);
focus_cycler_.reset();
AshTestBase::TearDown();
}
protected:
// Setup the system tray using StatusAreaWidgetTestHelper and focus_cycler.
void SetUpTrayFocusCycle() {
StatusAreaWidget* widget =
StatusAreaWidgetTestHelper::GetStatusAreaWidget();
widget->CreateTrayViews();
widget->Show();
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 Shelf::ForPrimaryDisplay()->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.
WmShell::Get()->focus_cycler()->RotateFocus(FocusCycler::FORWARD);
EXPECT_FALSE(wm::IsActiveWindow(window.get()));
// Cycle focus to the shelf.
WmShell::Get()->focus_cycler()->RotateFocus(FocusCycler::FORWARD);
// Cycle focus should go back to the browser.
WmShell::Get()->focus_cycler()->RotateFocus(FocusCycler::FORWARD);
EXPECT_TRUE(wm::IsActiveWindow(window.get()));
}
} // namespace test
} // namespace ash