diff --git a/extensions/feishu/src/monitor.cleanup.test.ts b/extensions/feishu/src/monitor.cleanup.test.ts index b74417f0dd5..ed103db2f9a 100644 --- a/extensions/feishu/src/monitor.cleanup.test.ts +++ b/extensions/feishu/src/monitor.cleanup.test.ts @@ -65,8 +65,10 @@ describe("feishu websocket cleanup", () => { eventDispatcher: {} as never, }); - expect(wsClient.start).toHaveBeenCalledTimes(1); - expect(wsClients.get(accountId)).toBe(wsClient); + await vi.waitFor(() => { + expect(wsClient.start).toHaveBeenCalledTimes(1); + expect(wsClients.get(accountId)).toBe(wsClient); + }); abortController.abort(); await monitorPromise;