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.3 KiB
TypeScript
62 lines
2.3 KiB
TypeScript
import { test } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import {
|
|
_clearOAuthSessionOccupancyForTest,
|
|
getForeignOAuthSessionCount,
|
|
getOAuthSessionAvailability,
|
|
reserveOAuthSession,
|
|
wrapResponseWithOAuthSessionRelease,
|
|
} from "../../open-sse/services/oauthSessionOccupancy.ts";
|
|
|
|
test.beforeEach(() => _clearOAuthSessionOccupancyForTest());
|
|
|
|
test("counts foreign sessions while ignoring parallel requests from the same session", () => {
|
|
const releaseA1 = reserveOAuthSession("account-a", "session-a", 60_000, 1_000);
|
|
const releaseA2 = reserveOAuthSession("account-a", "session-a", 60_000, 1_000);
|
|
|
|
assert.equal(getForeignOAuthSessionCount("account-a", "session-a", 1_001), 0);
|
|
assert.equal(getForeignOAuthSessionCount("account-a", "session-b", 1_001), 1);
|
|
assert.equal(getOAuthSessionAvailability("account-a", "session-b", 1_001), 0.5);
|
|
|
|
releaseA1();
|
|
assert.equal(getForeignOAuthSessionCount("account-a", "session-b", 1_001), 1);
|
|
releaseA2();
|
|
assert.equal(getForeignOAuthSessionCount("account-a", "session-b", 1_001), 0);
|
|
});
|
|
|
|
test("expired leases fail open", () => {
|
|
reserveOAuthSession("account-a", "session-a", 100, 1_000);
|
|
assert.equal(getForeignOAuthSessionCount("account-a", "session-b", 1_099), 1);
|
|
assert.equal(getForeignOAuthSessionCount("account-a", "session-b", 1_100), 0);
|
|
});
|
|
|
|
test("response wrapper releases only after streaming completes", async () => {
|
|
const release = reserveOAuthSession("account-a", "session-a", 60_000);
|
|
const wrapped = wrapResponseWithOAuthSessionRelease(
|
|
new Response(
|
|
new ReadableStream<Uint8Array>({
|
|
start(controller) {
|
|
controller.enqueue(new TextEncoder().encode("data"));
|
|
controller.close();
|
|
},
|
|
})
|
|
),
|
|
release
|
|
);
|
|
|
|
assert.equal(getForeignOAuthSessionCount("account-a", "session-b"), 1);
|
|
assert.equal(await wrapped.text(), "data");
|
|
assert.equal(getForeignOAuthSessionCount("account-a", "session-b"), 0);
|
|
});
|
|
|
|
test("response wrapper releases on cancellation", async () => {
|
|
const release = reserveOAuthSession("account-a", "session-a", 60_000);
|
|
const wrapped = wrapResponseWithOAuthSessionRelease(
|
|
new Response(new ReadableStream<Uint8Array>({ pull() {} })),
|
|
release
|
|
);
|
|
await wrapped.body?.cancel("client disconnected");
|
|
assert.equal(getForeignOAuthSessionCount("account-a", "session-b"), 0);
|
|
});
|