From 2dd0e7b82286343c350320dc2ca2e84181783fd2 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Thu, 16 Jul 2026 16:31:37 +0100
Subject: [PATCH] test(agents): harden SSH stream error timing

---
 src/agents/sandbox/ssh.stream-errors.test.ts | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/agents/sandbox/ssh.stream-errors.test.ts b/src/agents/sandbox/ssh.stream-errors.test.ts
index 1f8ce851374d..4dfc87dab7a9 100644
--- a/src/agents/sandbox/ssh.stream-errors.test.ts
+++ b/src/agents/sandbox/ssh.stream-errors.test.ts
@@ -75,8 +75,15 @@ describe("SSH sandbox stream errors", () => {
         localDir,
         remoteDir: "/remote/workspace",
       });
-      const rejection = expect(result).rejects.toThrow(expected);
-      await vi.waitFor(() => expect(spawnMock).toHaveBeenCalledTimes(2));
+      const rejection = result.then(
+        () => {
+          throw new Error(`expected rejection: ${expected}`);
+        },
+        (error: unknown) => {
+          expect(error).toEqual(expect.objectContaining({ message: expected }));
+        },
+      );
+      await vi.waitFor(() => expect(spawnMock).toHaveBeenCalledTimes(2), { timeout: 10_000 });
       const [childName, streamName] = stream.split(".") as ["tar" | "ssh", keyof MockChildProcess];
       const failedStream = { tar, ssh }[childName][streamName] as PassThrough;