watchClientDisconnect,

writeDone,

} from "./http-common.js";

import { makeMockHttpResponse } from "./test-http-response.js";

import { makeMockHttpReqRes, makeMockHttpResponse } from "./test-http-response.js";

const readJsonBodyMock = vi.hoisted(() => vi.fn());

});

describe("watchClientDisconnect", () => {

function buildReqRes(

reqSocket: EventEmitter | null,

resSocket: EventEmitter | null,

): { req: IncomingMessage; res: ServerResponse } {

return {

req: { socket: reqSocket } as unknown as IncomingMessage,

res: { socket: resSocket } as unknown as ServerResponse,

};

}

it("returns a no-op cleanup when no sockets are available", () => {

const { req, res } = buildReqRes(null, null);

const { req, res } = makeMockHttpReqRes(null, null);

const controller = new AbortController();

const cleanup = watchClientDisconnect(req, res, controller);

cleanup();

it("aborts the controller and calls onDisconnect when a socket closes", () => {

const socket = new EventEmitter();

const { req, res } = buildReqRes(socket, socket);

const { req, res } = makeMockHttpReqRes(socket, socket);

const controller = new AbortController();

const onDisconnect = vi.fn();

watchClientDisconnect(req, res, controller, onDisconnect);

it("does not double-abort when the controller is already aborted", () => {

const socket = new EventEmitter();

const { req, res } = buildReqRes(socket, null);

const { req, res } = makeMockHttpReqRes(socket, null);

const controller = new AbortController();

controller.abort();

const abortSpy = vi.spyOn(controller, "abort");

it("works without an onDisconnect callback", () => {

const socket = new EventEmitter();

const { req, res } = buildReqRes(null, socket);

const { req, res } = makeMockHttpReqRes(null, socket);

const controller = new AbortController();

watchClientDisconnect(req, res, controller);

socket.emit("close");

it("deduplicates identical request and response sockets", () => {

const socket = new EventEmitter();

const onSpy = vi.spyOn(socket, "on");

const { req, res } = buildReqRes(socket, socket);

const { req, res } = makeMockHttpReqRes(socket, socket);

const controller = new AbortController();

watchClientDisconnect(req, res, controller);

expect(onSpy).toHaveBeenCalledTimes(1);

const resSocket = new EventEmitter();

const reqOn = vi.spyOn(reqSocket, "on");

const resOn = vi.spyOn(resSocket, "on");

const { req, res } = buildReqRes(reqSocket, resSocket);

const { req, res } = makeMockHttpReqRes(reqSocket, resSocket);

const controller = new AbortController();

watchClientDisconnect(req, res, controller);

const reqOnCall = mockCallRecord(reqOn, 0);

it("cleanup detaches the close listener from each socket", () => {

const socket = new EventEmitter();

const { req, res } = buildReqRes(socket, null);

const { req, res } = makeMockHttpReqRes(socket, null);

const controller = new AbortController();

const cleanup = watchClientDisconnect(req, res, controller);

expect(socket.listenerCount("close")).toBe(1);