Files
OmniRoute/tests/unit/migration-149-api-key-combo-access.test.ts
Praveen K Palaniswamy 65e81158ab fix(ollama): route models by advertised capability (#11088)
Landed with the design call resolved per the owner's pick — **option 1**: the synced store is now endpoint-agnostic (persistDiscoveredModels and managedModelImport no longer drop non-chat models at write time), and chat selectability moved to read time (auto-pool expansion in autoStrategy applies filterChatSelectableModels; the models-route projection already had its chatOnly filter). Your discovery test now passes end-to-end (3/3): /api/show capabilities persist per connection and image/embedding requests route through the advertising host.

Reconciliation notes: conflicted areas merged onto the current tip (adobe discovery import, requestedModel preflight signature, resolvedProvider fast-path coexists with the synced-route override — explicit resolution wins); carried base-red drains (#10055 memoization, #11071 test variants) dropped as already-landed; the managed-model-import exclusion test was propagated to the new contract (image/video models persist; the read filter still hides them from chat pickers — pinned by a new assertion). Full battery: 205/206 focused (the one red is a confirmed periodic-timer timing flake on the loaded devbox — 20/20 isolated), autoCombo vitest 30/30, combo suites 46/46, gates + typecheck clean.

Thank you @yourspraveen — the capability probe + routing design was right; it just needed the store contract opened up. Fixes #11087.
2026-08-23 11:45:01 -03:00

57 lines
2.4 KiB
TypeScript

// ENVIRONMENT NOTE (sandbox better-sqlite3 / glibc limitation, not a code defect):
// This test constructs or exercises a real better-sqlite3-backed SQLite database.
// better-sqlite3 is a native addon; production and CI load it normally, but some
// sandboxes/dev boxes ship a system glibc older than the prebuilt binary requires
// ("GLIBC_2.29 not found"), so the native module fails to dlopen and any test that
// reaches better-sqlite3 directly (or asserts stdout that the load-failure warning
// would pollute) fails HERE while passing in CI. This is a known environment
// limitation, not a defect in the code under test: the OmniRoute runtime itself
// cascades to node:sqlite/sql.js when better-sqlite3 is unavailable. See
// tests/unit/_helpers/betterSqlite3Availability.ts for a guard helper.
import test from "node:test";
import assert from "node:assert/strict";
import fs from "node:fs";
import path from "node:path";
import { fileURLToPath } from "node:url";
import DatabaseSync from "better-sqlite3";
const repoRoot = path.resolve(path.dirname(fileURLToPath(import.meta.url)), "../..");
const migrationPath = path.join(repoRoot, "src/lib/db/migrations/149_api_key_combo_access.sql");
test("combo-access migration preserves legacy allow-all rows and named allowlists", () => {
const sql = fs.readFileSync(migrationPath, "utf8");
const db = new DatabaseSync(":memory:");
db.exec(`
CREATE TABLE api_keys (
id TEXT PRIMARY KEY,
allowed_combos TEXT
);
INSERT INTO api_keys (id, allowed_combos) VALUES
('legacy-null', NULL),
('legacy-empty', '[]'),
('legacy-blank', ''),
('legacy-malformed', 'not-json'),
('named', '["fast-chat"]'),
('all', '["combo/*"]');
`);
db.exec(sql);
db.exec(sql);
const rows = db.prepare("SELECT id, allowed_combos FROM api_keys ORDER BY id").all() as Array<{
id: string;
allowed_combos: string;
}>;
const combosById = new Map(
rows.map((row) => [row.id, JSON.parse(row.allowed_combos) as string[]])
);
assert.deepEqual(combosById.get("legacy-null"), ["combo/*"]);
assert.deepEqual(combosById.get("legacy-empty"), ["combo/*"]);
assert.deepEqual(combosById.get("legacy-blank"), ["combo/*"]);
assert.deepEqual(combosById.get("legacy-malformed"), ["combo/*"]);
assert.deepEqual(combosById.get("named"), ["fast-chat"]);
assert.deepEqual(combosById.get("all"), ["combo/*"]);
});