From 9ef77362ee848b468d81207a5fd9e108474e9ba1 Mon Sep 17 00:00:00 2001
From: Vincent Koc <vincentkoc@ieee.org>
Date: Wed, 11 Mar 2026 19:50:27 -0400
Subject: [PATCH] Tests: cover shared embedded run runtime state

---
 src/agents/pi-embedded-runner/runs.test.ts | 32 ++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/src/agents/pi-embedded-runner/runs.test.ts b/src/agents/pi-embedded-runner/runs.test.ts
index 73201749317..d9bf90f961d 100644
--- a/src/agents/pi-embedded-runner/runs.test.ts
+++ b/src/agents/pi-embedded-runner/runs.test.ts
@@ -1,4 +1,5 @@
 import { afterEach, describe, expect, it, vi } from "vitest";
+import { importFreshModule } from "../../../test/helpers/import-fresh.js";
 import {
   __testing,
   abortEmbeddedPiRun,
@@ -105,4 +106,35 @@ describe("pi-embedded runner run registry", () => {
       vi.useRealTimers();
     }
   });
+
+  it("shares active run state across distinct module instances", async () => {
+    const runsA = await importFreshModule<typeof import("./runs.js")>(
+      import.meta.url,
+      "./runs.js?scope=shared-a",
+    );
+    const runsB = await importFreshModule<typeof import("./runs.js")>(
+      import.meta.url,
+      "./runs.js?scope=shared-b",
+    );
+    const handle = {
+      queueMessage: async () => {},
+      isStreaming: () => true,
+      isCompacting: () => false,
+      abort: vi.fn(),
+    };
+
+    runsA.__testing.resetActiveEmbeddedRuns();
+    runsB.__testing.resetActiveEmbeddedRuns();
+
+    try {
+      runsA.setActiveEmbeddedRun("session-shared", handle);
+      expect(runsB.isEmbeddedPiRunActive("session-shared")).toBe(true);
+
+      runsB.clearActiveEmbeddedRun("session-shared", handle);
+      expect(runsA.isEmbeddedPiRunActive("session-shared")).toBe(false);
+    } finally {
+      runsA.__testing.resetActiveEmbeddedRuns();
+      runsB.__testing.resetActiveEmbeddedRuns();
+    }
+  });
 });