mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-09-10 17:02:22 +03:00
65e81158ab
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.
59 lines
2.2 KiB
TypeScript
59 lines
2.2 KiB
TypeScript
import { describe, it } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import { exchangeKimiRefreshToken } from "../../src/lib/kimi/tokenRefresh.ts";
|
|
|
|
describe("Kimi Token Refresh Exchange", () => {
|
|
it("exchanges valid refresh_token for new access_token and refresh_token", async () => {
|
|
const originalFetch = globalThis.fetch;
|
|
const now = Math.floor(Date.now() / 1000);
|
|
const mockNewAccessToken =
|
|
"eyJhbGciOiJIUzUxMiJ9." +
|
|
Buffer.from(JSON.stringify({ exp: now + 86400 * 30, iat: now })).toString("base64url") +
|
|
".sig";
|
|
const mockNewRefreshToken =
|
|
"eyJhbGciOiJIUzUxMiJ9." +
|
|
Buffer.from(JSON.stringify({ exp: now + 86400 * 90, iat: now })).toString("base64url") +
|
|
".sig";
|
|
|
|
try {
|
|
globalThis.fetch = (async (url: Parameters<typeof fetch>[0], init?: RequestInit) => {
|
|
assert.equal(String(url), "https://www.kimi.ai/api/auth/token/refresh");
|
|
assert.equal(init?.headers?.["Authorization"], "Bearer sample_refresh_token");
|
|
return new Response(
|
|
JSON.stringify({
|
|
access_token: mockNewAccessToken,
|
|
refresh_token: mockNewRefreshToken,
|
|
}),
|
|
{ status: 200, headers: { "Content-Type": "application/json" } }
|
|
);
|
|
}) as typeof fetch;
|
|
|
|
const res = await exchangeKimiRefreshToken("sample_refresh_token", "https://www.kimi.ai");
|
|
assert.equal(res.success, true);
|
|
assert.equal(res.accessToken, mockNewAccessToken);
|
|
assert.equal(res.refreshToken, mockNewRefreshToken);
|
|
assert.ok(res.expiresAtSec! > now);
|
|
} finally {
|
|
globalThis.fetch = originalFetch;
|
|
}
|
|
});
|
|
|
|
it("handles upstream refresh failure cleanly", async () => {
|
|
const originalFetch = globalThis.fetch;
|
|
try {
|
|
globalThis.fetch = (async () => {
|
|
return new Response(JSON.stringify({ error_type: "auth.token.invalid" }), {
|
|
status: 401,
|
|
headers: { "Content-Type": "application/json" },
|
|
});
|
|
}) as typeof fetch;
|
|
|
|
const res = await exchangeKimiRefreshToken("expired_refresh_token", "https://www.kimi.ai");
|
|
assert.equal(res.success, false);
|
|
assert.ok(res.error?.includes("401"));
|
|
} finally {
|
|
globalThis.fetch = originalFetch;
|
|
}
|
|
});
|
|
});
|