From b39f3cf2664c3b089fc17c574e2ecfddd261a8e7 Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Fri, 17 Apr 2026 19:31:40 +0100 Subject: [PATCH] test: avoid polling settled acp reconnect --- src/acp/translator.stop-reason.test.ts | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/src/acp/translator.stop-reason.test.ts b/src/acp/translator.stop-reason.test.ts index 39c36645047..a30fbcd8a55 100644 --- a/src/acp/translator.stop-reason.test.ts +++ b/src/acp/translator.stop-reason.test.ts @@ -537,11 +537,10 @@ describe("acp translator stop reason mapping", () => { await Promise.resolve(); agent.handleGatewayReconnect(); - await vi.waitFor(() => { - expect(settleSpy).toHaveBeenCalledWith({ - kind: "resolve", - value: { stopReason: "end_turn" }, - }); + await expect(promptPromise).resolves.toEqual({ stopReason: "end_turn" }); + expect(settleSpy).toHaveBeenCalledWith({ + kind: "resolve", + value: { stopReason: "end_turn" }, }); });