wehub-resource-sync
62 lines
2.3 KiB
TypeScript
62 lines
2.3 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import {
|
|
getAvailabilityReport,
|
|
clearModelUnavailability,
|
|
resetAllAvailability,
|
|
} from "../../src/domain/modelAvailability.ts";
|
|
import { lockModel, clearModelLock } from "@omniroute/open-sse/services/accountFallback";
|
|
|
|
const TEST_CONN = "test-conn-001";
|
|
|
|
function seed(provider: string, model: string, cooldownMs = 60_000) {
|
|
lockModel(provider, TEST_CONN, model, "quota_exhausted", cooldownMs, {});
|
|
}
|
|
|
|
function cleanup(provider: string, model: string) {
|
|
clearModelLock(provider, TEST_CONN, model);
|
|
}
|
|
|
|
test("getAvailabilityReport: returns empty array when no lockouts", () => {
|
|
const report = getAvailabilityReport();
|
|
const forProvider = report.filter((e) => e.provider === "test-empty-provider");
|
|
assert.equal(forProvider.length, 0);
|
|
});
|
|
|
|
test("getAvailabilityReport: returns active lockout with positive remainingMs", () => {
|
|
seed("test-prov", "test-model");
|
|
try {
|
|
const report = getAvailabilityReport();
|
|
const entry = report.find((e) => e.provider === "test-prov" && e.model === "test-model");
|
|
assert.ok(entry, "lockout should appear in report");
|
|
assert.ok(entry.remainingMs > 0, "remainingMs should be positive");
|
|
} finally {
|
|
cleanup("test-prov", "test-model");
|
|
}
|
|
});
|
|
|
|
test("clearModelUnavailability: removes matching lockout and returns true", () => {
|
|
seed("prov-clear", "model-clear");
|
|
const removed = clearModelUnavailability("prov-clear", "model-clear");
|
|
assert.equal(removed, true);
|
|
const report = getAvailabilityReport();
|
|
const stillThere = report.find((e) => e.provider === "prov-clear" && e.model === "model-clear");
|
|
assert.equal(stillThere, undefined);
|
|
});
|
|
|
|
test("clearModelUnavailability: returns false when no matching lockout", () => {
|
|
const removed = clearModelUnavailability("nonexistent-prov", "nonexistent-model");
|
|
assert.equal(removed, false);
|
|
});
|
|
|
|
test("resetAllAvailability: clears all seeded lockouts", () => {
|
|
seed("reset-prov-a", "reset-model-a");
|
|
seed("reset-prov-b", "reset-model-b");
|
|
resetAllAvailability();
|
|
const report = getAvailabilityReport();
|
|
const a = report.find((e) => e.provider === "reset-prov-a");
|
|
const b = report.find((e) => e.provider === "reset-prov-b");
|
|
assert.equal(a, undefined);
|
|
assert.equal(b, undefined);
|
|
});
|