@alan-ai-hq/agent-manager 0.1.113 → 0.1.115

package/dist/__tests__/ws-client-connection.test.js

@@ -0,0 +1,295 @@
+import { randomUUID } from "node:crypto";
+import { rmSync } from "node:fs";
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { eventJournalDir } from "../event-journal-paths.js";
+import { createSandboxEventOutbox, sandboxOutboxKeyPath, sandboxOutboxSpoolPath, } from "../sandbox-outbox.js";
+import { isolateSandboxOutbox } from "./helpers/sandbox-outbox.js";
+isolateSandboxOutbox();
+const hoisted = vi.hoisted(() => {
+    const handlers = new Map();
+    const addHandler = (event, handler) => {
+        const eventHandlers = handlers.get(event) ?? new Set();
+        eventHandlers.add(handler);
+        handlers.set(event, eventHandlers);
+    };
+    const socket = {
+        connected: false,
+        active: true,
+        id: "fake-socket",
+        io: { engine: {}, on: vi.fn() },
+        emit: vi.fn(),
+        volatile: { emit: vi.fn() },
+        on: vi.fn((event, handler) => {
+            addHandler(event, handler);
+            return socket;
+        }),
+        once: vi.fn((event, handler) => {
+            const onceHandler = (...args) => {
+                socket.off(event, onceHandler);
+                handler(...args);
+            };
+            addHandler(event, onceHandler);
+            return socket;
+        }),
+        off: vi.fn((event, handler) => {
+            handlers.get(event)?.delete(handler);
+            return socket;
+        }),
+        connect: vi.fn(),
+        disconnect: vi.fn(),
+    };
+    return {
+        handlers,
+        socket,
+        ioMock: vi.fn(() => socket),
+        trigger(event, ...args) {
+            for (const handler of [...(handlers.get(event) ?? [])])
+                handler(...args);
+        },
+    };
+});
+vi.mock("socket.io-client", () => ({
+    io: hoisted.ioMock,
+    Socket: class {
+    },
+}));
+import { WSClient } from "../ws-client.js";
+describe("WSClient startup connection", () => {
+    let client;
+    beforeEach(() => {
+        vi.useFakeTimers();
+        hoisted.handlers.clear();
+        hoisted.socket.connected = false;
+        hoisted.socket.active = true;
+        hoisted.socket.emit.mockClear();
+        hoisted.socket.io.on.mockClear();
+        hoisted.socket.off.mockClear();
+        hoisted.socket.connect.mockClear();
+        hoisted.socket.disconnect.mockClear();
+    });
+    afterEach(async () => {
+        await client?.close();
+        client = undefined;
+        vi.restoreAllMocks();
+        vi.useRealTimers();
+    });
+    function createClient() {
+        return new WSClient("wss://sandbox-runner.example", "conversation-1", "task-1", "token-1", { onUserMessage: () => { }, onStop: () => { } }, "1.2.3");
+    }
+    it("keeps waiting after a transient connect error and resolves on a later retry", async () => {
+        client = createClient();
+        let settled = false;
+        const connection = client.waitForConnection(15_000).finally(() => {
+            settled = true;
+        });
+        hoisted.trigger("connect_error", new Error("websocket error"));
+        await Promise.resolve();
+        expect(settled).toBe(false);
+        hoisted.socket.connected = true;
+        hoisted.trigger("connect");
+        await expect(connection).resolves.toBeUndefined();
+    });
+    it("waits for a delayed event receipt before closing the transport", async () => {
+        client = createClient();
+        hoisted.socket.connected = true;
+        client.emitEvent({ type: "text", text: "final output", ts: 1 });
+        const send = hoisted.socket.emit.mock.calls.find((call) => call[0] === "agent_event");
+        const ack = send?.[2];
+        expect(ack).toBeTypeOf("function");
+        const closing = client.closeGracefully(1_000);
+        await vi.advanceTimersByTimeAsync(100);
+        expect(hoisted.socket.disconnect).not.toHaveBeenCalled();
+        ack({ ok: true });
+        await vi.advanceTimersByTimeAsync(25);
+        await closing;
+        expect(hoisted.socket.disconnect).toHaveBeenCalledOnce();
+    });
+    it("bounds offline shutdown and replays persisted events on the next process", async () => {
+        client = createClient();
+        client.emitEvent({ type: "text", text: "offline final output", ts: 1 });
+        const closing = client.closeGracefully(100);
+        await vi.advanceTimersByTimeAsync(99);
+        expect(hoisted.socket.disconnect).not.toHaveBeenCalled();
+        await vi.advanceTimersByTimeAsync(1);
+        await closing;
+        expect(hoisted.socket.disconnect).toHaveBeenCalledOnce();
+        hoisted.handlers.clear();
+        hoisted.socket.connected = true;
+        client = createClient();
+        hoisted.trigger("connect");
+        expect(hoisted.socket.emit).toHaveBeenCalledWith("agent_event", expect.objectContaining({ text: "offline final output" }), expect.any(Function));
+    });
+    it("rejects newly delivered messages while gracefully closing", async () => {
+        client = createClient();
+        const closing = client.closeGracefully(0);
+        const ack = vi.fn();
+        hoisted.trigger("user_message", { text: "too late", messageId: "shutdown-message" }, ack);
+        await closing;
+        expect(ack).toHaveBeenCalledWith({ ok: false, message: "Agent is shutting down" });
+    });
+    it("fails at the deadline and preserves the last connection error", async () => {
+        client = createClient();
+        const connection = client.waitForConnection(1_000);
+        const rejection = expect(connection).rejects.toThrow("WSClient connection timeout after 1000ms; last error: websocket error");
+        hoisted.trigger("connect_error", new Error("websocket error"));
+        await vi.advanceTimersByTimeAsync(1_000);
+        await rejection;
+    });
+    it("exposes one stable agent session id for heartbeat and lease fencing", () => {
+        client = createClient();
+        expect(client.getAgentSessionId()).toMatch(/^[0-9a-f-]{36}$/);
+        expect(client.getAgentSessionId()).toBe(client.getAgentSessionId());
+        expect(hoisted.ioMock).toHaveBeenLastCalledWith("wss://sandbox-runner.example", expect.objectContaining({
+            auth: {
+                token: "token-1",
+                agentSessionId: client.getAgentSessionId(),
+                terminationCauseV1: true,
+            },
+        }));
+    });
+    it("subscribes to Socket.IO Manager reconnect telemetry", () => {
+        client = createClient();
+        expect(hoisted.socket.io.on).toHaveBeenCalledWith("reconnect_attempt", expect.any(Function));
+        expect(hoisted.socket.io.on).toHaveBeenCalledWith("reconnect", expect.any(Function));
+        expect(hoisted.socket.io.on).toHaveBeenCalledWith("reconnect_error", expect.any(Function));
+        expect(hoisted.socket.io.on).toHaveBeenCalledWith("reconnect_failed", expect.any(Function));
+    });
+    it("reports only well-formed server capabilities", () => {
+        const onCapabilities = vi.fn();
+        client = new WSClient("wss://sandbox-runner.example", "conversation-1", "task-1", "token-1", { onUserMessage: () => { }, onStop: () => { }, onCapabilities }, "1.2.3");
+        hoisted.trigger("agent.capabilities", { executionLeaseV1: "yes" });
+        expect(onCapabilities).not.toHaveBeenCalled();
+        hoisted.trigger("agent.capabilities", { executionLeaseV1: true });
+        expect(onCapabilities).toHaveBeenCalledOnce();
+        expect(onCapabilities).toHaveBeenCalledWith({ executionLeaseV1: true });
+    });
+    it("lets the Socket.IO Manager own retryable transport reconnection", async () => {
+        client = createClient();
+        hoisted.socket.active = true;
+        hoisted.trigger("disconnect", "transport close");
+        await vi.advanceTimersByTimeAsync(10_000);
+        expect(hoisted.socket.connect).not.toHaveBeenCalled();
+    });
+    it("supervises an inactive socket after a legacy server shutdown", async () => {
+        vi.spyOn(Math, "random").mockReturnValue(0);
+        client = createClient();
+        hoisted.trigger("server_shutdown", { code: 1001, reason: "Server shutting down" });
+        hoisted.socket.active = false;
+        hoisted.trigger("disconnect", "io server disconnect");
+        await vi.advanceTimersByTimeAsync(999);
+        expect(hoisted.socket.connect).not.toHaveBeenCalled();
+        await vi.advanceTimersByTimeAsync(1);
+        expect(hoisted.socket.connect).toHaveBeenCalledTimes(1);
+    });
+    it("does not supervise a terminal server directive", async () => {
+        const onTerminalDirective = vi.fn();
+        client = new WSClient("wss://sandbox-runner.example", "conversation-1", "task-1", "token-1", { onUserMessage: () => { }, onStop: () => { }, onTerminalDirective }, "1.2.3");
+        hoisted.trigger("connection_directive", {
+            code: "invalid_session",
+            reason: "Invalid session",
+            retryable: false,
+        });
+        hoisted.socket.active = false;
+        hoisted.trigger("disconnect", "io server disconnect");
+        await vi.advanceTimersByTimeAsync(10_000);
+        expect(hoisted.socket.connect).not.toHaveBeenCalled();
+        expect(onTerminalDirective).toHaveBeenCalledWith({
+            code: "invalid_session",
+            reason: "Invalid session",
+        });
+    });
+    it("honors retryAfterMs for a supervised rate-limit reconnect", async () => {
+        vi.spyOn(Math, "random").mockReturnValue(0);
+        client = createClient();
+        hoisted.trigger("connection_directive", {
+            code: "event_rate_limit_exceeded",
+            reason: "Slow down",
+            retryable: true,
+            retryAfterMs: 10_000,
+        });
+        hoisted.socket.active = false;
+        hoisted.trigger("disconnect", "io server disconnect");
+        await vi.advanceTimersByTimeAsync(9_999);
+        expect(hoisted.socket.connect).not.toHaveBeenCalled();
+        await vi.advanceTimersByTimeAsync(1);
+        expect(hoisted.socket.connect).toHaveBeenCalledOnce();
+    });
+    it("supervises an inactive initial connection after a structured retryable error", async () => {
+        vi.spyOn(Math, "random").mockReturnValue(0);
+        client = createClient();
+        hoisted.socket.active = false;
+        const error = Object.assign(new Error("Authentication temporarily unavailable"), {
+            data: {
+                code: "agent_auth_unavailable",
+                retryable: true,
+                retryAfterMs: 2_000,
+            },
+        });
+        hoisted.trigger("connect_error", error);
+        await vi.advanceTimersByTimeAsync(1_999);
+        expect(hoisted.socket.connect).not.toHaveBeenCalled();
+        await vi.advanceTimersByTimeAsync(1);
+        expect(hoisted.socket.connect).toHaveBeenCalledTimes(1);
+    });
+    it("does not retry an inactive initial connection after a terminal auth error", async () => {
+        const onTerminalDirective = vi.fn();
+        client = new WSClient("wss://sandbox-runner.example", "conversation-1", "task-1", "token-1", { onUserMessage: () => { }, onStop: () => { }, onTerminalDirective }, "1.2.3");
+        hoisted.socket.active = false;
+        const error = Object.assign(new Error("Invalid session token"), {
+            data: { code: "agent_session_token_invalid", retryable: false },
+        });
+        hoisted.trigger("connect_error", error);
+        await vi.advanceTimersByTimeAsync(10_000);
+        expect(hoisted.socket.connect).not.toHaveBeenCalled();
+        expect(onTerminalDirective).toHaveBeenCalledWith({
+            code: "agent_session_token_invalid",
+            reason: undefined,
+        });
+    });
+    it("never infers initial retryability from an error message", async () => {
+        client = createClient();
+        hoisted.socket.active = false;
+        hoisted.trigger("connect_error", new Error("temporary database error, please retry"));
+        await vi.advanceTimersByTimeAsync(10_000);
+        expect(hoisted.socket.connect).not.toHaveBeenCalled();
+    });
+    it("cancels a supervised reconnect when the client intentionally closes", async () => {
+        vi.spyOn(Math, "random").mockReturnValue(0);
+        client = createClient();
+        hoisted.trigger("server_shutdown", { reason: "Server shutting down" });
+        hoisted.socket.active = false;
+        hoisted.trigger("disconnect", "io server disconnect");
+        await client.close();
+        await vi.advanceTimersByTimeAsync(10_000);
+        expect(hoisted.socket.connect).not.toHaveBeenCalled();
+        expect(hoisted.socket.disconnect).toHaveBeenCalledTimes(1);
+    });
+    it("flushes a final offline event before close resolves", async () => {
+        const conversationId = `close-${randomUUID()}`;
+        const path = sandboxOutboxSpoolPath(conversationId);
+        client = new WSClient("wss://sandbox-runner.example", conversationId, "task", "token", { onUserMessage: () => { }, onStop: () => { } }, "1.2.3");
+        try {
+            client.emitEvent({
+                type: "session_complete",
+                result: {
+                    success: true,
+                    summary: "final output",
+                    filesModified: [],
+                    planFilesCreated: [],
+                    iterations: 1,
+                },
+            });
+            await client.close();
+            const recovered = createSandboxEventOutbox({ conversationId, sessionToken: "token" });
+            expect(recovered.pendingCount()).toBe(1);
+            await recovered.flushDurable();
+            recovered.disconnect();
+        }
+        finally {
+            await client.close();
+            for (const entry of [path, sandboxOutboxKeyPath(path), eventJournalDir(path)])
+                rmSync(entry, { force: true, recursive: true });
+        }
+    });
+});
+//# sourceMappingURL=ws-client-connection.test.js.map

