1 files changed, 29 insertions, 0 deletions
diff --git a/src/features/tabs/tab-groups.test.ts b/src/features/tabs/tab-groups.test.ts
index 83f273c..a89a469 100644
--- a/src/features/tabs/tab-groups.test.ts
+++ b/src/features/tabs/tab-groups.test.ts
@@ -1,6 +1,7 @@
 import { describe, expect, it } from "vitest";
 import {
   buildTabGroups,
+  cascadeCloseIds,
   directGroupChildren,
   groupBleedPx,
   groupDescendants,
@@ -103,6 +104,34 @@ describe("groupDescendants / directGroupChildren / groupParentCandidates", () =>
   });
 });
 
+describe("cascadeCloseIds (the ids a cascading tab close removes)", () => {
+  const tabs = [
+    tab("p"),
+    tab("c1", "p"),
+    tab("c2", "p"),
+    tab("g", "c1"), // sub-sub-agent: grouped under c1, still cascades from p
+    tab("solo"),
+  ];
+
+  it("returns the tab first, then its transitive descendants in tab order", () => {
+    expect(cascadeCloseIds(tabs, "p")).toEqual(["p", "c1", "c2", "g"]);
+  });
+
+  it("starts the cascade at a mid-level tab, taking only its own subtree", () => {
+    expect(cascadeCloseIds(tabs, "c1")).toEqual(["c1", "g"]);
+  });
+
+  it("returns just the tab itself when it has no subagents (closes without a prompt)", () => {
+    expect(cascadeCloseIds(tabs, "solo")).toEqual(["solo"]);
+    expect(cascadeCloseIds(tabs, "c2")).toEqual(["c2"]);
+  });
+
+  it("ignores tabs whose group parent is no longer open", () => {
+    const stale = [tab("p"), tab("orphan", "gone")];
+    expect(cascadeCloseIds(stale, "p")).toEqual(["p"]);
+  });
+});
+
 describe("groupBleedPx / groupGutterPx", () => {
   it("is zero without groups", () => {
     expect(groupGutterPx(0)).toBe(0);