tests/unit/cli/setup-claude.test.ts

import { test } from "node:test";
import assert from "node:assert/strict";
import { buildProfileSettings } from "../../../bin/cli/commands/setup-claude.mjs";
import { buildClaudeEnv, resolveLaunchTarget } from "../../../bin/cli/commands/launch.mjs";
import { categoriseModel } from "../../../bin/cli/commands/setup-codex.mjs";

// ── setup-claude profile generation ──────────────────────────────────────────

test("buildProfileSettings pins the model + base URL + gateway discovery", () => {
  const cfg = categoriseModel("glm/glm-5.2"); // thinking → effort xhigh
  const json = JSON.parse(buildProfileSettings("glm/glm-5.2", "http://vps:20128", cfg));
  assert.equal(json.model, "glm/glm-5.2");
  assert.equal(json.env.ANTHROPIC_BASE_URL, "http://vps:20128");
  assert.equal(json.env.ANTHROPIC_MODEL, "glm/glm-5.2");
  assert.equal(json.env.CLAUDE_CODE_ENABLE_GATEWAY_MODEL_DISCOVERY, "1");
  assert.equal(json.effortLevel, "xhigh");
});

test("buildProfileSettings NEVER writes the auth token to disk", () => {
  const cfg = categoriseModel("kmc/kimi-k2.7");
  const raw = buildProfileSettings("kmc/kimi-k2.7", "http://vps:20128", cfg);
  assert.equal(raw.includes("ANTHROPIC_AUTH_TOKEN"), false);
  assert.equal(raw.includes("ANTHROPIC_API_KEY"), false);
});

test("buildProfileSettings omits effortLevel for the simple tier", () => {
  const cfg = categoriseModel("ollamacloud/gemma4:31b"); // simple → no effort
  const json = JSON.parse(buildProfileSettings("ollamacloud/gemma4:31b", "http://x:20128", cfg));
  assert.equal("effortLevel" in json, false);
});

test("profile names match setup-codex (cross-CLI consistency)", () => {
  assert.equal(categoriseModel("glm/glm-5.2").name, "glm52");
  assert.equal(categoriseModel("kmc/kimi-k2.7").name, "kimi-k27");
});

// ── launch env (Claude Code) ─────────────────────────────────────────────────

test("buildClaudeEnv still accepts a bare port (backward compatible)", () => {
  const env = buildClaudeEnv({ PATH: "/bin" }, 20128, "secret");
  assert.equal(env.ANTHROPIC_BASE_URL, "http://localhost:20128");
  assert.equal(env.ANTHROPIC_AUTH_TOKEN, "secret");
  assert.equal(env.CLAUDE_CODE_ENABLE_GATEWAY_MODEL_DISCOVERY, "1");
});

test("buildClaudeEnv accepts a full base URL and strips /v1", () => {
  const env = buildClaudeEnv({}, "https://vps.example.com:20128/v1", "t");
  assert.equal(env.ANTHROPIC_BASE_URL, "https://vps.example.com:20128");
});

test("buildClaudeEnv sets CLAUDE_CONFIG_DIR for a profile", () => {
  const env = buildClaudeEnv({}, 20128, "t", { configDir: "/home/u/.claude/profiles/glm52" });
  assert.equal(env.CLAUDE_CONFIG_DIR, "/home/u/.claude/profiles/glm52");
});

test("buildClaudeEnv strips inherited ANTHROPIC_* and does not mutate input", () => {
  const input = { ANTHROPIC_API_KEY: "leak", PATH: "/bin" };
  const env = buildClaudeEnv(input, 20128, "x");
  assert.equal(env.ANTHROPIC_API_KEY, undefined);
  assert.equal(input.ANTHROPIC_API_KEY, "leak");
});

test("resolveLaunchTarget: explicit --remote wins, strips /v1", () => {
  const { baseUrl } = resolveLaunchTarget({ remote: "https://vps:20128/v1" });
  assert.equal(baseUrl, "https://vps:20128");
});

test("resolveLaunchTarget: explicit token wins over everything", () => {
  const { authToken } = resolveLaunchTarget({ remote: "http://x:20128", token: "tok-explicit" });
  assert.equal(authToken, "tok-explicit");
});