Files
diegosouzapw--omniroute/tests/unit/cli/setup-opencode.test.ts
T
2026-07-13 13:39:12 +08:00

62 lines
2.4 KiB
TypeScript

import { test } from "node:test";
import assert from "node:assert/strict";
import {
postProcessOpencodeConfig,
resolveOpencodeTarget,
} from "../../../bin/cli/commands/setup-opencode.mjs";
const RAW = JSON.stringify({
$schema: "https://opencode.ai/config.json",
provider: {
omniroute: {
name: "OmniRoute",
npm: "@ai-sdk/openai-compatible",
options: { baseURL: "http://vps:20128/v1", apiKey: "sk-secret-literal" },
models: {
"glm/glm-5.2": { name: "glm/glm-5.2", limit: { context: 131072, output: 32768 } },
"kmc/kimi-k2.7": { name: "kmc/kimi-k2.7", limit: { context: 131072 } },
"openai/gpt-4o": { name: "openai/gpt-4o", limit: { context: 128000 } },
},
},
},
});
test("postProcessOpencodeConfig replaces the literal API key with an env ref (no secret on disk)", () => {
const { json } = postProcessOpencodeConfig(RAW);
assert.equal(json.includes("sk-secret-literal"), false);
const cfg = JSON.parse(json);
assert.equal(cfg.provider.omniroute.options.apiKey, "{env:OMNIROUTE_API_KEY}");
assert.equal(cfg.provider.omniroute.options.baseURL, "http://vps:20128/v1");
});
test("postProcessOpencodeConfig keeps all models by default", () => {
const { modelCount } = postProcessOpencodeConfig(RAW);
assert.equal(modelCount, 3);
});
test("postProcessOpencodeConfig --only filters the model map by substring", () => {
const { json, modelCount } = postProcessOpencodeConfig(RAW, { only: ["glm", "kimi"] });
const cfg = JSON.parse(json);
assert.equal(modelCount, 2);
assert.ok(cfg.provider.omniroute.models["glm/glm-5.2"]);
assert.ok(cfg.provider.omniroute.models["kmc/kimi-k2.7"]);
assert.equal("openai/gpt-4o" in cfg.provider.omniroute.models, false);
});
test("postProcessOpencodeConfig preserves $schema, provider name and npm", () => {
const { json } = postProcessOpencodeConfig(RAW);
const cfg = JSON.parse(json);
assert.equal(cfg.$schema, "https://opencode.ai/config.json");
assert.equal(cfg.provider.omniroute.npm, "@ai-sdk/openai-compatible");
});
test("resolveOpencodeTarget: --remote wins and trailing slashes are trimmed", () => {
const { baseUrl } = resolveOpencodeTarget({ remote: "http://vps:20128/" });
assert.equal(baseUrl, "http://vps:20128");
});
test("resolveOpencodeTarget: explicit --api-key wins", () => {
const { apiKey } = resolveOpencodeTarget({ remote: "http://x:20128", apiKey: "sk-explicit" });
assert.equal(apiKey, "sk-explicit");
});