From e583e716f2301a3270c51416ea30588fa062f3ab Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Wed, 18 Feb 2026 18:00:07 +0000 Subject: [PATCH] perf(test): use expect.poll for background abort completion --- ...bash-tools.exec.background-abort.e2e.test.ts | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/src/agents/bash-tools.exec.background-abort.e2e.test.ts b/src/agents/bash-tools.exec.background-abort.e2e.test.ts index 0a8779faf4..8ee19937aa 100644 --- a/src/agents/bash-tools.exec.background-abort.e2e.test.ts +++ b/src/agents/bash-tools.exec.background-abort.e2e.test.ts @@ -14,11 +14,18 @@ afterEach(() => { async function waitForFinishedSession(sessionId: string) { let finished = getFinishedSession(sessionId); - const deadline = Date.now() + (process.platform === "win32" ? 10_000 : 2_000); - while (!finished && Date.now() < deadline) { - await sleep(20); - finished = getFinishedSession(sessionId); - } + await expect + .poll( + () => { + finished = getFinishedSession(sessionId); + return Boolean(finished); + }, + { + timeout: process.platform === "win32" ? 10_000 : 2_000, + interval: 20, + }, + ) + .toBe(true); return finished; }