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/first_run/first_run_helper.h"
6
7 #include "ash/first_run/desktop_cleaner.h"
8 #include "ash/shell.h"
9 #include "ash/shell_window_ids.h"
10 #include "ash/test/ash_test_base.h"
11 #include "ui/aura/test/event_generator.h"
12 #include "ui/events/event_handler.h"
13 #include "ui/views/widget/widget.h"
14 #include "ui/views/window/dialog_delegate.h"
15
16 namespace ash {
17 namespace test {
18
19 namespace {
20
21 class TestModalDialogDelegate : public views::DialogDelegateView {
22 public:
TestModalDialogDelegate()23 TestModalDialogDelegate() {}
~TestModalDialogDelegate()24 virtual ~TestModalDialogDelegate() {}
25
26 // Overridden from views::WidgetDelegate:
GetModalType() const27 virtual ui::ModalType GetModalType() const OVERRIDE {
28 return ui::MODAL_TYPE_SYSTEM;
29 }
30
31 private:
32 DISALLOW_COPY_AND_ASSIGN(TestModalDialogDelegate);
33 };
34
35 class CountingEventHandler : public ui::EventHandler {
36 public:
37 // Handler resets |*mouse_events_registered_| during construction and updates
38 // it after each registered event.
CountingEventHandler(int * mouse_events_registered)39 explicit CountingEventHandler(int* mouse_events_registered)
40 : mouse_events_registered_(mouse_events_registered) {
41 *mouse_events_registered = 0;
42 }
43
~CountingEventHandler()44 virtual ~CountingEventHandler() {}
45
46 private:
47 // ui::EventHandler overrides.
OnMouseEvent(ui::MouseEvent * event)48 virtual void OnMouseEvent(ui::MouseEvent* event) OVERRIDE {
49 ++*mouse_events_registered_;
50 }
51
52 int* mouse_events_registered_;
53
54 DISALLOW_COPY_AND_ASSIGN(CountingEventHandler);
55 };
56
57 } // namespace
58
59 class FirstRunHelperTest : public AshTestBase,
60 public FirstRunHelper::Observer {
61 public:
FirstRunHelperTest()62 FirstRunHelperTest() : cancelled_times_(0) {}
63
~FirstRunHelperTest()64 virtual ~FirstRunHelperTest() {}
65
SetUp()66 virtual void SetUp() OVERRIDE {
67 AshTestBase::SetUp();
68 CheckContainersAreVisible();
69 helper_.reset(ash::Shell::GetInstance()->CreateFirstRunHelper());
70 helper_->AddObserver(this);
71 helper_->GetOverlayWidget()->Show();
72 }
73
TearDown()74 virtual void TearDown() OVERRIDE {
75 EXPECT_TRUE(helper_.get());
76 helper_.reset();
77 CheckContainersAreVisible();
78 AshTestBase::TearDown();
79 }
80
CheckContainersAreVisible() const81 void CheckContainersAreVisible() const {
82 aura::Window* root_window = Shell::GetInstance()->GetPrimaryRootWindow();
83 std::vector<int> containers_to_check =
84 DesktopCleaner::GetContainersToHideForTest();
85 for (size_t i = 0; i < containers_to_check.size(); ++i) {
86 aura::Window* container =
87 Shell::GetContainer(root_window, containers_to_check[i]);
88 EXPECT_TRUE(container->IsVisible());
89 }
90 }
91
CheckContainersAreHidden() const92 void CheckContainersAreHidden() const {
93 aura::Window* root_window = Shell::GetInstance()->GetPrimaryRootWindow();
94 std::vector<int> containers_to_check =
95 DesktopCleaner::GetContainersToHideForTest();
96 for (size_t i = 0; i < containers_to_check.size(); ++i) {
97 aura::Window* container =
98 Shell::GetContainer(root_window, containers_to_check[i]);
99 EXPECT_TRUE(!container->IsVisible());
100 }
101 }
102
helper()103 FirstRunHelper* helper() { return helper_.get(); }
104
cancelled_times() const105 int cancelled_times() const { return cancelled_times_; }
106
107 private:
108 // FirstRunHelper::Observer overrides.
OnCancelled()109 virtual void OnCancelled() OVERRIDE {
110 ++cancelled_times_;
111 }
112
113 scoped_ptr<FirstRunHelper> helper_;
114 int cancelled_times_;
115
116 DISALLOW_COPY_AND_ASSIGN(FirstRunHelperTest);
117 };
118
119 // This test creates helper, checks that containers are hidden and then
120 // destructs helper.
TEST_F(FirstRunHelperTest,ContainersAreHidden)121 TEST_F(FirstRunHelperTest, ContainersAreHidden) {
122 CheckContainersAreHidden();
123 }
124
125 // Tests that helper correctly handles Escape key press.
TEST_F(FirstRunHelperTest,Cancel)126 TEST_F(FirstRunHelperTest, Cancel) {
127 GetEventGenerator().PressKey(ui::VKEY_ESCAPE, 0);
128 EXPECT_EQ(cancelled_times(), 1);
129 }
130
131 // Tests that modal window doesn't block events for overlay window.
TEST_F(FirstRunHelperTest,ModalWindowDoesNotBlock)132 TEST_F(FirstRunHelperTest, ModalWindowDoesNotBlock) {
133 views::Widget* modal_dialog = views::DialogDelegate::CreateDialogWidget(
134 new TestModalDialogDelegate(), CurrentContext(), NULL);
135 modal_dialog->Show();
136 // TODO(dzhioev): modal window should not steal focus from overlay window.
137 aura::Window* overlay_window = helper()->GetOverlayWidget()->GetNativeView();
138 overlay_window->Focus();
139 EXPECT_TRUE(overlay_window->HasFocus());
140 int mouse_events;
141 CountingEventHandler handler(&mouse_events);
142 overlay_window->AddPreTargetHandler(&handler);
143 GetEventGenerator().PressLeftButton();
144 GetEventGenerator().ReleaseLeftButton();
145 EXPECT_EQ(mouse_events, 2);
146 overlay_window->RemovePreTargetHandler(&handler);
147 }
148
149 } // namespace test
150 } // namespace ash
151