Backend code for quitOnClose option

background.ts

@@ -399,6 +399,10 @@ Electron.app.on('before-quit', async(event) => {
 });
 
 Electron.app.on('window-all-closed', () => {
+  if (cfg.window.quitOnClose) {
+    Electron.app.quit();
+  }
+
   // On macOS, hide the dock icon.
   Electron.app.dock?.hide();
 });

e2e/quit-on-close.e2e.spec.ts

@@ -0,0 +1,59 @@
+import path from 'path';
+
+import { test, expect, _electron, ElectronApplication } from '@playwright/test';
+
+import { createDefaultSettings, packageLogs, startRancherDesktop } from './utils/TestUtils';
+
+/**
+ * Using test.describe.serial make the test execute step by step, as described on each `test()` order
+ * Playwright executes test in parallel by default and it will not work for our app backend loading process.
+ * */
+test.describe.serial('quitOnClose setting', () => {
+  test.afterAll(async() => {
+    await packageLogs(__filename);
+  });
+
+  test('should quit when quitOnClose is true and window is closed', async() => {
+    createDefaultSettings({ window: { quitOnClose: true } });
+    const { electronApp } = await startRancherDesktop(__filename, false);
+
+    await expect(closeWindowsAndCheckQuit(electronApp)).resolves.toBe(true);
+  });
+
+  test('should not quit when quitOnClose is false and window is closed', async() => {
+    createDefaultSettings({ window: { quitOnClose: false } });
+    const { electronApp } = await startRancherDesktop(__filename, true);
+
+    await expect(closeWindowsAndCheckQuit(electronApp)).resolves.toBe(false);
+    const tracePath = path.join(__dirname, 'reports', `${ path.basename(__filename) }-quitOnCloseFalse.zip`);
+
+    electronApp.context().tracing.stop({ path: tracePath });
+    await electronApp.close();
+  });
+});
+
+/**
+ * Closes all of the windows in a running app. Returns a promise that
+ * resolves to true when the app has quit within a certain period of time,
+ * or that resolves to false when the app does not quit within that period
+ * of time.
+ * */
+function closeWindowsAndCheckQuit(electronApp: ElectronApplication): Promise<boolean> {
+  return electronApp.evaluate(async({ app, BrowserWindow }) => {
+    const quitReady = new Promise<boolean>((resolve) => {
+      app.on('will-quit', () => resolve(true));
+      app.on('window-all-closed', () => {
+        setTimeout(() => resolve(false), 3_000);
+      });
+    });
+
+    await Promise.all(BrowserWindow.getAllWindows().map((window) => {
+      return new Promise<void>((resolve) => {
+        window.on('closed', resolve);
+        window.close();
+      });
+    }));
+
+    return await quitReady;
+  });
+}

e2e/utils/TestUtils.ts

@@ -6,7 +6,7 @@ import os from 'os';
 import path from 'path';
 import util from 'util';
 
-import { ElectronApplication, expect } from '@playwright/test';
+import { expect, _electron, ElectronApplication, Page } from '@playwright/test';
 import _ from 'lodash';
 
 import { defaultSettings, Settings } from '@pkg/config/settings';
@@ -65,7 +65,7 @@ export function reportAsset(testPath: string, type: 'trace' | 'log' = 'trace') {
   return path.join(__dirname, '..', 'reports', `${ path.basename(testPath) }-${ name }`);
 }
 
-async function packageLogs(testPath: string) {
+export async function packageLogs(testPath: string) {
   if (!process.env.CIRRUS_CI) {
     console.log('Skipping packaging logs, not running in Cirrus CI');
 
@@ -171,3 +171,35 @@ export async function retry<T>(proc: () => Promise<T>, options?: { delay?: numbe
     }
   }
 }
+
+/**
+ * Run Rancher Desktop; return promise that resolves to commonly-used
+ * playwright objects when it has started.
+ * @param testPath The path to the test file.
+ * @param tracing Whether to start tracing.
+ */
+export async function startRancherDesktop(testPath: string, tracing: boolean): Promise<{electronApp: ElectronApplication, page: Page}> {
+  const electronApp = await _electron.launch({
+    args: [
+      path.join(__dirname, '../../'),
+      '--disable-gpu',
+      '--whitelisted-ips=',
+      // See pkg/rancher-desktop/utils/commandLine.ts before changing the next item as the final option.
+      '--disable-dev-shm-usage',
+      '--no-modal-dialogs',
+    ],
+    env: {
+      ...process.env,
+      RD_LOGS_DIR:     reportAsset(testPath, 'log'),
+      RD_MOCK_BACKEND: '1',
+    },
+  });
+
+  if (tracing) {
+    electronApp.context().tracing.start({ screenshots: true, snapshots: true });
+  }
+
+  const page = await electronApp.firstWindow();
+
+  return { electronApp, page };
+}