wehub-resource-sync
66 lines
2.2 KiB
TypeScript
66 lines
2.2 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import {
|
|
buildAliasEnvVar,
|
|
buildCustomCliEnvScript,
|
|
buildCustomCliJsonConfig,
|
|
normalizeOpenAiBaseUrl,
|
|
} from "../../src/app/(dashboard)/dashboard/cli-code/components/customCliConfig.ts";
|
|
|
|
test("normalizeOpenAiBaseUrl appends /v1 only when needed", () => {
|
|
assert.equal(normalizeOpenAiBaseUrl("http://localhost:20128"), "http://localhost:20128/v1");
|
|
assert.equal(
|
|
normalizeOpenAiBaseUrl("https://example.com/proxy/v1"),
|
|
"https://example.com/proxy/v1"
|
|
);
|
|
});
|
|
|
|
test("buildAliasEnvVar sanitizes aliases for env variable export", () => {
|
|
assert.equal(buildAliasEnvVar("review"), "OMNIROUTE_MODEL_REVIEW");
|
|
assert.equal(buildAliasEnvVar("plan mode"), "OMNIROUTE_MODEL_PLAN_MODE");
|
|
assert.equal(buildAliasEnvVar(""), null);
|
|
});
|
|
|
|
test("custom CLI generators include default model and alias mappings", () => {
|
|
const envScript = buildCustomCliEnvScript({
|
|
cliName: "My Team CLI",
|
|
baseUrl: "http://localhost:20128",
|
|
apiKey: "sk_test_123",
|
|
defaultModel: "omniroute/fast",
|
|
aliasMappings: [
|
|
{ alias: "review", model: "cc/claude-sonnet-4-5-20250929" },
|
|
{ alias: "vision", model: "gemini/gemini-3-flash" },
|
|
],
|
|
});
|
|
|
|
assert.match(envScript, /export OPENAI_BASE_URL="http:\/\/localhost:20128\/v1"/);
|
|
assert.match(envScript, /export OPENAI_MODEL="omniroute\/fast"/);
|
|
assert.match(envScript, /export OMNIROUTE_MODEL_REVIEW="cc\/claude-sonnet-4-5-20250929"/);
|
|
assert.match(envScript, /# http:\/\/localhost:20128\/v1\/chat\/completions/);
|
|
assert.match(envScript, /my-team-cli --base-url "\$OPENAI_BASE_URL"/);
|
|
|
|
const jsonConfig = JSON.parse(
|
|
buildCustomCliJsonConfig({
|
|
cliName: "My Team CLI",
|
|
baseUrl: "http://localhost:20128",
|
|
apiKey: "sk_test_123",
|
|
defaultModel: "omniroute/fast",
|
|
aliasMappings: [{ alias: "review", model: "cc/claude-sonnet-4-5-20250929" }],
|
|
})
|
|
);
|
|
|
|
assert.deepEqual(jsonConfig, {
|
|
name: "My Team CLI",
|
|
provider: {
|
|
type: "openai",
|
|
baseURL: "http://localhost:20128/v1",
|
|
apiKey: "sk_test_123",
|
|
model: "omniroute/fast",
|
|
},
|
|
modelAliases: {
|
|
review: "cc/claude-sonnet-4-5-20250929",
|
|
},
|
|
});
|
|
});
|