wehub-resource-sync
76 lines
2.7 KiB
TypeScript
76 lines
2.7 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import fs from "node:fs";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
|
|
const TEST_DATA_DIR = fs.mkdtempSync(path.join(os.tmpdir(), "omniroute-alias-seed-"));
|
|
process.env.DATA_DIR = TEST_DATA_DIR;
|
|
|
|
const core = await import("../../src/lib/db/core.ts");
|
|
const modelsDb = await import("../../src/lib/db/models.ts");
|
|
const sseModelService = await import("../../src/sse/services/model.ts");
|
|
const { DEFAULT_MODEL_ALIAS_SEED, seedDefaultModelAliases } =
|
|
await import("../../src/lib/modelAliasSeed.ts");
|
|
|
|
async function resetStorage() {
|
|
core.resetDbInstance();
|
|
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
|
|
fs.mkdirSync(TEST_DATA_DIR, { recursive: true });
|
|
}
|
|
|
|
test.beforeEach(async () => {
|
|
await resetStorage();
|
|
});
|
|
|
|
test.after(async () => {
|
|
core.resetDbInstance();
|
|
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
|
|
});
|
|
|
|
test("default model alias seed writes missing aliases and is idempotent", async () => {
|
|
const first = await seedDefaultModelAliases();
|
|
const aliases = await modelsDb.getModelAliases();
|
|
|
|
assert.deepEqual(first.failed, []);
|
|
assert.equal(first.applied.length, Object.keys(DEFAULT_MODEL_ALIAS_SEED).length);
|
|
assert.equal(aliases["gemini-3-pro-high"], "agy/gemini-3.1-pro-high");
|
|
assert.equal(aliases["gemini-3-pro-low"], "agy/gemini-3.1-pro-low");
|
|
assert.equal(aliases["gemini-3-pro-preview"], "agy/gemini-pro-agent");
|
|
assert.equal(aliases["gemini-3.1-pro-preview"], "agy/gemini-pro-agent");
|
|
assert.equal(aliases["gemini-3-flash-preview"], "agy/gemini-3.5-flash-medium");
|
|
|
|
const routed = await sseModelService.getModelInfo("gemini-3-pro-high");
|
|
assert.deepEqual(routed, {
|
|
provider: "agy",
|
|
model: "gemini-3.1-pro-high",
|
|
extendedContext: false,
|
|
});
|
|
|
|
const second = await seedDefaultModelAliases();
|
|
assert.equal(second.applied.length, 0);
|
|
assert.equal(second.failed.length, 0);
|
|
assert.equal(second.skipped.length, Object.keys(DEFAULT_MODEL_ALIAS_SEED).length);
|
|
});
|
|
|
|
test("default model alias seed preserves existing aliases and skips invalid entries", async () => {
|
|
await modelsDb.setModelAlias("gemini-3-pro-high", "custom/provider-model");
|
|
|
|
const warnings = [];
|
|
const result = await seedDefaultModelAliases({
|
|
logger: {
|
|
warn: (message) => warnings.push(String(message)),
|
|
},
|
|
seedMap: {
|
|
...DEFAULT_MODEL_ALIAS_SEED,
|
|
"broken-entry": null,
|
|
},
|
|
});
|
|
const aliases = await modelsDb.getModelAliases();
|
|
|
|
assert.equal(aliases["gemini-3-pro-high"], "custom/provider-model");
|
|
assert.ok(result.skipped.includes("gemini-3-pro-high"));
|
|
assert.ok(result.failed.includes("broken-entry"));
|
|
assert.ok(warnings.some((message) => message.includes("broken-entry")));
|
|
});
|