diff --git a/packages/dashboard/app/hooks/__tests__/useChat.test.ts b/packages/dashboard/app/hooks/__tests__/useChat.test.ts index 2f0d702dc3..674811c653 100644 --- a/packages/dashboard/app/hooks/__tests__/useChat.test.ts +++ b/packages/dashboard/app/hooks/__tests__/useChat.test.ts @@ -2173,8 +2173,10 @@ describe("useChat", () => { result.current.selectSession("session-001"); }); - await act(async () => { - await new Promise((resolve) => setTimeout(resolve, 25)); + await waitFor(() => { + expect(result.current.pendingMessage).toBe("Queued follow-up"); + expect(mockStreamChatResponse).not.toHaveBeenCalled(); + expect(localStorage.getItem(getChatPendingMessageKey("session-001"))).toBe("Queued follow-up"); }); expect(result.current.pendingMessage).toBe("Queued follow-up");