diff --git a/src/gateway/server-methods/agent.create-event.test.ts b/src/gateway/server-methods/agent.create-event.test.ts index d28c5005fbb..95f5a84c742 100644 --- a/src/gateway/server-methods/agent.create-event.test.ts +++ b/src/gateway/server-methods/agent.create-event.test.ts @@ -98,14 +98,16 @@ describe("agent handler session create events", () => { undefined, { runId: "idem-agent-create-event" }, ); - expect(broadcastToConnIds).toHaveBeenCalledWith( - "sessions.changed", - expect.objectContaining({ - sessionKey: "agent:main:subagent:create-test", - reason: "create", - }), - new Set(["conn-1"]), - { dropIfSlow: true }, + await vi.waitFor(() => + expect(broadcastToConnIds).toHaveBeenCalledWith( + "sessions.changed", + expect.objectContaining({ + sessionKey: "agent:main:subagent:create-test", + reason: "create", + }), + new Set(["conn-1"]), + { dropIfSlow: true }, + ), ); }); });