Files
OmniRoute/tests/unit/aihorde-key-validation.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

72 lines
2.3 KiB
TypeScript

import test from "node:test";
import assert from "node:assert/strict";
import { readFileSync } from "node:fs";
import { validateAiHordeProvider } from "../../src/lib/providers/validation/aihorde.ts";
function jsonResponse(status: number, body: unknown): Response {
return new Response(JSON.stringify(body), {
status,
headers: { "content-type": "application/json" },
});
}
test("empty Horde key is valid because the provider is optional", async () => {
const result = await validateAiHordeProvider({
apiKey: " ",
fetchImpl: async () => {
throw new Error("find_user must not run when no key is pasted");
},
});
assert.equal(result.valid, true);
assert.equal(result.method, "aihorde_anonymous");
});
test("junk Horde key is rejected by find_user 404", async () => {
const result = await validateAiHordeProvider({
apiKey: "junk-key-not-real",
fetchImpl: async () =>
jsonResponse(404, {
message: "User with api_key 'junk-key-not-real' not found.",
rc: "UserNotFound",
}),
});
assert.equal(result.valid, false);
assert.equal(result.error, "Invalid API key");
});
test("missing Horde key header is rejected by find_user 401", async () => {
const result = await validateAiHordeProvider({
apiKey: "not-a-real-key",
fetchImpl: async () =>
jsonResponse(401, { message: "No user matching sent API Key.", rc: "InvalidAPIKey" }),
});
assert.equal(result.valid, false);
assert.equal(result.error, "Invalid API key");
});
test("registered Horde key is accepted when find_user returns a username", async () => {
let sentKey = "";
let sentUrl = "";
const result = await validateAiHordeProvider({
apiKey: "horde-registered-key-123",
fetchImpl: async (url, init) => {
sentUrl = String(url);
sentKey = new Headers(init?.headers).get("apikey") || "";
return jsonResponse(200, { username: "tester#1234", kudos: 100 });
},
});
assert.equal(result.valid, true);
assert.equal(result.method, "aihorde_find_user");
assert.equal(sentKey, "horde-registered-key-123");
assert.match(sentUrl, /\/v2\/find_user$/);
});
test("validation.ts registers the Horde find_user specialty validator", () => {
const src = readFileSync(
new URL("../../src/lib/providers/validation.ts", import.meta.url),
"utf8"
);
assert.match(src, /aihorde:\s*validateAiHordeProvider/);
});