import { EventEmitter } from "node:events";

import { beforeEach, describe, expect, it, vi } from "vitest";

import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";

import type { WebSocketServer } from "ws";

import type { ResolvedGatewayAuth } from "../auth.js";

attachGatewayWsMessageHandlerMock.mockReset();

});

afterEach(() => {

vi.useRealTimers();

});

it("threads current auth getters into the handshake handler instead of a stale snapshot", () => {

const listeners = new Map<string, (...args: unknown[]) => void>();

const wss = {

port: 19001,

canvasHostEnabled: false,

resolvedAuth: createResolvedAuth("token"),

preauthHandshakeTimeoutMs: 60_000,

gatewayMethods: [],

events: [],

refreshHealthSnapshot: vi.fn(),

expect(registered).toBe(false);

expect(clients.size).toBe(0);

});

it("sends protocol pings until the connection closes", () => {

vi.useFakeTimers();

const listeners = new Map<string, (...args: unknown[]) => void>();

const wss = {

on: vi.fn((event: string, handler: (...args: unknown[]) => void) => {

listeners.set(event, handler);

}),

} as unknown as WebSocketServer;

const socket = Object.assign(new EventEmitter(), {

_socket: {

remoteAddress: "127.0.0.1",

remotePort: 1234,

localAddress: "127.0.0.1",

localPort: 5678,

},

send: vi.fn(),

ping: vi.fn(),

close: vi.fn(),

});

const upgradeReq = {

headers: { host: "127.0.0.1:19001" },

socket: { localAddress: "127.0.0.1" },

};

attachGatewayWsConnectionHandler({

wss,

clients: new Set(),

preauthConnectionBudget: { release: vi.fn() } as never,

port: 19001,

canvasHostEnabled: false,

resolvedAuth: createResolvedAuth("token"),

preauthHandshakeTimeoutMs: 60_000,

gatewayMethods: [],

events: [],

refreshHealthSnapshot: vi.fn(),

logGateway: createLogger() as never,

logHealth: createLogger() as never,

logWsControl: createLogger() as never,

extraHandlers: {},

broadcast: vi.fn(),

buildRequestContext: () =>

({

unsubscribeAllSessionEvents: vi.fn(),

nodeRegistry: { unregister: vi.fn() },

nodeUnsubscribeAll: vi.fn(),

}) as never,

});

const onConnection = listeners.get("connection");

expect(onConnection).toBeTypeOf("function");

onConnection?.(socket, upgradeReq);

const passed = attachGatewayWsMessageHandlerMock.mock.calls[0]?.[0] as {

setClient: (client: unknown) => boolean;

};

expect(

passed.setClient({

socket,

connect: { client: { id: "openclaw-control-ui", mode: "webchat" } },

connId: "ping-client",

usesSharedGatewayAuth: false,

}),

).toBe(true);

vi.advanceTimersByTime(25_000);

expect(socket.ping).toHaveBeenCalledTimes(1);

socket.emit("close", 1000, Buffer.from("done"));

vi.advanceTimersByTime(25_000);

expect(socket.ping).toHaveBeenCalledTimes(1);

});

});