123 lines
3.5 KiB
TypeScript
123 lines
3.5 KiB
TypeScript
import { describe, expect, it, vi } from "vitest";
|
|
import type { RuntimeEnv } from "../runtime.js";
|
|
import type { WizardPrompter } from "./prompts.js";
|
|
|
|
const mocks = vi.hoisted(() => ({
|
|
randomToken: vi.fn(),
|
|
}));
|
|
|
|
vi.mock("../commands/onboard-helpers.js", async (importActual) => {
|
|
const actual = await importActual<typeof import("../commands/onboard-helpers.js")>();
|
|
return {
|
|
...actual,
|
|
randomToken: mocks.randomToken,
|
|
};
|
|
});
|
|
|
|
vi.mock("../infra/tailscale.js", () => ({
|
|
findTailscaleBinary: vi.fn(async () => undefined),
|
|
}));
|
|
|
|
import { configureGatewayForOnboarding } from "./onboarding.gateway-config.js";
|
|
|
|
describe("configureGatewayForOnboarding", () => {
|
|
function createPrompter(params: { selectQueue: string[]; textQueue: Array<string | undefined> }) {
|
|
const selectQueue = [...params.selectQueue];
|
|
const textQueue = [...params.textQueue];
|
|
|
|
return {
|
|
intro: vi.fn(async () => {}),
|
|
outro: vi.fn(async () => {}),
|
|
note: vi.fn(async () => {}),
|
|
select: vi.fn(async () => selectQueue.shift() as string),
|
|
multiselect: vi.fn(async () => []),
|
|
text: vi.fn(async () => textQueue.shift() as string),
|
|
confirm: vi.fn(async () => false),
|
|
progress: vi.fn(() => ({ update: vi.fn(), stop: vi.fn() })),
|
|
} satisfies WizardPrompter;
|
|
}
|
|
|
|
function createRuntime(): RuntimeEnv {
|
|
return {
|
|
log: vi.fn(),
|
|
error: vi.fn(),
|
|
exit: vi.fn(),
|
|
};
|
|
}
|
|
|
|
it("generates a token when the prompt returns undefined", async () => {
|
|
mocks.randomToken.mockReturnValue("generated-token");
|
|
|
|
const prompter = createPrompter({
|
|
selectQueue: ["loopback", "token", "off"],
|
|
textQueue: ["18789", undefined],
|
|
});
|
|
const runtime = createRuntime();
|
|
|
|
const result = await configureGatewayForOnboarding({
|
|
flow: "advanced",
|
|
baseConfig: {},
|
|
nextConfig: {},
|
|
localPort: 18789,
|
|
quickstartGateway: {
|
|
hasExisting: false,
|
|
port: 18789,
|
|
bind: "loopback",
|
|
authMode: "token",
|
|
tailscaleMode: "off",
|
|
token: undefined,
|
|
password: undefined,
|
|
customBindHost: undefined,
|
|
tailscaleResetOnExit: false,
|
|
},
|
|
prompter,
|
|
runtime,
|
|
});
|
|
|
|
expect(result.settings.gatewayToken).toBe("generated-token");
|
|
expect(result.nextConfig.gateway?.nodes?.denyCommands).toEqual([
|
|
"camera.snap",
|
|
"camera.clip",
|
|
"screen.record",
|
|
"calendar.add",
|
|
"contacts.add",
|
|
"reminders.add",
|
|
]);
|
|
});
|
|
it("does not set password to literal 'undefined' when prompt returns undefined", async () => {
|
|
mocks.randomToken.mockReturnValue("unused");
|
|
|
|
// Flow: loopback bind → password auth → tailscale off
|
|
const prompter = createPrompter({
|
|
selectQueue: ["loopback", "password", "off"],
|
|
textQueue: ["18789", undefined],
|
|
});
|
|
const runtime = createRuntime();
|
|
|
|
const result = await configureGatewayForOnboarding({
|
|
flow: "advanced",
|
|
baseConfig: {},
|
|
nextConfig: {},
|
|
localPort: 18789,
|
|
quickstartGateway: {
|
|
hasExisting: false,
|
|
port: 18789,
|
|
bind: "loopback",
|
|
authMode: "password",
|
|
tailscaleMode: "off",
|
|
token: undefined,
|
|
password: undefined,
|
|
customBindHost: undefined,
|
|
tailscaleResetOnExit: false,
|
|
},
|
|
prompter,
|
|
runtime,
|
|
});
|
|
|
|
const authConfig = result.nextConfig.gateway?.auth as { mode?: string; password?: string };
|
|
expect(authConfig?.mode).toBe("password");
|
|
expect(authConfig?.password).toBe("");
|
|
expect(authConfig?.password).not.toBe("undefined");
|
|
});
|
|
});
|