package/dist/__tests__/ws-client-connection.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-client-connection.test.js","sourceRoot":"","sources":["../../src/__tests__/ws-client-connection.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,eAAe,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,EACL,wBAAwB,EACxB,oBAAoB,EACpB,sBAAsB,GACvB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AAEnE,oBAAoB,EAAE,CAAC;AAEvB,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;IAE9B,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAwB,CAAC;IAEjD,MAAM,UAAU,GAAG,CAAC,KAAa,EAAE,OAAgB,EAAE,EAAE;QACrD,MAAM,aAAa,GAAG,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,GAAG,EAAW,CAAC;QAChE,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC3B,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;IACrC,CAAC,CAAC;IAEF,MAAM,MAAM,GAAG;QACb,SAAS,EAAE,KAAK;QAChB,MAAM,EAAE,IAAI;QACZ,EAAE,EAAE,aAAa;QACjB,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE;QAC/B,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE;QAC3B,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAAa,EAAE,OAAgB,EAAE,EAAE;YAC5C,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAC3B,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;QACF,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAAa,EAAE,OAAgB,EAAE,EAAE;YAC9C,MAAM,WAAW,GAAY,CAAC,GAAG,IAAI,EAAE,EAAE;gBACvC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;gBAC/B,OAAO,CAAC,GAAG,IAAI,CAAC,CAAC;YACnB,CAAC,CAAC;YACF,UAAU,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;YAC/B,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;QACF,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAAa,EAAE,OAAgB,EAAE,EAAE;YAC7C,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;YACrC,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;QACF,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;QAChB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;KACpB,CAAC;IAEF,OAAO;QACL,QAAQ;QACR,MAAM;QACN,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC;QAC3B,OAAO,CAAC,KAAa,EAAE,GAAG,IAAe;YACvC,KAAK,MAAM,OAAO,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;gBAAE,OAAO,CAAC,GAAG,IAAI,CAAC,CAAC;QAC3E,CAAC;KACF,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACjC,EAAE,EAAE,OAAO,CAAC,MAAM;IAClB,MAAM,EAAE;KAAQ;CACjB,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAE3C,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,IAAI,MAA4B,CAAC;IAEjC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,OAAO,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACzB,OAAO,CAAC,MAAM,CAAC,SAAS,GAAG,KAAK,CAAC;QACjC,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC;QAC7B,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC;QACjC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC;QAC/B,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;QACnC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,MAAM,EAAE,KAAK,EAAE,CAAC;QACtB,MAAM,GAAG,SAAS,CAAC;QACnB,EAAE,CAAC,eAAe,EAAE,CAAC;QACrB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,SAAS,YAAY;QACnB,OAAO,IAAI,QAAQ,CACjB,8BAA8B,EAC9B,gBAAgB,EAChB,QAAQ,EACR,SAAS,EACT,EAAE,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,EAC7C,OAAO,CACR,CAAC;IACJ,CAAC;IAED,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QAC3F,MAAM,GAAG,YAAY,EAAE,CAAC;QACxB,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,MAAM,UAAU,GAAG,MAAM,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;YAC/D,OAAO,GAAG,IAAI,CAAC;QACjB,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,OAAO,CAAC,eAAe,EAAE,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC;QAC/D,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;QACxB,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE5B,OAAO,CAAC,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC;QAChC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAE3B,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,GAAG,YAAY,EAAE,CAAC;QACxB,OAAO,CAAC,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC;QAChC,MAAM,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;QAChE,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,aAAa,CAAC,CAAC;QACtF,MAAM,GAAG,GAAG,IAAI,EAAE,CAAC,CAAC,CAAsC,CAAC;QAC3D,MAAM,CAAC,GAAG,CAAC,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;QACnC,MAAM,OAAO,GAAG,MAAM,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC9C,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAC;QACvC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzD,GAAG,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QAClB,MAAM,EAAE,CAAC,wBAAwB,CAAC,EAAE,CAAC,CAAC;QACtC,MAAM,OAAO,CAAC;QACd,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,EAAE,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,MAAM,GAAG,YAAY,EAAE,CAAC;QACxB,MAAM,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,sBAAsB,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;QACxE,MAAM,OAAO,GAAG,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;QAC5C,MAAM,EAAE,CAAC,wBAAwB,CAAC,EAAE,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzD,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,OAAO,CAAC;QACd,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,EAAE,CAAC;QAEzD,OAAO,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACzB,OAAO,CAAC,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC;QAChC,MAAM,GAAG,YAAY,EAAE,CAAC;QACxB,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC3B,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC9C,aAAa,EACb,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC,EACzD,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CACrB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,GAAG,YAAY,EAAE,CAAC;QACxB,MAAM,OAAO,GAAG,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACpB,OAAO,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,kBAAkB,EAAE,EAAE,GAAG,CAAC,CAAC;QAC1F,MAAM,OAAO,CAAC;QACd,MAAM,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,wBAAwB,EAAE,CAAC,CAAC;IACrF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,GAAG,YAAY,EAAE,CAAC;QACxB,MAAM,UAAU,GAAG,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACnD,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,OAAO,CAClD,uEAAuE,CACxE,CAAC;QAEF,OAAO,CAAC,OAAO,CAAC,eAAe,EAAE,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC;QAC/D,MAAM,EAAE,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;QAEzC,MAAM,SAAS,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,MAAM,GAAG,YAAY,EAAE,CAAC;QAExB,MAAM,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAC9D,MAAM,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC;QACpE,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,wBAAwB,CAC7C,8BAA8B,EAC9B,MAAM,CAAC,gBAAgB,CAAC;YACtB,IAAI,EAAE;gBACJ,KAAK,EAAE,SAAS;gBAChB,cAAc,EAAE,MAAM,CAAC,iBAAiB,EAAE;gBAC1C,kBAAkB,EAAE,IAAI;aACzB;SACF,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,GAAG,YAAY,EAAE,CAAC;QAExB,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,mBAAmB,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC7F,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QACrF,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC3F,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,kBAAkB,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC9F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/B,MAAM,GAAG,IAAI,QAAQ,CACnB,8BAA8B,EAC9B,gBAAgB,EAChB,QAAQ,EACR,SAAS,EACT,EAAE,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,cAAc,EAAE,EAC7D,OAAO,CACR,CAAC;QAEF,OAAO,CAAC,OAAO,CAAC,oBAAoB,EAAE,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAC;QACnE,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAE9C,OAAO,CAAC,OAAO,CAAC,oBAAoB,EAAE,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC;QAClE,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,EAAE,CAAC;QAC9C,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,MAAM,GAAG,YAAY,EAAE,CAAC;QACxB,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC;QAE7B,OAAO,CAAC,OAAO,CAAC,YAAY,EAAE,iBAAiB,CAAC,CAAC;QACjD,MAAM,EAAE,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAE1C,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,GAAG,YAAY,EAAE,CAAC;QACxB,OAAO,CAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,sBAAsB,EAAE,CAAC,CAAC;QACnF,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;QAE9B,OAAO,CAAC,OAAO,CAAC,YAAY,EAAE,sBAAsB,CAAC,CAAC;QACtD,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAC;QACvC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACtD,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC;QAErC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,mBAAmB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACpC,MAAM,GAAG,IAAI,QAAQ,CACnB,8BAA8B,EAC9B,gBAAgB,EAChB,QAAQ,EACR,SAAS,EACT,EAAE,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,mBAAmB,EAAE,EAClE,OAAO,CACR,CAAC;QACF,OAAO,CAAC,OAAO,CAAC,sBAAsB,EAAE;YACtC,IAAI,EAAE,iBAAiB;YACvB,MAAM,EAAE,iBAAiB;YACzB,SAAS,EAAE,KAAK;SACjB,CAAC,CAAC;QACH,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;QAE9B,OAAO,CAAC,OAAO,CAAC,YAAY,EAAE,sBAAsB,CAAC,CAAC;QACtD,MAAM,EAAE,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAE1C,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACtD,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAAC;YAC/C,IAAI,EAAE,iBAAiB;YACvB,MAAM,EAAE,iBAAiB;SAC1B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,GAAG,YAAY,EAAE,CAAC;QACxB,OAAO,CAAC,OAAO,CAAC,sBAAsB,EAAE;YACtC,IAAI,EAAE,2BAA2B;YACjC,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,IAAI;YACf,YAAY,EAAE,MAAM;SACrB,CAAC,CAAC;QACH,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;QAE9B,OAAO,CAAC,OAAO,CAAC,YAAY,EAAE,sBAAsB,CAAC,CAAC;QACtD,MAAM,EAAE,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACtD,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC;QAErC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,EAAE,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;QAC5F,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,GAAG,YAAY,EAAE,CAAC;QACxB,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;QAC9B,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,wCAAwC,CAAC,EAAE;YAC/E,IAAI,EAAE;gBACJ,IAAI,EAAE,wBAAwB;gBAC9B,SAAS,EAAE,IAAI;gBACf,YAAY,EAAE,KAAK;aACpB;SACF,CAAC,CAAC;QAEH,OAAO,CAAC,OAAO,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;QACxC,MAAM,EAAE,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACtD,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC;QAErC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,MAAM,mBAAmB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACpC,MAAM,GAAG,IAAI,QAAQ,CACnB,8BAA8B,EAC9B,gBAAgB,EAChB,QAAQ,EACR,SAAS,EACT,EAAE,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,mBAAmB,EAAE,EAClE,OAAO,CACR,CAAC;QACF,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;QAC9B,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,EAAE;YAC9D,IAAI,EAAE,EAAE,IAAI,EAAE,6BAA6B,EAAE,SAAS,EAAE,KAAK,EAAE;SAChE,CAAC,CAAC;QAEH,OAAO,CAAC,OAAO,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;QACxC,MAAM,EAAE,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAE1C,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACtD,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAAC;YAC/C,IAAI,EAAE,6BAA6B;YACnC,MAAM,EAAE,SAAS;SAClB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,GAAG,YAAY,EAAE,CAAC;QACxB,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;QAE9B,OAAO,CAAC,OAAO,CAAC,eAAe,EAAE,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC,CAAC;QACtF,MAAM,EAAE,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAE1C,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,GAAG,YAAY,EAAE,CAAC;QACxB,OAAO,CAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,MAAM,EAAE,sBAAsB,EAAE,CAAC,CAAC;QACvE,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;QAC9B,OAAO,CAAC,OAAO,CAAC,YAAY,EAAE,sBAAsB,CAAC,CAAC;QAEtD,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACrB,MAAM,EAAE,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAE1C,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACtD,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,cAAc,GAAG,SAAS,UAAU,EAAE,EAAE,CAAC;QAC/C,MAAM,IAAI,GAAG,sBAAsB,CAAC,cAAc,CAAC,CAAC;QACpD,MAAM,GAAG,IAAI,QAAQ,CACnB,8BAA8B,EAC9B,cAAc,EACd,MAAM,EACN,OAAO,EACP,EAAE,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,EAC7C,OAAO,CACR,CAAC;QACF,IAAI,CAAC;YACH,MAAM,CAAC,SAAS,CAAC;gBACf,IAAI,EAAE,kBAAkB;gBACxB,MAAM,EAAE;oBACN,OAAO,EAAE,IAAI;oBACb,OAAO,EAAE,cAAc;oBACvB,aAAa,EAAE,EAAE;oBACjB,gBAAgB,EAAE,EAAE;oBACpB,UAAU,EAAE,CAAC;iBACd;aACF,CAAC,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACrB,MAAM,SAAS,GAAG,wBAAwB,CAAC,EAAE,cAAc,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC;YACtF,MAAM,CAAC,SAAS,CAAC,YAAY,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,SAAS,CAAC,YAAY,EAAE,CAAC;YAC/B,SAAS,CAAC,UAAU,EAAE,CAAC;QACzB,CAAC;gBAAS,CAAC;YACT,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACrB,KAAK,MAAM,KAAK,IAAI,CAAC,IAAI,EAAE,oBAAoB,CAAC,IAAI,CAAC,EAAE,eAAe,CAAC,IAAI,CAAC,CAAC;gBAC3E,MAAM,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACpD,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/ws-client-heartbeat-volatile.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=ws-client-heartbeat-volatile.test.d.ts.map

