mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-09-13 18:32:12 +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.
62 lines
2.2 KiB
TypeScript
62 lines
2.2 KiB
TypeScript
import { describe, it } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import {
|
|
parseKimiJwt,
|
|
getKimiTokenExpiration,
|
|
isKimiTokenExpiringSoon,
|
|
} from "../../open-sse/utils/kimiJwt.ts";
|
|
|
|
describe("Kimi JWT Decoder", () => {
|
|
const makeToken = (payload: Record<string, unknown>) => {
|
|
const header = Buffer.from(JSON.stringify({ alg: "HS512", typ: "JWT" })).toString("base64url");
|
|
const body = Buffer.from(JSON.stringify(payload)).toString("base64url");
|
|
return `${header}.${body}.signature`;
|
|
};
|
|
|
|
it("parses valid Kimi JWT payload", () => {
|
|
const now = Math.floor(Date.now() / 1000);
|
|
const token = makeToken({
|
|
sub: "user_123",
|
|
region: "overseas",
|
|
exp: now + 900,
|
|
iat: now,
|
|
});
|
|
const parsed = parseKimiJwt(token);
|
|
assert.ok(parsed);
|
|
assert.equal(parsed?.sub, "user_123");
|
|
assert.equal(parsed?.region, "overseas");
|
|
assert.equal(parsed?.exp, now + 900);
|
|
});
|
|
|
|
it("handles malformed tokens safely without throwing", () => {
|
|
assert.equal(parseKimiJwt(""), null);
|
|
assert.equal(parseKimiJwt("invalid.token"), null);
|
|
assert.equal(parseKimiJwt("a.invalid_json.c"), null);
|
|
});
|
|
|
|
it("calculates remaining seconds and expiration status", () => {
|
|
const now = Math.floor(Date.now() / 1000);
|
|
const validToken = makeToken({ exp: now + 300, iat: now });
|
|
const expiredToken = makeToken({ exp: now - 100, iat: now - 400 });
|
|
|
|
const validExp = getKimiTokenExpiration(validToken);
|
|
assert.ok(validExp);
|
|
assert.equal(validExp?.isExpired, false);
|
|
assert.ok(validExp!.remainingSec > 250);
|
|
|
|
const expiredExp = getKimiTokenExpiration(expiredToken);
|
|
assert.ok(expiredExp);
|
|
assert.equal(expiredExp?.isExpired, true);
|
|
assert.ok(expiredExp!.remainingSec <= 0);
|
|
});
|
|
|
|
it("correctly identifies tokens expiring soon within threshold", () => {
|
|
const now = Math.floor(Date.now() / 1000);
|
|
const tokenExpiringIn2Min = makeToken({ exp: now + 120, iat: now });
|
|
const tokenExpiringIn10Min = makeToken({ exp: now + 600, iat: now });
|
|
|
|
assert.equal(isKimiTokenExpiringSoon(tokenExpiringIn2Min, 180), true);
|
|
assert.equal(isKimiTokenExpiringSoon(tokenExpiringIn10Min, 180), false);
|
|
});
|
|
});
|