// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/ui/cocoa/apps/quit_with_apps_controller_mac.h"
#import "base/apple/foundation_util.h"
#include "base/command_line.h"
#include "base/memory/raw_ptr.h"
#include "base/run_loop.h"
#import "chrome/browser/app_controller_mac.h"
#include "chrome/browser/apps/platform_apps/app_browsertest_util.h"
#include "chrome/browser/apps/platform_apps/app_window_registry_util.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/chrome_browser_application_mac.h"
#include "chrome/browser/extensions/extension_service.h"
#include "chrome/browser/lifetime/application_lifetime.h"
#include "chrome/browser/lifetime/browser_shutdown.h"
#include "chrome/browser/lifetime/termination_notification.h"
#include "chrome/browser/notifications/notification_display_service_tester.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/profiles/profile_manager.h"
#include "chrome/browser/ui/browser_list.h"
#include "chrome/browser/ui/browser_window.h"
#include "chrome/common/chrome_switches.h"
#include "chrome/test/base/ui_test_utils.h"
#include "content/public/test/browser_test.h"
#include "content/public/test/test_utils.h"
#include "extensions/common/extension.h"
#include "extensions/test/extension_test_message_listener.h"
namespace {
// The param selects whether to use ChromeNativeAppWindowViewsMac, otherwise it
// will use NativeAppWindowCocoa.
class QuitWithAppsControllerInteractiveTest
: public extensions::PlatformAppBrowserTest {
public:
QuitWithAppsControllerInteractiveTest(
const QuitWithAppsControllerInteractiveTest&) = delete;
QuitWithAppsControllerInteractiveTest& operator=(
const QuitWithAppsControllerInteractiveTest&) = delete;
protected:
QuitWithAppsControllerInteractiveTest() : app_(nullptr) {}
void SetUpCommandLine(base::CommandLine* command_line) override {
PlatformAppBrowserTest::SetUpCommandLine(command_line);
command_line->AppendSwitch(switches::kAppsKeepChromeAliveInTests);
}
raw_ptr<const extensions::Extension> app_;
};
} // namespace
// Test that quitting while apps are open shows a notification instead.
IN_PROC_BROWSER_TEST_F(QuitWithAppsControllerInteractiveTest, QuitBehavior) {
scoped_refptr<QuitWithAppsController> controller =
new QuitWithAppsController();
ProfileManager* profile_manager = g_browser_process->profile_manager();
ASSERT_TRUE(profile_manager);
std::vector<Profile*> profiles(profile_manager->GetLoadedProfiles());
ASSERT_NE(0u, profiles.size());
NotificationDisplayServiceTester display_service(profiles[0]);
// With no app windows open, ShouldQuit returns true.
EXPECT_TRUE(controller->ShouldQuit());
EXPECT_FALSE(display_service.GetNotification(
QuitWithAppsController::kQuitWithAppsNotificationID));
// Open an app window.
ExtensionTestMessageListener listener("Launched");
app_ = InstallAndLaunchPlatformApp("minimal_id");
ASSERT_TRUE(listener.WaitUntilSatisfied());
// One browser and one app window at this point.
EXPECT_FALSE(BrowserList::GetInstance()->empty());
EXPECT_TRUE(AppWindowRegistryUtil::IsAppWindowVisibleInAnyProfile(0));
// On the first quit, show notification.
EXPECT_FALSE(controller->ShouldQuit());
EXPECT_TRUE(AppWindowRegistryUtil::IsAppWindowVisibleInAnyProfile(0));
EXPECT_TRUE(display_service.GetNotification(
QuitWithAppsController::kQuitWithAppsNotificationID));
// If notification was dismissed by click, show again on next quit.
display_service.SimulateClick(
NotificationHandler::Type::TRANSIENT,
QuitWithAppsController::kQuitWithAppsNotificationID, std::nullopt,
std::nullopt);
EXPECT_FALSE(display_service.GetNotification(
QuitWithAppsController::kQuitWithAppsNotificationID));
EXPECT_FALSE(controller->ShouldQuit());
EXPECT_TRUE(AppWindowRegistryUtil::IsAppWindowVisibleInAnyProfile(0));
EXPECT_TRUE(display_service.GetNotification(
QuitWithAppsController::kQuitWithAppsNotificationID));
EXPECT_FALSE(BrowserList::GetInstance()->empty());
EXPECT_TRUE(AppWindowRegistryUtil::IsAppWindowVisibleInAnyProfile(0));
// If notification is closed by user, don't show it next time.
display_service.RemoveNotification(
NotificationHandler::Type::TRANSIENT,
QuitWithAppsController::kQuitWithAppsNotificationID, true);
EXPECT_FALSE(controller->ShouldQuit());
EXPECT_TRUE(AppWindowRegistryUtil::IsAppWindowVisibleInAnyProfile(0));
EXPECT_FALSE(display_service.GetNotification(
QuitWithAppsController::kQuitWithAppsNotificationID));
EXPECT_FALSE(BrowserList::GetInstance()->empty());
EXPECT_TRUE(AppWindowRegistryUtil::IsAppWindowVisibleInAnyProfile(0));
// Get a reference to the open app window before the browser closes.
extensions::AppWindow* app_window = GetFirstAppWindow();
// Quitting should not quit but close all browsers
chrome_browser_application_mac::Terminate();
ui_test_utils::WaitForBrowserToClose();
EXPECT_TRUE(BrowserList::GetInstance()->empty());
EXPECT_TRUE(AppWindowRegistryUtil::IsAppWindowVisibleInAnyProfile(0));
// Trying to quit while there are no browsers always shows notification.
EXPECT_FALSE(controller->ShouldQuit());
EXPECT_TRUE(display_service.GetNotification(
QuitWithAppsController::kQuitWithAppsNotificationID));
// Clicking "Quit All Apps." button closes all app windows. With no browsers
// open, this should also quit Chrome.
base::RunLoop quit_observer;
auto subscription =
browser_shutdown::AddAppTerminatingCallback(quit_observer.QuitClosure());
// Since closing app windows may be an async operation, use a watcher.
content::WebContentsDestroyedWatcher destroyed_watcher(
app_window->web_contents());
display_service.SimulateClick(
NotificationHandler::Type::TRANSIENT,
QuitWithAppsController::kQuitWithAppsNotificationID,
0 /* kQuitAllAppsButtonIndex */, std::nullopt);
destroyed_watcher.Wait();
EXPECT_FALSE(AppWindowRegistryUtil::IsAppWindowVisibleInAnyProfile(0));
quit_observer.Run();
}
// Test that, when powering off, Chrome will quit even if there are apps open.
IN_PROC_BROWSER_TEST_F(QuitWithAppsControllerInteractiveTest, QuitOnPowerOff) {
// Open an app window.
app_ = LoadAndLaunchPlatformApp("minimal_id", "Launched");
// First try to terminate with a packaged app running. Chrome should stay
// running in the background.
[NSApp terminate:nil];
EXPECT_FALSE(browser_shutdown::IsTryingToQuit());
// Simulate a terminate triggered by a power off or log out.
// Cocoa will send an NSWorkspaceWillPowerOffNotification followed by
// -[NSApplication terminate:].
NSNotification* notification =
[NSNotification notificationWithName:NSWorkspaceWillPowerOffNotification
object:nil];
[AppController.sharedController willPowerOff:notification];
[NSApp terminate:nil];
EXPECT_TRUE(browser_shutdown::IsTryingToQuit());
}