package/dist/__tests__/ws-client-heartbeat-volatile.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-client-heartbeat-volatile.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/ws-client-heartbeat-volatile.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/ws-client-heartbeat-volatile.test.js

@@ -0,0 +1,63 @@
+import { AGENT_HEARTBEAT_INTERVAL_MS } from "@alan-ai/shared/agent-liveness";
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { isolateSandboxOutbox } from "./helpers/sandbox-outbox.js";
+isolateSandboxOutbox();
+// Fake socket.io-client socket whose `connect` handler and interval timers we can
+// drive, and whose reliable (`emit`) vs volatile (`volatile.emit`) channels are
+// separate spies — so we can assert WHICH channel each frame uses.
+const hoisted = vi.hoisted(() => {
+    const handlers = {};
+    const socket = {
+        connected: true,
+        id: "fake-socket",
+        io: { engine: {}, on: vi.fn() },
+        emit: vi.fn(),
+        volatile: { emit: vi.fn() },
+        on: vi.fn((event, cb) => {
+            handlers[event] = cb;
+        }),
+        once: vi.fn((event, cb) => {
+            handlers[event] = cb;
+        }),
+        disconnect: vi.fn(),
+        connect: vi.fn(),
+    };
+    return { socket, handlers, ioMock: vi.fn(() => socket) };
+});
+vi.mock("socket.io-client", () => ({
+    io: hoisted.ioMock,
+    Socket: class {
+    },
+}));
+import { WSClient } from "../ws-client.js";
+describe("WSClient heartbeat channel (head-of-line fix)", () => {
+    let client;
+    beforeEach(() => {
+        hoisted.socket.emit.mockClear();
+        hoisted.socket.volatile.emit.mockClear();
+        for (const key of Object.keys(hoisted.handlers))
+            delete hoisted.handlers[key];
+        hoisted.socket.connected = true;
+    });
+    afterEach(() => {
+        vi.useRealTimers();
+        client?.close();
+        client = undefined;
+    });
+    it("sends hello reliably but periodic heartbeats over the volatile channel", () => {
+        client = new WSClient("ws://fake", "conversation-hb", "task-hb", "token-hb", { onUserMessage: () => { }, onStop: () => { } }, "1.2.3");
+        // Fake ONLY interval timers so the heartbeat interval (created on connect) is
+        // drivable; the connect handler runs synchronously below.
+        vi.useFakeTimers({ toFake: ["setInterval", "clearInterval"] });
+        hoisted.handlers.connect?.();
+        // hello is reliable (must never be dropped — it establishes the marker).
+        expect(hoisted.socket.emit).toHaveBeenCalledWith("agent.hello", expect.objectContaining({ seq: 1 }));
+        expect(hoisted.socket.volatile.emit).not.toHaveBeenCalledWith("agent.hello", expect.anything());
+        // The periodic heartbeat is volatile: it must never queue behind bulk frames.
+        vi.advanceTimersByTime(AGENT_HEARTBEAT_INTERVAL_MS);
+        expect(hoisted.socket.volatile.emit).toHaveBeenCalledWith("agent.heartbeat", expect.objectContaining({ seq: 2 }));
+        // And the heartbeat must NOT also go out on the reliable channel.
+        expect(hoisted.socket.emit).not.toHaveBeenCalledWith("agent.heartbeat", expect.anything());
+    });
+});
+//# sourceMappingURL=ws-client-heartbeat-volatile.test.js.map

