Files
agents-marketplace/tests/unit/supplier-config.test.ts
david 2f7b0194a5 feat(supplier,gateway,buyer): N-slot chat suppliers + entry-ticket settle mode
Two structural fixes for demo concurrency (3 single-slot suppliers thrashed
under Open WebUI's ~3 conversations per message):

Multi-slot suppliers (MAX_CHAT_SESSIONS, default 1):
- SupplierState is now capacity-aware: a Set of active escrow refs with
  free-until-full status semantics, ref-scoped release(ref) (also fixes the
  latent bug where handler catch-alls released another request's slot), and
  a walletMutex serializing every wallet-spending chain op — Claim, Submit,
  and wallet-health consolidation queue instead of racing the single
  working/collateral UTxO. /status gains active_sessions/max_sessions.

Entry-ticket settle mode (CHAT_SETTLE_MODE=ticket, default full, both sides):
- The buyer's escrow post at session open is the ONLY chain op. The supplier
  verifies the Open escrow as an entry ticket but never Claims/Submits
  (/v1/chat/start → {status:"ticket"}; /v1/chat/end → receipt-less
  {status:"ended"}); no hard-cap timer, so sessions may outlive deliver_by.
  Used tickets can't reopen sessions (records retained; 409 session_exists).
- Buyer SDK: startChat/endChat are settle-mode aware, driven by the
  supplier's response — mixed-mode rollout windows settle correctly.
- Gateway: ticket closes bill cost 0; the sweeper Reclaims Open tickets
  after deliver_by WITHOUT closing live sessions (the conversation and its
  affinity mirror survive their reclaimed ticket), gates reclaims on
  deliver_by, and keeps an in-process settled-refs memory so terminal
  indexer rows aren't retried every sweep forever.

Deployment (kimi-chat trio + gateway): ticket mode, 8 slots per supplier,
idle policy gateway 10 min / supplier 15 min (fixes the inversion where the
supplier idle-closed before the gateway janitor).

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
2026-07-09 20:48:52 +02:00

257 lines
8.9 KiB
TypeScript

/**
* supplier-config.test.ts — RED phase tests for supplier/src/config.ts
*
* Tests loadConfig(env) and parseAdvertRef(ref):
* - Happy paths: all required vars present
* - Defaults: PORT and OLLAMA_TIMEOUT_MS
* - Errors: missing required vars, malformed values
*/
import { describe, it, expect } from "vitest";
import { loadConfig, parseAdvertRef } from "../../supplier/src/config.js";
import { buildSampleEnv } from "../fixtures/supplier-side/sample-config.js";
import { SAMPLE_ADVERT_TX_HASH, SAMPLE_ADVERT_INDEX } from "../fixtures/supplier-side/sample-config.js";
import { SUPPLIER_PRIVATE_KEY_HEX } from "../fixtures/supplier-side/wallet-keys.js";
// ─── loadConfig — happy paths ────────────────────────────────────────────────
describe("loadConfig() — happy path", () => {
it("returns a valid SupplierConfig for a complete env map", () => {
const cfg = loadConfig(buildSampleEnv());
expect(cfg.supplierPrivKeyHex).toBe(SUPPLIER_PRIVATE_KEY_HEX);
expect(cfg.ogmiosUrl).toBe("ws://localhost:1337");
expect(cfg.ollamaUrl).toBe("http://localhost:11434");
expect(cfg.advertRef.txHash).toBe(SAMPLE_ADVERT_TX_HASH);
expect(cfg.advertRef.index).toBe(SAMPLE_ADVERT_INDEX);
expect(cfg.networkId).toBe(0);
expect(cfg.port).toBe(8080);
expect(cfg.ollamaTimeoutMs).toBe(120_000);
});
it("parses NETWORK_ID 1 as mainnet", () => {
const cfg = loadConfig({ ...buildSampleEnv(), NETWORK_ID: "1" });
expect(cfg.networkId).toBe(1);
});
it("defaults PORT to 8080 when omitted", () => {
const env = { ...buildSampleEnv() };
delete env.PORT;
const cfg = loadConfig(env);
expect(cfg.port).toBe(8080);
});
it("defaults OLLAMA_TIMEOUT_MS to 120_000 when omitted", () => {
const env = { ...buildSampleEnv() };
delete env.OLLAMA_TIMEOUT_MS;
const cfg = loadConfig(env);
expect(cfg.ollamaTimeoutMs).toBe(120_000);
});
it("accepts custom PORT", () => {
const cfg = loadConfig({ ...buildSampleEnv(), PORT: "9000" });
expect(cfg.port).toBe(9000);
});
it("accepts custom OLLAMA_TIMEOUT_MS", () => {
const cfg = loadConfig({ ...buildSampleEnv(), OLLAMA_TIMEOUT_MS: "30000" });
expect(cfg.ollamaTimeoutMs).toBe(30_000);
});
});
// ─── loadConfig — missing required vars ─────────────────────────────────────
describe("loadConfig() — missing required vars", () => {
it("throws when SUPPLIER_PRIV_KEY_HEX is missing", () => {
const env = { ...buildSampleEnv() };
delete env.SUPPLIER_PRIV_KEY_HEX;
expect(() => loadConfig(env)).toThrow();
});
it("throws when OGMIOS_URL is missing", () => {
const env = { ...buildSampleEnv() };
delete env.OGMIOS_URL;
expect(() => loadConfig(env)).toThrow();
});
it("throws when OLLAMA_URL is missing", () => {
const env = { ...buildSampleEnv() };
delete env.OLLAMA_URL;
expect(() => loadConfig(env)).toThrow();
});
it("throws when ADVERT_REF is missing", () => {
const env = { ...buildSampleEnv() };
delete env.ADVERT_REF;
expect(() => loadConfig(env)).toThrow();
});
it("throws when NETWORK_ID is missing", () => {
const env = { ...buildSampleEnv() };
delete env.NETWORK_ID;
expect(() => loadConfig(env)).toThrow();
});
});
// ─── loadConfig — malformed values ──────────────────────────────────────────
describe("loadConfig() — malformed values", () => {
it("throws when SUPPLIER_PRIV_KEY_HEX is not 64-char hex", () => {
expect(() =>
loadConfig({ ...buildSampleEnv(), SUPPLIER_PRIV_KEY_HEX: "short" })
).toThrow();
});
it("throws when SUPPLIER_PRIV_KEY_HEX is 63 chars (odd length)", () => {
expect(() =>
loadConfig({ ...buildSampleEnv(), SUPPLIER_PRIV_KEY_HEX: "a".repeat(63) })
).toThrow();
});
it("throws when SUPPLIER_PRIV_KEY_HEX is 64 chars but contains non-hex", () => {
expect(() =>
loadConfig({ ...buildSampleEnv(), SUPPLIER_PRIV_KEY_HEX: "z".repeat(64) })
).toThrow();
});
it("throws when NETWORK_ID is not '0' or '1'", () => {
expect(() =>
loadConfig({ ...buildSampleEnv(), NETWORK_ID: "2" })
).toThrow();
});
it("throws when ADVERT_REF is malformed (no # separator)", () => {
expect(() =>
loadConfig({ ...buildSampleEnv(), ADVERT_REF: "nohashsep" })
).toThrow();
});
it("throws when ADVERT_REF txHash is not 64 hex chars", () => {
expect(() =>
loadConfig({ ...buildSampleEnv(), ADVERT_REF: "abc#0" })
).toThrow();
});
it("throws when ADVERT_REF index is not a non-negative integer", () => {
expect(() =>
loadConfig({ ...buildSampleEnv(), ADVERT_REF: `${"a".repeat(64)}#-1` })
).toThrow();
});
it("throws when PORT is not a positive integer string", () => {
expect(() =>
loadConfig({ ...buildSampleEnv(), PORT: "not-a-port" })
).toThrow();
});
it("throws when OLLAMA_TIMEOUT_MS is not a positive integer string", () => {
expect(() =>
loadConfig({ ...buildSampleEnv(), OLLAMA_TIMEOUT_MS: "0" })
).toThrow();
});
it("error message mentions the missing/malformed field name", () => {
const env = { ...buildSampleEnv() };
delete env.SUPPLIER_PRIV_KEY_HEX;
let errorMsg = "";
try {
loadConfig(env);
} catch (e) {
errorMsg = (e as Error).message;
}
expect(errorMsg.toLowerCase()).toMatch(/supplier_priv_key_hex/i);
});
});
// ─── loadConfig — LIVE_CHAIN (M1-F-2) ───────────────────────────────────────
// SPEC FIX 2026-04-27: M1-F-2 LIVE_CHAIN env support
describe("loadConfig() — LIVE_CHAIN parsing", () => {
it("LIVE_CHAIN='1' sets liveChain: true", () => {
const cfg = loadConfig({ ...buildSampleEnv(), LIVE_CHAIN: "1" });
expect(cfg.liveChain).toBe(true);
});
it("LIVE_CHAIN omitted defaults to liveChain: false", () => {
const env = { ...buildSampleEnv() };
delete (env as Record<string, string | undefined>).LIVE_CHAIN;
const cfg = loadConfig(env);
expect(cfg.liveChain).toBe(false);
});
it("LIVE_CHAIN='true' does NOT set liveChain (only literal '1' opts in)", () => {
const cfg = loadConfig({ ...buildSampleEnv(), LIVE_CHAIN: "true" });
expect(cfg.liveChain).toBe(false);
});
});
// ─── parseAdvertRef ──────────────────────────────────────────────────────────
describe("parseAdvertRef()", () => {
it("parses a well-formed ref into { txHash, index }", () => {
const ref = parseAdvertRef(`${"b".repeat(64)}#0`);
expect(ref.txHash).toBe("b".repeat(64));
expect(ref.index).toBe(0);
});
it("parses index > 0", () => {
const ref = parseAdvertRef(`${"a".repeat(64)}#7`);
expect(ref.index).toBe(7);
});
it("throws on empty string", () => {
expect(() => parseAdvertRef("")).toThrow();
});
it("throws when # separator is missing", () => {
expect(() => parseAdvertRef("a".repeat(64))).toThrow();
});
it("throws when txHash part is shorter than 64 chars", () => {
expect(() => parseAdvertRef("abc#0")).toThrow();
});
it("throws when index part is not a number", () => {
expect(() => parseAdvertRef(`${"a".repeat(64)}#notanumber`)).toThrow();
});
it("throws when index part is negative", () => {
expect(() => parseAdvertRef(`${"a".repeat(64)}#-1`)).toThrow();
});
});
// ─── loadConfig — chat-session concurrency + settle mode ─────────────────────
describe("loadConfig() — MAX_CHAT_SESSIONS / CHAT_SETTLE_MODE", () => {
const chatSessionEnv = () => ({
...buildSampleEnv(),
CAPABILITY_KIND: "chat-session",
LLM_BACKEND: "openai",
OPENAI_BASE_URL: "http://up",
});
it("defaults to 1 slot and full settle mode", () => {
const cfg = loadConfig(buildSampleEnv());
expect(cfg.maxChatSessions).toBe(1);
expect(cfg.chatSettleMode).toBe("full");
});
it("parses MAX_CHAT_SESSIONS and CHAT_SETTLE_MODE for chat-session kind", () => {
const cfg = loadConfig({ ...chatSessionEnv(), MAX_CHAT_SESSIONS: "8", CHAT_SETTLE_MODE: "ticket" });
expect(cfg.maxChatSessions).toBe(8);
expect(cfg.chatSettleMode).toBe("ticket");
});
it("rejects MAX_CHAT_SESSIONS > 1 for non-chat-session kinds", () => {
expect(() => loadConfig({ ...buildSampleEnv(), MAX_CHAT_SESSIONS: "4" })).toThrow();
});
it("rejects CHAT_SETTLE_MODE=ticket for non-chat-session kinds", () => {
expect(() => loadConfig({ ...buildSampleEnv(), CHAT_SETTLE_MODE: "ticket" })).toThrow();
});
it("rejects invalid values", () => {
expect(() => loadConfig({ ...chatSessionEnv(), MAX_CHAT_SESSIONS: "0" })).toThrow();
expect(() => loadConfig({ ...chatSessionEnv(), CHAT_SETTLE_MODE: "nope" })).toThrow();
});
});