1 // Copyright 2014 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 "chrome/browser/ui/cocoa/apps/quit_with_apps_controller_mac.h"
6
7 #include "apps/app_window_registry.h"
8 #include "apps/ui/native_app_window.h"
9 #include "base/command_line.h"
10 #include "base/run_loop.h"
11 #include "chrome/browser/apps/app_browsertest_util.h"
12 #include "chrome/browser/browser_process.h"
13 #include "chrome/browser/chrome_browser_application_mac.h"
14 #include "chrome/browser/extensions/extension_service.h"
15 #include "chrome/browser/extensions/extension_test_message_listener.h"
16 #include "chrome/browser/lifetime/application_lifetime.h"
17 #include "chrome/browser/notifications/message_center_notification_manager.h"
18 #include "chrome/browser/notifications/notification_ui_manager.h"
19 #include "chrome/browser/profiles/profile.h"
20 #include "chrome/browser/ui/browser_iterator.h"
21 #include "chrome/browser/ui/browser_window.h"
22 #include "chrome/common/chrome_switches.h"
23 #include "content/public/browser/notification_service.h"
24 #include "content/public/test/test_utils.h"
25 #include "extensions/common/extension.h"
26 #include "ui/message_center/message_center.h"
27
28 typedef apps::AppWindowRegistry::AppWindowList AppWindowList;
29
30 namespace {
31
32 class QuitWithAppsControllerInteractiveTest
33 : public extensions::PlatformAppBrowserTest {
34 protected:
QuitWithAppsControllerInteractiveTest()35 QuitWithAppsControllerInteractiveTest() : app_(NULL) {}
36
~QuitWithAppsControllerInteractiveTest()37 virtual ~QuitWithAppsControllerInteractiveTest() {}
38
SetUpCommandLine(CommandLine * command_line)39 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
40 PlatformAppBrowserTest::SetUpCommandLine(command_line);
41 command_line->AppendSwitch(switches::kAppsKeepChromeAlive);
42 }
43
44 const extensions::Extension* app_;
45
46 private:
47 DISALLOW_COPY_AND_ASSIGN(QuitWithAppsControllerInteractiveTest);
48 };
49
50 } // namespace
51
52 // Test that quitting while apps are open shows a notification instead.
IN_PROC_BROWSER_TEST_F(QuitWithAppsControllerInteractiveTest,QuitBehavior)53 IN_PROC_BROWSER_TEST_F(QuitWithAppsControllerInteractiveTest, QuitBehavior) {
54 scoped_refptr<QuitWithAppsController> controller =
55 new QuitWithAppsController();
56 const Notification* notification;
57 message_center::MessageCenter* message_center =
58 message_center::MessageCenter::Get();
59
60 // With no app windows open, ShouldQuit returns true.
61 EXPECT_TRUE(controller->ShouldQuit());
62 notification = g_browser_process->notification_ui_manager()->FindById(
63 QuitWithAppsController::kQuitWithAppsNotificationID);
64 EXPECT_EQ(NULL, notification);
65
66 // Open an app window.
67 ExtensionTestMessageListener listener("Launched", false);
68 app_ = InstallAndLaunchPlatformApp("minimal_id");
69 ASSERT_TRUE(listener.WaitUntilSatisfied());
70
71 // One browser and one app window at this point.
72 EXPECT_FALSE(chrome::BrowserIterator().done());
73 EXPECT_TRUE(apps::AppWindowRegistry::IsAppWindowRegisteredInAnyProfile(0));
74
75 // On the first quit, show notification.
76 EXPECT_FALSE(controller->ShouldQuit());
77 EXPECT_TRUE(apps::AppWindowRegistry::IsAppWindowRegisteredInAnyProfile(0));
78 notification = g_browser_process->notification_ui_manager()->FindById(
79 QuitWithAppsController::kQuitWithAppsNotificationID);
80 ASSERT_TRUE(notification);
81
82 // If notification was dismissed by click, show again on next quit.
83 notification->delegate()->Click();
84 message_center->RemoveAllNotifications(false);
85 EXPECT_FALSE(controller->ShouldQuit());
86 EXPECT_TRUE(apps::AppWindowRegistry::IsAppWindowRegisteredInAnyProfile(0));
87 notification = g_browser_process->notification_ui_manager()->FindById(
88 QuitWithAppsController::kQuitWithAppsNotificationID);
89 ASSERT_TRUE(notification);
90
91 EXPECT_FALSE(chrome::BrowserIterator().done());
92 EXPECT_TRUE(apps::AppWindowRegistry::IsAppWindowRegisteredInAnyProfile(0));
93
94 // If notification is closed by user, don't show it next time.
95 notification->delegate()->Close(true);
96 message_center->RemoveAllNotifications(false);
97 EXPECT_FALSE(controller->ShouldQuit());
98 EXPECT_TRUE(apps::AppWindowRegistry::IsAppWindowRegisteredInAnyProfile(0));
99 notification = g_browser_process->notification_ui_manager()->FindById(
100 QuitWithAppsController::kQuitWithAppsNotificationID);
101 EXPECT_EQ(NULL, notification);
102
103 EXPECT_FALSE(chrome::BrowserIterator().done());
104 EXPECT_TRUE(apps::AppWindowRegistry::IsAppWindowRegisteredInAnyProfile(0));
105
106 // Quitting should not quit but close all browsers
107 content::WindowedNotificationObserver observer(
108 chrome::NOTIFICATION_BROWSER_CLOSED,
109 content::NotificationService::AllSources());
110 chrome_browser_application_mac::Terminate();
111 observer.Wait();
112
113 EXPECT_TRUE(chrome::BrowserIterator().done());
114 EXPECT_TRUE(apps::AppWindowRegistry::IsAppWindowRegisteredInAnyProfile(0));
115
116 // Trying to quit while there are no browsers always shows notification.
117 EXPECT_FALSE(controller->ShouldQuit());
118 notification = g_browser_process->notification_ui_manager()->FindById(
119 QuitWithAppsController::kQuitWithAppsNotificationID);
120 ASSERT_TRUE(notification);
121
122 // Clicking "Quit All Apps." button closes all app windows. With no browsers
123 // open, this should also quit Chrome.
124 content::WindowedNotificationObserver quit_observer(
125 chrome::NOTIFICATION_APP_TERMINATING,
126 content::NotificationService::AllSources());
127 notification->delegate()->ButtonClick(0);
128 message_center->RemoveAllNotifications(false);
129 EXPECT_FALSE(apps::AppWindowRegistry::IsAppWindowRegisteredInAnyProfile(0));
130 quit_observer.Wait();
131 }
132