wehub-resource-sync
43 lines
1.2 KiB
TypeScript
43 lines
1.2 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import {
|
|
clearProxyFallbackCache,
|
|
findWorkingProxy,
|
|
__setProxyFallbackTestHooks,
|
|
} from "../../open-sse/utils/proxyFallback.ts";
|
|
|
|
test.afterEach(() => {
|
|
__setProxyFallbackTestHooks(null);
|
|
clearProxyFallbackCache();
|
|
});
|
|
|
|
test("proxy fallback negative cache is scoped by target URL, not only hostname", async () => {
|
|
const proxyUrl = "http://127.0.0.1:18080";
|
|
const probes: string[] = [];
|
|
|
|
__setProxyFallbackTestHooks({
|
|
getProxyCandidates: async () => [proxyUrl],
|
|
testSingleProxy: async (_proxyUrl, targetUrl) => {
|
|
probes.push(targetUrl);
|
|
return {
|
|
ok: targetUrl.endsWith("/v1/chat/completions"),
|
|
latencyMs: targetUrl.endsWith("/v1/chat/completions") ? 12 : null,
|
|
};
|
|
},
|
|
});
|
|
|
|
assert.equal(
|
|
await findWorkingProxy("api.example.test", "https://api.example.test/v1/models"),
|
|
null
|
|
);
|
|
assert.equal(
|
|
await findWorkingProxy("api.example.test", "https://api.example.test/v1/chat/completions"),
|
|
proxyUrl
|
|
);
|
|
assert.deepEqual(probes, [
|
|
"https://api.example.test/v1/models",
|
|
"https://api.example.test/v1/chat/completions",
|
|
]);
|
|
});
|