101 lines
3.1 KiB
TypeScript
101 lines
3.1 KiB
TypeScript
import fs from "node:fs/promises";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
|
|
import { afterEach, describe, expect, it, vi } from "vitest";
|
|
|
|
import type { RuntimeEnv } from "../runtime.js";
|
|
import type { WizardPrompter } from "../wizard/prompts.js";
|
|
import { applyAuthChoice } from "./auth-choice.js";
|
|
|
|
const noopAsync = async () => {};
|
|
const noop = () => {};
|
|
|
|
describe("applyAuthChoice", () => {
|
|
const previousStateDir = process.env.CLAWDBOT_STATE_DIR;
|
|
const previousAgentDir = process.env.CLAWDBOT_AGENT_DIR;
|
|
const previousPiAgentDir = process.env.PI_CODING_AGENT_DIR;
|
|
let tempStateDir: string | null = null;
|
|
|
|
afterEach(async () => {
|
|
if (tempStateDir) {
|
|
await fs.rm(tempStateDir, { recursive: true, force: true });
|
|
tempStateDir = null;
|
|
}
|
|
if (previousStateDir === undefined) {
|
|
delete process.env.CLAWDBOT_STATE_DIR;
|
|
} else {
|
|
process.env.CLAWDBOT_STATE_DIR = previousStateDir;
|
|
}
|
|
if (previousAgentDir === undefined) {
|
|
delete process.env.CLAWDBOT_AGENT_DIR;
|
|
} else {
|
|
process.env.CLAWDBOT_AGENT_DIR = previousAgentDir;
|
|
}
|
|
if (previousPiAgentDir === undefined) {
|
|
delete process.env.PI_CODING_AGENT_DIR;
|
|
} else {
|
|
process.env.PI_CODING_AGENT_DIR = previousPiAgentDir;
|
|
}
|
|
});
|
|
|
|
it("prompts and writes MiniMax API key when selecting minimax-api", async () => {
|
|
tempStateDir = await fs.mkdtemp(path.join(os.tmpdir(), "clawdbot-auth-"));
|
|
process.env.CLAWDBOT_STATE_DIR = tempStateDir;
|
|
process.env.CLAWDBOT_AGENT_DIR = path.join(tempStateDir, "agent");
|
|
process.env.PI_CODING_AGENT_DIR = process.env.CLAWDBOT_AGENT_DIR;
|
|
|
|
const text = vi.fn().mockResolvedValue("sk-minimax-test");
|
|
const select: WizardPrompter["select"] = vi.fn(
|
|
async (params) => params.options[0]?.value as never,
|
|
);
|
|
const multiselect: WizardPrompter["multiselect"] = vi.fn(async () => []);
|
|
const prompter: WizardPrompter = {
|
|
intro: vi.fn(noopAsync),
|
|
outro: vi.fn(noopAsync),
|
|
note: vi.fn(noopAsync),
|
|
select,
|
|
multiselect,
|
|
text,
|
|
confirm: vi.fn(async () => false),
|
|
progress: vi.fn(() => ({ update: noop, stop: noop })),
|
|
};
|
|
const runtime: RuntimeEnv = {
|
|
log: vi.fn(),
|
|
error: vi.fn(),
|
|
exit: vi.fn((code: number) => {
|
|
throw new Error(`exit:${code}`);
|
|
}),
|
|
};
|
|
|
|
const result = await applyAuthChoice({
|
|
authChoice: "minimax-api",
|
|
config: {},
|
|
prompter,
|
|
runtime,
|
|
setDefaultModel: true,
|
|
});
|
|
|
|
expect(text).toHaveBeenCalledWith(
|
|
expect.objectContaining({ message: "Enter MiniMax API key" }),
|
|
);
|
|
expect(result.config.auth?.profiles?.["minimax:default"]).toMatchObject({
|
|
provider: "minimax",
|
|
mode: "api_key",
|
|
});
|
|
|
|
const authProfilePath = path.join(
|
|
tempStateDir,
|
|
"agents",
|
|
"main",
|
|
"agent",
|
|
"auth-profiles.json",
|
|
);
|
|
const raw = await fs.readFile(authProfilePath, "utf8");
|
|
const parsed = JSON.parse(raw) as {
|
|
profiles?: Record<string, { key?: string }>;
|
|
};
|
|
expect(parsed.profiles?.["minimax:default"]?.key).toBe("sk-minimax-test");
|
|
});
|
|
});
|