} from "./chat.abort.test-helpers.js";

import { chatHandlers } from "./chat.js";

type AbortResponsePayload = {

aborted?: boolean;

runIds?: string[];

};

async function invokeSingleRunAbort({

context,

runId = "run-1",

const [ok, payload, error] = respond.mock.calls.at(-1) ?? [];

expect(ok).toBe(false);

expect(payload).toBeUndefined();

expect(error).toMatchObject({ code: "INVALID_REQUEST", message: "unauthorized" });

expect(error?.code).toBe("INVALID_REQUEST");

expect(error?.message).toBe("unauthorized");

expect(context.chatAbortControllers.has("run-1")).toBe(true);

});

const [ok, payload] = respond.mock.calls.at(-1) ?? [];

expect(ok).toBe(true);

expect(payload).toMatchObject({ aborted: true, runIds: ["run-1"] });

const abortPayload = payload as AbortResponsePayload | undefined;

expect(abortPayload?.aborted).toBe(true);

expect(abortPayload?.runIds).toEqual(["run-1"]);

expect(context.chatAbortControllers.has("run-1")).toBe(false);

});

const [ok, payload] = respond.mock.calls.at(-1) ?? [];

expect(ok).toBe(true);

expect(payload).toMatchObject({ aborted: true, runIds: ["run-mine"] });

const abortPayload = payload as AbortResponsePayload | undefined;

expect(abortPayload?.aborted).toBe(true);

expect(abortPayload?.runIds).toEqual(["run-mine"]);

expect(context.chatAbortControllers.has("run-mine")).toBe(false);

expect(context.chatAbortControllers.has("run-other")).toBe(true);

});

const [ok, payload] = respond.mock.calls.at(-1) ?? [];

expect(ok).toBe(true);

expect(payload).toMatchObject({ aborted: true, runIds: ["run-1"] });

const abortPayload = payload as AbortResponsePayload | undefined;

expect(abortPayload?.aborted).toBe(true);

expect(abortPayload?.runIds).toEqual(["run-1"]);

});

});