package/dist/__tests__/ws-client-heartbeat-volatile.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-client-heartbeat-volatile.test.js","sourceRoot":"","sources":["../../src/__tests__/ws-client-heartbeat-volatile.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,2BAA2B,EAAE,MAAM,gCAAgC,CAAC;AAC7E,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AAEnE,oBAAoB,EAAE,CAAC;AAEvB,kFAAkF;AAClF,gFAAgF;AAChF,mEAAmE;AACnE,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;IAC9B,MAAM,QAAQ,GAAiD,EAAE,CAAC;IAClE,MAAM,MAAM,GAAG;QACb,SAAS,EAAE,IAAI;QACf,EAAE,EAAE,aAAa;QACjB,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE;QAC/B,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE;QAC3B,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAAa,EAAE,EAAgC,EAAE,EAAE;YAC5D,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;QACvB,CAAC,CAAC;QACF,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAAa,EAAE,EAAgC,EAAE,EAAE;YAC9D,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;QACvB,CAAC,CAAC;QACF,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;QACnB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;KACjB,CAAC;IACF,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;AAC3D,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACjC,EAAE,EAAE,OAAO,CAAC,MAAM;IAClB,MAAM,EAAE;KAAQ;CACjB,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAE3C,QAAQ,CAAC,+CAA+C,EAAE,GAAG,EAAE;IAC7D,IAAI,MAA4B,CAAC;IAEjC,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QACzC,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;YAAE,OAAO,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;QAC9E,OAAO,CAAC,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,EAAE,KAAK,EAAE,CAAC;QAChB,MAAM,GAAG,SAAS,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;QAChF,MAAM,GAAG,IAAI,QAAQ,CACnB,WAAW,EACX,iBAAiB,EACjB,SAAS,EACT,UAAU,EACV,EAAE,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,EAC7C,OAAO,CACR,CAAC;QAEF,8EAA8E;QAC9E,0DAA0D;QAC1D,EAAE,CAAC,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC,aAAa,EAAE,eAAe,CAAC,EAAE,CAAC,CAAC;QAC/D,OAAO,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC;QAE7B,yEAAyE;QACzE,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC9C,aAAa,EACb,MAAM,CAAC,gBAAgB,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,CACpC,CAAC;QACF,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAAC,aAAa,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QAEhG,8EAA8E;QAC9E,EAAE,CAAC,mBAAmB,CAAC,2BAA2B,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,oBAAoB,CACvD,iBAAiB,EACjB,MAAM,CAAC,gBAAgB,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,CACpC,CAAC;QACF,kEAAkE;QAClE,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/ws-client-heartbeat.integration.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=ws-client-heartbeat.integration.test.d.ts.map

