From b0e3606382742e7a08a7a108aad3a5bf6baa2330 Mon Sep 17 00:00:00 2001
From: atharva kamble <60008419+athkdev@users.noreply.github.com>
Date: Thu, 16 Oct 2025 18:24:12 -0400
Subject: [PATCH] tests: add unit tests for unloading workspaces, p=#10840

---
 .../workspaces/browser_workspace_unload.js    | 113 ++++++++++++++++++
 1 file changed, 113 insertions(+)
 create mode 100644 src/zen/tests/workspaces/browser_workspace_unload.js

diff --git a/src/zen/tests/workspaces/browser_workspace_unload.js b/src/zen/tests/workspaces/browser_workspace_unload.js
new file mode 100644
index 000000000..7f2c60f99
--- /dev/null
+++ b/src/zen/tests/workspaces/browser_workspace_unload.js
@@ -0,0 +1,113 @@
+/* Any copyright is dedicated to the Public Domain.
+   https://creativecommons.org/publicdomain/zero/1.0/ */
+
+'use strict';
+
+// verify that workspace unloading works
+add_task(async function test_UnloadWorkspace_WithMultipleTabs() {
+  const workspaceId = await gZenWorkspaces.createAndSaveWorkspace('Test Workspace 1');
+  const tabs = [];
+  for (let i = 0; i < 3; i++) {
+    const tab = await BrowserTestUtils.openNewForegroundTab(
+      window.gBrowser,
+      `data:text/html,<title>Workspace Tab ${i}</title>`,
+      true,
+      { skipAnimation: true }
+    );
+    tab.setAttribute('zen-workspace-id', workspaceId);
+    tabs.push(tab);
+  }
+
+  for (const tab of tabs) {
+    ok(!tab.hasAttribute('pending'), 'Tab should not be pending before unload');
+    ok(tab.linkedPanel, 'Tab should have linked panel before unload');
+  }
+
+  await gZenWorkspaces.unloadWorkspace();
+
+  for (const tab of tabs) {
+    ok(tab.hasAttribute('pending'), 'Tab should be pending after unload');
+    ok(!tab.linkedPanel, 'Tab should not have linked panel after unload');
+  }
+
+  await gZenWorkspaces.removeWorkspace(workspaceId);
+});
+
+// verify that essential tabs are not unloaded
+add_task(async function test_UnloadWorkspace_WithEssentialTabs() {
+  const workspaceId = await gZenWorkspaces.createAndSaveWorkspace('Test Workspace 2');
+
+  const regularTab = await BrowserTestUtils.openNewForegroundTab(
+    window.gBrowser,
+    'data:text/html,<title>Hi! I am a Regular Tab</title>',
+    true,
+    { skipAnimation: true }
+  );
+  regularTab.setAttribute('zen-workspace-id', workspaceId);
+
+  const essentialTab = await BrowserTestUtils.openNewForegroundTab(
+    window.gBrowser,
+    'data:text/html,<title>Hi! I am an Essential Tab</title>',
+    true,
+    { skipAnimation: true }
+  );
+  essentialTab.setAttribute('zen-workspace-id', workspaceId);
+  essentialTab.setAttribute('zen-essential', 'true');
+
+  await gZenWorkspaces.unloadWorkspace();
+
+  ok(regularTab.hasAttribute('pending'), 'Regular tab should be unloaded');
+  ok(!regularTab.linkedPanel, 'Regular tab should not have linked panel');
+
+  ok(!essentialTab.hasAttribute('pending'), 'Essential tab should not be unloaded');
+  ok(essentialTab.linkedPanel, 'Essential tab should still have linked panel');
+
+  await gZenWorkspaces.removeWorkspace(workspaceId);
+});
+
+// only tabs from the targeted workspace should be unloaded
+add_task(async function test_UnloadWorkspace_TargetedWorkspaceIsolation() {
+  const inActiveWorkspaceId = await gZenWorkspaces.createAndSaveWorkspace(
+    'Test In-Active Workspace'
+  );
+  const activeWorkspaceId = await gZenWorkspaces.createAndSaveWorkspace('Test Active Workspace');
+
+  const inActiveWorkspaceTabs = [];
+  for (let i = 0; i < 2; i++) {
+    const tab = await BrowserTestUtils.openNewForegroundTab(
+      window.gBrowser,
+      `data:text/html,<title>In-Active Workspace Tab ${i}</title>`,
+      true,
+      { skipAnimation: true }
+    );
+    tab.setAttribute('zen-workspace-id', inActiveWorkspaceId);
+    inActiveWorkspaceTabs.push(tab);
+  }
+
+  const activeWorkspaceTabs = [];
+  for (let i = 0; i < 2; i++) {
+    const tab = await BrowserTestUtils.openNewForegroundTab(
+      window.gBrowser,
+      `data:text/html,<title>Active Workspace Tab ${i}</title>`,
+      true,
+      { skipAnimation: true }
+    );
+    tab.setAttribute('zen-workspace-id', activeWorkspaceId);
+    activeWorkspaceTabs.push(tab);
+  }
+
+  await gZenWorkspaces.unloadWorkspace(); // this unloads the latest created workspace -> activeWorkspaceId
+
+  for (const tab of activeWorkspaceTabs) {
+    ok(tab.hasAttribute('pending'), 'Active workspace tab should be pending after unload');
+    ok(!tab.linkedPanel, 'Active workspace tab should not have linked panel after unload');
+  }
+
+  for (const tab of inActiveWorkspaceTabs) {
+    ok(!tab.hasAttribute('pending'), 'In-Active workspace tab should NOT be pending after unload');
+    ok(tab.linkedPanel, 'In-Active workspace tab should still have linked panel after unload');
+  }
+
+  await gZenWorkspaces.removeWorkspace(inActiveWorkspaceId);
+  await gZenWorkspaces.removeWorkspace(activeWorkspaceId);
+});