Files
OmniRoute/tests/unit/provider-refresh-token-route.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

58 lines
2.2 KiB
TypeScript

import { describe, it } from "node:test";
import assert from "node:assert/strict";
import { POST } from "../../src/app/api/providers/[id]/refresh-token/route.ts";
import { createProviderConnection } from "../../src/lib/db/providers.ts";
describe("POST /api/providers/[id]/refresh-token", () => {
it("returns 404 for nonexistent connection", async () => {
const req = new Request("http://localhost/api/providers/nonexistent/refresh-token", {
method: "POST",
});
const res = await POST(req, { params: Promise.resolve({ id: "nonexistent" }) });
assert.equal(res.status, 404);
});
it("refreshes active kimi-web connection", async () => {
const originalFetch = globalThis.fetch;
const now = Math.floor(Date.now() / 1000);
const mockAccessToken =
"eyJhbGciOiJIUzUxMiJ9." +
Buffer.from(JSON.stringify({ sub: "user_test_99", exp: now + 3600, iat: now })).toString("base64url") +
".sig";
const mockRefreshToken =
"eyJhbGciOiJIUzUxMiJ9." +
Buffer.from(JSON.stringify({ exp: now + 86400 * 90, iat: now })).toString("base64url") +
".sig";
try {
globalThis.fetch = (async (url: Parameters<typeof fetch>[0]) => {
if (String(url).includes("/api/auth/token/refresh")) {
return new Response(
JSON.stringify({ access_token: mockAccessToken, refresh_token: mockRefreshToken }),
{ status: 200, headers: { "Content-Type": "application/json" } }
);
}
return new Response("Not found", { status: 404 });
}) as typeof fetch;
const conn = await createProviderConnection({
provider: "kimi-web",
name: "Test Kimi Connection",
apiKey: "old_access_token",
refreshToken: "valid_refresh_token",
});
const req = new Request(`http://localhost/api/providers/${conn.id}/refresh-token`, {
method: "POST",
});
const res = await POST(req, { params: Promise.resolve({ id: conn.id }) });
assert.equal(res.status, 200);
const data = await res.json();
assert.equal(data.success, true);
assert.equal(data.user.userId, "user_test_99");
} finally {
globalThis.fetch = originalFetch;
}
});
});