Files
OmniRoute/tests/unit/custom-cli-config.test.ts
diegosouzapw 01dd72cf1f feat(dashboard): add custom CLI builder and eval suite management
Introduce a custom CLI tools card that generates OpenAI-compatible
env vars and JSON config, and add a cost overview tab with pricing
source visibility.

Add persisted custom eval suites with authenticated CRUD routes and
validation, plus new translator stream transformation tooling and
richer live monitor routing details.

Improve localization coverage across dashboard flows and add unit
tests for custom CLI config, pricing sources, eval suites, and
stream transformation.
2026-04-23 16:57:43 -03:00

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-tools/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",
},
});
});