19 lines
2.9 KiB
TypeScript
19 lines
2.9 KiB
TypeScript
import { describe, expect, it, vi } from "vitest";
|
|
import { MemoryApprovalBroker } from "../src/approval.js";
|
|
import { MemoryEventStore } from "../src/event-store.js";
|
|
|
|
describe("event store", () => {
|
|
it("allocates monotonic per-session sequences", () => { const store = new MemoryEventStore(); const a = store.append("s", { type: "run.started", runId: "r", payload: { runId: "r" } }); const b = store.append("s", { type: "run.completed", runId: "r", payload: { runId: "r" } }); expect([a.sequence,b.sequence]).toEqual([1,2]); });
|
|
it("replays only after cursor and notifies subscribers", () => { const store = new MemoryEventStore(); const listener = vi.fn(); const off = store.subscribe("s", listener); const event = store.append("s", { type: "session.started", payload: { cwd: "safe" } }); off(); store.append("s", { type: "session.ended", payload: { reason: "x" } }); expect(listener).toHaveBeenCalledTimes(1); expect(store.listAfter("s", { eventId: event.eventId })).toHaveLength(1); });
|
|
it("keeps a bounded history", () => { const store = new MemoryEventStore(2); for (let i=0;i<3;i++) store.append("s", { type: "run.started", runId: `r${i}`, payload: { runId: `r${i}` } }); expect(store.listAfter("s").map((e)=>e.sequence)).toEqual([2,3]); });
|
|
});
|
|
|
|
describe("approval broker", () => {
|
|
const input = { sessionId: "s", runId: "r", toolCallId: "t", toolName: "save_report_draft", arguments: { body: "draft", apiKey: "secret" }, riskLevel: "medium" as const };
|
|
it("approves and redacts arguments", async () => { let required = ""; const broker = new MemoryApprovalBroker(1000, undefined, (request) => { required=request.approvalId; expect(request.arguments).toMatchObject({ apiKey: "[REDACTED]" }); }); const pending = broker.request(input, new AbortController().signal); broker.approve(required); expect((await pending).decision).toBe("approved"); });
|
|
it("denies and is idempotent for the same decision", async () => { let id=""; const broker = new MemoryApprovalBroker(1000, undefined, (r)=>{id=r.approvalId}); const pending=broker.request(input,new AbortController().signal); const first=broker.deny(id); expect(broker.deny(id)).toEqual(first); expect((await pending).decision).toBe("denied"); });
|
|
it("rejects a conflicting decision", () => { let id=""; const broker=new MemoryApprovalBroker(1000,undefined,(r)=>{id=r.approvalId}); void broker.request(input,new AbortController().signal); broker.deny(id); expect(()=>broker.approve(id)).toThrow(/already resolved/i); });
|
|
it("expires", async () => { let id=""; const broker=new MemoryApprovalBroker(5,undefined,(r)=>{id=r.approvalId}); const result=await broker.request(input,new AbortController().signal); expect(id).toMatch(/^apr_/); expect(result.decision).toBe("expired"); });
|
|
it("cancels on abort", async () => { const controller=new AbortController(); const broker=new MemoryApprovalBroker(1000); const result=broker.request(input,controller.signal); controller.abort(); expect((await result).decision).toBe("cancelled"); });
|
|
});
|