Files
OmniRoute/tests/unit/kiro-social-poll.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

67 lines
2.3 KiB
TypeScript

import test from "node:test";
import assert from "node:assert/strict";
import { classifyKiroSocialPoll, getNextKiroSocialPollInterval } from "@/lib/oauth/kiroSocialPoll";
test("slow_down permanently increases the polling interval for this authorization flow", () => {
const slowed = getNextKiroSocialPollInterval(5_000, "slow_down");
assert.equal(slowed, 10_000);
assert.equal(getNextKiroSocialPollInterval(slowed, "authorization_pending"), 10_000);
assert.equal(getNextKiroSocialPollInterval(slowed, "network_error"), 10_000);
});
test("classifyKiroSocialPoll keeps only documented pending states retryable", () => {
assert.deepEqual(classifyKiroSocialPoll(false, 400, { error: "authorization_pending" }), {
kind: "pending",
error: "authorization_pending",
});
assert.deepEqual(classifyKiroSocialPoll(false, 429, { error: "slow_down" }), {
kind: "pending",
error: "slow_down",
});
});
test("classifyKiroSocialPoll treats status as alias of error for pending states", () => {
// Kiro upstream returns progress in `status`, not `error`
assert.deepEqual(classifyKiroSocialPoll(true, 200, { status: "authorization_pending" }), {
kind: "pending",
error: "authorization_pending",
});
assert.deepEqual(classifyKiroSocialPoll(true, 200, { status: "slow_down" }), {
kind: "pending",
error: "slow_down",
});
// error takes precedence over status if both present
assert.deepEqual(
classifyKiroSocialPoll(true, 200, { error: "slow_down", status: "authorization_pending" }),
{
kind: "pending",
error: "slow_down",
}
);
});
test("classifyKiroSocialPoll stops on denied, expired and malformed responses", () => {
assert.deepEqual(classifyKiroSocialPoll(false, 403, { error: "access_denied" }), {
kind: "error",
error: "access_denied",
status: 403,
});
assert.deepEqual(classifyKiroSocialPoll(true, 200, { error: "expired_token" }), {
kind: "error",
error: "expired_token",
status: 400,
});
assert.deepEqual(classifyKiroSocialPoll(true, 200, {}), {
kind: "error",
error: "invalid_token_response",
status: 502,
});
});
test("classifyKiroSocialPoll accepts a token response", () => {
assert.deepEqual(classifyKiroSocialPoll(true, 200, { accessToken: "token" }), {
kind: "success",
});
});