package/dist/__tests__/ws-client-heartbeat.integration.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-client-heartbeat.integration.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/ws-client-heartbeat.integration.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/ws-client-heartbeat.integration.test.js

@@ -0,0 +1,236 @@
+import { once } from "node:events";
+import { createServer } from "node:http";
+import { AGENT_HEARTBEAT_INTERVAL_MS, AGENT_LIVENESS_PROTOCOL_VERSION, } from "@alan-ai/shared/agent-liveness";
+import { Server as SocketServer } from "socket.io";
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { WSClient } from "../ws-client.js";
+import { isolateSandboxOutbox } from "./helpers/sandbox-outbox.js";
+isolateSandboxOutbox();
+const resourceIo = vi.hoisted(() => ({
+    readFile: vi.fn(),
+    collected: vi.fn(),
+}));
+vi.mock("../resource-diagnostics.js", async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        // Keep the real sampler and Linux parser. Only kernel I/O is controlled,
+        // so this exercises the same asynchronous collection on every host OS.
+        AgentResourceSampler: class extends actual.AgentResourceSampler {
+            constructor() {
+                super(async () => {
+                    const sample = await actual.collectAgentResourceDiagnostics({
+                        platform: "linux",
+                        readFile: resourceIo.readFile,
+                    });
+                    resourceIo.collected();
+                    return sample;
+                });
+            }
+        },
+    };
+});
+function nextEvent(socket, event) {
+    return new Promise((resolve) => socket.once(event, (payload) => resolve(payload)));
+}
+describe("WSClient sandbox liveness heartbeat", () => {
+    let io;
+    let httpServer;
+    let client;
+    beforeEach(() => {
+        vi.useRealTimers();
+        resourceIo.readFile.mockReset().mockResolvedValue(undefined);
+        resourceIo.collected.mockReset();
+    });
+    afterEach(async () => {
+        vi.useRealTimers();
+        client?.close();
+        client = undefined;
+        await new Promise((resolve) => (io ? io.close(() => resolve()) : resolve()));
+        await new Promise((resolve) => httpServer?.listening ? httpServer.close(() => resolve()) : resolve());
+        io = undefined;
+        httpServer = undefined;
+    });
+    it("emits hello on connect, heartbeats with advancing seq, and acks probes", async () => {
+        const files = {
+            "/proc/meminfo": "MemAvailable: 2048 kB\n",
+            "/proc/pressure/memory": "some avg10=12.34 avg60=0.00 avg300=0.00 total=100\n",
+            "/proc/self/cgroup": "0::/agent\n",
+            "/sys/fs/cgroup/agent/memory.events": "oom 3\noom_kill 2\n",
+        };
+        let finishReads;
+        const readsReleased = new Promise((resolve) => {
+            finishReads = resolve;
+        });
+        resourceIo.readFile.mockImplementation(async (path) => {
+            await readsReleased;
+            return files[path];
+        });
+        const collected = new Promise((resolve) => {
+            resourceIo.collected.mockImplementationOnce(resolve);
+        });
+        httpServer = createServer();
+        io = new SocketServer(httpServer, { cors: { origin: "*" } });
+        httpServer.listen(0, "127.0.0.1");
+        await once(httpServer, "listening");
+        const address = httpServer.address();
+        if (!address || typeof address === "string")
+            throw new Error("no TCP port");
+        const wsUrl = `ws://127.0.0.1:${address.port}`;
+        // Fake ONLY interval timers (leave setTimeout real so the socket handshake +
+        // server .timeout() acks still work). Enabled before the client connects so
+        // the heartbeat interval created in the connect handler is a fake timer we
+        // can drive with advanceTimersByTime — otherwise a 15s real wait is needed.
+        vi.useFakeTimers({ toFake: ["setInterval", "clearInterval"] });
+        // Attach the hello listener synchronously inside the connection handler so a
+        // hello emitted immediately on connect is never dropped before we listen.
+        const connectionPromise = new Promise((resolve) => {
+            io.once("connection", (socket) => {
+                socket.once("agent.hello", (hello) => resolve({ socket, hello }));
+            });
+        });
+        let runIsActive = false;
+        client = new WSClient(wsUrl, "conversation-hb", "task-hb", "session-token-hb", {
+            onUserMessage: () => { },
+            onStop: () => { },
+            getRunState: () => runIsActive
+                ? { idle: false, activeRunIds: ["run-hb"] }
+                : { idle: true, activeRunIds: [] },
+        }, "9.9.9");
+        // 1. hello on connect — seq 1, protocol marker, idle run state.
+        const { socket: serverSocket, hello } = await connectionPromise;
+        expect(hello.protocolVersion).toBe(AGENT_LIVENESS_PROTOCOL_VERSION);
+        expect(hello.seq).toBe(1);
+        expect(hello.agentSessionId).toMatch(/[0-9a-f-]{36}/);
+        expect(hello.runState).toEqual({ idle: true, activeRunIds: [] });
+        expect(hello.agentVersion).toBe("9.9.9");
+        expect(hello.resources).toBeUndefined();
+        // 2. Advance the fake interval to fire heartbeats with an advancing seq.
+        runIsActive = true;
+        const hb2Promise = nextEvent(serverSocket, "agent.heartbeat");
+        vi.advanceTimersByTime(AGENT_HEARTBEAT_INTERVAL_MS);
+        const hb2 = await hb2Promise;
+        expect(hb2.seq).toBe(2);
+        expect(hb2.agentSessionId).toBe(hello.agentSessionId);
+        expect(hb2.runState).toEqual({ idle: false, activeRunIds: ["run-hb"] });
+        // Slow /proc reads cannot delay or suppress the heartbeat.
+        expect(hb2.resources).toBeUndefined();
+        expect(resourceIo.readFile).toHaveBeenCalledWith("/proc/meminfo");
+        finishReads();
+        await collected;
+        // A real socket round trip gives the sampler's collection continuation a
+        // turn to cache the completed sample; no guessed wall-clock sleep is used.
+        const ack = await serverSocket
+            .timeout(2_000)
+            .emitWithAck("agent.probe", {})
+            .then((res) => res);
+        expect(ack.agentSessionId).toBe(hello.agentSessionId);
+        expect(ack.protocolVersion).toBe(AGENT_LIVENESS_PROTOCOL_VERSION);
+        const hb3Promise = nextEvent(serverSocket, "agent.heartbeat");
+        vi.advanceTimersByTime(AGENT_HEARTBEAT_INTERVAL_MS);
+        const hb3 = await hb3Promise;
+        expect(hb3.seq).toBe(3);
+        expect(hb3.resources).toMatchObject({
+            memoryAvailableBytes: 2097152,
+            memoryPressureSomeAvg10: 12.34,
+            cgroupMemoryOomEvents: 3,
+            cgroupMemoryOomKillEvents: 2,
+        });
+        expect(hb3.resources?.sampledAtMs).toBeGreaterThan(0);
+        expect(hb3.resources?.memoryTotalBytes).toBeGreaterThan(0);
+        expect(hb3.resources?.processRssBytes).toBeGreaterThan(0);
+        vi.useRealTimers();
+    }, 20_000);
+    it("does not accept an Alan MCP registration on the user_message payload", async () => {
+        httpServer = createServer();
+        io = new SocketServer(httpServer, { cors: { origin: "*" } });
+        httpServer.listen(0, "127.0.0.1");
+        await once(httpServer, "listening");
+        const address = httpServer.address();
+        if (!address || typeof address === "string")
+            throw new Error("no TCP port");
+        const connectionPromise = new Promise((resolve) => {
+            io.once("connection", resolve);
+        });
+        const messagePromise = new Promise((resolve) => {
+            client = new WSClient(`ws://127.0.0.1:${address.port}`, "conversation-mcp", "task-mcp", "session-token-mcp", {
+                onUserMessage: (payload) => resolve(payload),
+                onStop: () => { },
+            });
+        });
+        const serverSocket = await connectionPromise;
+        const alanMcp = {
+            url: "https://api.alan.test/mcp",
+            headers: {
+                Authorization: "Bearer alan_runtime_credential",
+                "x-alan-run-id": "run-1",
+            },
+        };
+        serverSocket.emit("user_message", { text: "mcp payload", alanMcp });
+        const payload = await messagePromise;
+        expect(payload).not.toHaveProperty("alanMcp");
+    });
+    // AL-3926 audit B1: the payload is rebuilt from an explicit field list, and
+    // the per-turn provider key and terminal context were missing from it. The
+    // sandbox then ran every turn on the key frozen at provision time.
+    it("forwards the per-turn provider key and terminal context on the user_message payload", async () => {
+        httpServer = createServer();
+        io = new SocketServer(httpServer, { cors: { origin: "*" } });
+        httpServer.listen(0, "127.0.0.1");
+        await once(httpServer, "listening");
+        const address = httpServer.address();
+        if (!address || typeof address === "string")
+            throw new Error("no TCP port");
+        const connectionPromise = new Promise((resolve) => {
+            io.once("connection", resolve);
+        });
+        const messagePromise = new Promise((resolve) => {
+            client = new WSClient(`ws://127.0.0.1:${address.port}`, "conversation-key", "task-key", "session-token-key", {
+                onUserMessage: (payload) => resolve(payload),
+                onStop: () => { },
+            });
+        });
+        const serverSocket = await connectionPromise;
+        const terminalContextRefs = [{ id: "term-1", kind: "terminal" }];
+        const terminalSnapshots = [{ id: "term-1", content: "$ pnpm test" }];
+        const externalMcpServers = { signoz: { type: "http", url: "https://mcp.signoz.io/mcp" } };
+        serverSocket.emit("user_message", {
+            text: "per-turn credentials",
+            providerApiKey: "sk-turn-scoped",
+            terminalContextRefs,
+            terminalSnapshots,
+            externalMcpServers,
+        });
+        await expect(messagePromise).resolves.toEqual(expect.objectContaining({
+            providerApiKey: "sk-turn-scoped",
+            terminalContextRefs,
+            terminalSnapshots,
+            externalMcpServers,
+        }));
+    });
+    // AL-3926 audit B8: the watchdog's abort for a failed cloud run went to the
+    // agent socket, which never listened for it.
+    it("delivers a server-initiated agent.abort with its run id and reason", async () => {
+        httpServer = createServer();
+        io = new SocketServer(httpServer, { cors: { origin: "*" } });
+        httpServer.listen(0, "127.0.0.1");
+        await once(httpServer, "listening");
+        const address = httpServer.address();
+        if (!address || typeof address === "string")
+            throw new Error("no TCP port");
+        const connectionPromise = new Promise((resolve) => {
+            io.once("connection", resolve);
+        });
+        const abortPromise = new Promise((resolve) => {
+            client = new WSClient(`ws://127.0.0.1:${address.port}`, "conversation-abort", "task-abort", "session-token-abort", {
+                onUserMessage: () => { },
+                onStop: () => { },
+                onAbort: (payload) => resolve(payload),
+            });
+        });
+        const serverSocket = await connectionPromise;
+        serverSocket.emit("agent.abort", { runId: "run-42", reason: "watchdog_timeout" });
+        await expect(abortPromise).resolves.toEqual({ runId: "run-42", reason: "watchdog_timeout" });
+    });
+});
+//# sourceMappingURL=ws-client-heartbeat.integration.test.js.map