From 34649d3257656aadd0922ce23d83de2e650e27df Mon Sep 17 00:00:00 2001
From: Shakker <shakkerdroid@gmail.com>
Date: Sat, 9 May 2026 00:52:58 +0100
Subject: [PATCH] test: use vitest polling for session memory

---
 src/hooks/bundled/session-memory/handler.test.ts | 12 +-----------
 1 file changed, 1 insertion(+), 11 deletions(-)

diff --git a/src/hooks/bundled/session-memory/handler.test.ts b/src/hooks/bundled/session-memory/handler.test.ts
index 01f8607984f..e48e42a5d64 100644
--- a/src/hooks/bundled/session-memory/handler.test.ts
+++ b/src/hooks/bundled/session-memory/handler.test.ts
@@ -193,16 +193,6 @@ function expectMemoryConversation(params: {
   }
 }
 
-async function waitUntil(condition: () => boolean, timeoutMs = 500): Promise<void> {
-  const deadline = Date.now() + timeoutMs;
-  while (!condition()) {
-    if (Date.now() > deadline) {
-      throw new Error("condition was not met before timeout");
-    }
-    await new Promise((resolve) => setTimeout(resolve, 5));
-  }
-}
-
 describe("session-memory hook", () => {
   it("skips non-command events", async () => {
     const tempDir = await createCaseWorkspace("workspace");
@@ -370,7 +360,7 @@ describe("session-memory hook", () => {
         await handler(event);
         expect(Date.now() - startedAt).toBeLessThan(100);
 
-        await waitUntil(() => generateSlug.mock.calls.length === 1);
+        await vi.waitFor(() => expect(generateSlug).toHaveBeenCalledTimes(1), { interval: 1 });
         resolveSlug?.("slow-reset");
         await flushSessionMemoryWritesForTest();