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

import type { resolveGatewayConnectionAuth } from "./connection-auth.js";

type AuthResolutionParams = Parameters<typeof resolveGatewayConnectionAuth>[0];

const mockState = vi.hoisted(() => ({

buildGatewayConnectionDetails: vi.fn(),

const { resolveGatewayClientBootstrap, resolveGatewayUrlOverrideSource } =

await import("./client-bootstrap.js");

function expectLastAuthResolutionParams(expected: {

urlOverride?: string;

urlOverrideSource?: "cli" | "env";

}) {

const [params] = mockState.resolveGatewayConnectionAuth.mock.calls.at(-1) ?? [];

if (params === undefined) {

throw new Error("Expected shared auth resolution to be called");

}

const authParams = params as AuthResolutionParams;

expect(authParams.env).toBe(process.env);

expect(authParams.urlOverride).toBe(expected.urlOverride);

expect(authParams.urlOverrideSource).toBe(expected.urlOverrideSource);

}

describe("resolveGatewayUrlOverrideSource", () => {

it("maps override url sources only", () => {

expect(resolveGatewayUrlOverrideSource("cli --url")).toBe("cli");

password: undefined,

},

});

expect(mockState.resolveGatewayConnectionAuth).toHaveBeenCalledWith(

expect.objectContaining({

env: process.env,

urlOverride: "wss://override.example/ws",

urlOverrideSource: "cli",

}),

);

expectLastAuthResolutionParams({

urlOverride: "wss://override.example/ws",

urlOverrideSource: "cli",

});

});

it("does not mark config-derived urls as overrides", async () => {

env: process.env,

});

expect(mockState.resolveGatewayConnectionAuth).toHaveBeenCalledWith(

expect.objectContaining({

env: process.env,

urlOverride: undefined,

urlOverrideSource: undefined,

}),

);

expectLastAuthResolutionParams({

urlOverride: undefined,

urlOverrideSource: undefined,

});

});

it("carries configured preauth handshake timeout for GatewayClient callers", async () => {