@vellumai/assistant 0.12.0-staging.1 → 0.12.0

package/src/__tests__/subagent-manager-notify.test.ts

@@ -209,15 +209,36 @@ describe("SubagentManager abort notification", () => {
     manager.abort(subagentId, sendToClient);
 
     const statusMsg = clientMessages.find(
-      (m) => m.type === "subagent_status_changed",
+      (m): m is Extract<AssistantEvent, { type: "subagent_status_changed" }> =>
+        m.type === "subagent_status_changed",
     );
     expect(statusMsg).toBeDefined();
-    expect((statusMsg as unknown as Record<string, unknown>).subagentId).toBe(
+    expect(statusMsg!.subagentId).toBe(subagentId);
+    expect(statusMsg!.status).toBe("aborted");
+  });
+
+  test("reannounce sends each child's current status through the stored sender", () => {
+    const manager = new SubagentManager();
+    const subagentId = "sub-1";
+
+    const clientMessages: AssistantEvent[] = [];
+    const sendToClient = (msg: AssistantEvent) => clientMessages.push(msg);
+    injectFakeSubagent(
+      manager,
       subagentId,
+      makeState(subagentId),
+      sendToClient,
     );
-    expect((statusMsg as unknown as Record<string, unknown>).status).toBe(
-      "aborted",
+
+    manager.reannounceChildStatuses("parent-sess-1");
+
+    const statusMsg = clientMessages.find(
+      (m): m is Extract<AssistantEvent, { type: "subagent_status_changed" }> =>
+        m.type === "subagent_status_changed",
     );
+    expect(statusMsg).toBeDefined();
+    expect(statusMsg!.subagentId).toBe(subagentId);
+    expect(statusMsg!.status).toBe("running");
   });
 
   test("abort returns false for unknown subagent", () => {