test(agents): stabilize non-finite run drain timeout

compoodment · compoodment · commit 19dfdad770ef · 2026-05-29T14:32:48.000+02:00
diff --git a/src/agents/run-wait.test.ts b/src/agents/run-wait.test.ts
@@ -478,14 +478,21 @@ describe("waitForAgentRunsToDrain", () => {
     callGatewayMock.mockResolvedValue({ status: "ok" });
     let activeRunIds = ["run-1"];
 
-    const result = await waitForAgentRunsToDrain({
-      timeoutMs: Number.NaN,
-      getPendingRunIds: () => {
-        const current = activeRunIds;
-        activeRunIds = [];
-        return current;
-      },
-    });
+    vi.useFakeTimers();
+    vi.setSystemTime(new Date("2026-01-01T00:00:00.000Z"));
+    let result: Awaited<ReturnType<typeof waitForAgentRunsToDrain>>;
+    try {
+      result = await waitForAgentRunsToDrain({
+        timeoutMs: Number.NaN,
+        getPendingRunIds: () => {
+          const current = activeRunIds;
+          activeRunIds = [];
+          return current;
+        },
+      });
+    } finally {
+      vi.useRealTimers();
+    }
 
     expect(result.timedOut).toBe(false);
     expect(Number.isFinite(result.deadlineAtMs)).toBe(true);
