Files
OmniRoute/tests/unit/proxy-family-resolve-cache.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

77 lines
2.6 KiB
TypeScript

import { describe, it, beforeEach } from "node:test";
import assert from "node:assert/strict";
import {
assertHostnameSupportsFamily,
__clearFamilyCheckCacheForTest,
} from "../../open-sse/utils/proxyFamilyResolve.ts";
describe("proxyFamilyResolve DNS family cache (#9158)", () => {
beforeEach(() => __clearFamilyCheckCacheForTest());
it("caches a positive family check so repeated calls reuse DNS", async () => {
let lookupCount = 0;
const lookupFn = async () => {
lookupCount += 1;
return [{ address: "::1", family: 6 }];
};
await assertHostnameSupportsFamily("relay.example.test", 6, lookupFn);
await assertHostnameSupportsFamily("relay.example.test", 6, lookupFn);
assert.equal(lookupCount, 1, "second call must hit the positive cache");
});
it("re-probes after the test hook clears the cache", async () => {
let lookupCount = 0;
const lookupFn = async () => {
lookupCount += 1;
return [{ address: "::1", family: 6 }];
};
await assertHostnameSupportsFamily("relay.example.test", 6, lookupFn);
__clearFamilyCheckCacheForTest();
await assertHostnameSupportsFamily("relay.example.test", 6, lookupFn);
assert.equal(lookupCount, 2, "cleared cache must re-invoke DNS");
});
it("caches a negative result without re-invoking DNS", async () => {
let lookupCount = 0;
const failingFn = async () => {
lookupCount += 1;
throw new Error("boom");
};
await assert.rejects(
assertHostnameSupportsFamily("relay.example.test", 6, failingFn),
/resolution/i
);
await assert.rejects(
assertHostnameSupportsFamily("relay.example.test", 6, failingFn),
/resolution/i
);
assert.equal(lookupCount, 1, "negative result must be cached");
});
it("no-ops for bracketed IPv6 literals without any DNS lookup", async () => {
const failingFn = async () => {
throw new Error("must not be called");
};
await assertHostnameSupportsFamily("[2001:db8::1]", 6, failingFn);
});
it("keys the cache by lookup function so a different resolver re-probes", async () => {
let countA = 0;
const fnA = async () => {
countA += 1;
return [{ address: "::1", family: 6 }];
};
let countB = 0;
const fnB = async () => {
countB += 1;
return [{ address: "::1", family: 6 }];
};
await assertHostnameSupportsFamily("relay.example.test", 6, fnA);
await assertHostnameSupportsFamily("relay.example.test", 6, fnA);
await assertHostnameSupportsFamily("relay.example.test", 6, fnB);
assert.equal(countA, 1);
assert.equal(countB, 1, "different lookupFn must bypass the cached result");
});
});