From 361d561cb762217ebffa3925b249400eee331dbd Mon Sep 17 00:00:00 2001 From: Alassane <79718216+yattdev@users.noreply.github.com> Date: Mon, 31 Aug 2026 08:44:13 +0000 Subject: [PATCH] test(web): stabilize session hydration ordering (#3168) Co-authored-by: ayattara --- apps/web/hooks/domains/session/use-session-messages.test.ts | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/apps/web/hooks/domains/session/use-session-messages.test.ts b/apps/web/hooks/domains/session/use-session-messages.test.ts index 86d9a38411..af0a2aeab5 100644 --- a/apps/web/hooks/domains/session/use-session-messages.test.ts +++ b/apps/web/hooks/domains/session/use-session-messages.test.ts @@ -38,6 +38,7 @@ const mockState = { setActiveTurn: vi.fn(), reconcileActiveTurnAfterHydration: vi.fn(), }; +const mockStoreApi = { getState: () => mockState }; vi.mock("@/lib/api/domains/session-api", () => ({ listSessionTurns: (...args: unknown[]) => mockListSessionTurns(...args), @@ -49,7 +50,7 @@ vi.mock("@/lib/ws/connection", () => ({ vi.mock("@/components/state-provider", () => ({ useAppStore: (selector: (state: typeof mockState) => unknown) => selector(mockState), - useAppStoreApi: () => ({ getState: () => mockState }), + useAppStoreApi: () => mockStoreApi, })); import { taskId, sessionId } from "@/lib/types/ids"; @@ -364,6 +365,7 @@ describe("session subscription hydration ordering", () => { await readiness.promise; }); + expect(mockWebSocketClient.subscribeSessionWithReady).toHaveBeenCalledTimes(1); expect(mockWebSocketClient.request).toHaveBeenCalledWith( "message.list", expect.objectContaining({ session_id: "sess-1" }),