68 lines
4.8 KiB
TypeScript
68 lines
4.8 KiB
TypeScript
import { describe, expect, it, vi } from "vitest";
|
|
import { MessageDeliveryAutoRecoveryHandler } from "../auto-recovery-handlers/message-delivery.js";
|
|
|
|
describe("MessageDeliveryAutoRecoveryHandler", () => {
|
|
it("returns delivered on first attempt", async () => {
|
|
const runAudit = { database: vi.fn(), git: vi.fn(), filesystem: vi.fn() } as any;
|
|
const handler = new MessageDeliveryAutoRecoveryHandler({ runAudit, sleep: vi.fn() });
|
|
const result = await handler.runWithBoundedRetry({ run: async () => ({ id: "m1" }), correlation: { kind: "direct", fromAgentId: "a1", toId: "a2" } }, { mode: "programmatic", maxRetries: 3 });
|
|
expect(result.outcome).toBe("delivered");
|
|
});
|
|
|
|
it("retries transient failures then delivers", async () => {
|
|
const runAudit = { database: vi.fn(), git: vi.fn(), filesystem: vi.fn() } as any;
|
|
const sleep = vi.fn(async () => {});
|
|
const handler = new MessageDeliveryAutoRecoveryHandler({ runAudit, sleep });
|
|
const run = vi
|
|
.fn<() => Promise<{ id: string }>>()
|
|
.mockRejectedValueOnce(Object.assign(new Error("SQLITE_BUSY"), { code: "SQLITE_BUSY" }))
|
|
.mockRejectedValueOnce(new Error("timeout"))
|
|
.mockResolvedValue({ id: "m2" });
|
|
const result = await handler.runWithBoundedRetry({ run, correlation: { kind: "room", fromAgentId: "a1", roomId: "r1" } }, { mode: "programmatic", maxRetries: 3 });
|
|
expect(result.outcome).toBe("delivered");
|
|
expect(run).toHaveBeenCalledTimes(3);
|
|
expect(runAudit.database).toHaveBeenCalledWith(expect.objectContaining({ type: "message-delivery:retry-issued" }));
|
|
});
|
|
|
|
it("parks after transient retries are exhausted", async () => {
|
|
const runAudit = { database: vi.fn(), git: vi.fn(), filesystem: vi.fn() } as any;
|
|
const run = vi.fn().mockRejectedValue(Object.assign(new Error("SQLITE_BUSY"), { code: "SQLITE_BUSY" }));
|
|
const handler = new MessageDeliveryAutoRecoveryHandler({ runAudit, sleep: vi.fn(async () => {}) });
|
|
const result = await handler.runWithBoundedRetry({ run, correlation: { kind: "room", fromAgentId: "a1", roomId: "r1" } }, { mode: "programmatic", maxRetries: 3 });
|
|
expect(result.outcome).toBe("parked");
|
|
expect(run).toHaveBeenCalledTimes(3);
|
|
expect(runAudit.database).toHaveBeenCalledWith(expect.objectContaining({ type: "message-delivery:park", metadata: expect.objectContaining({ attempts: 3 }) }));
|
|
});
|
|
|
|
it("parks permanent errors without retry", async () => {
|
|
const runAudit = { database: vi.fn(), git: vi.fn(), filesystem: vi.fn() } as any;
|
|
const run = vi.fn().mockRejectedValue(new Error("Room membership required"));
|
|
const handler = new MessageDeliveryAutoRecoveryHandler({ runAudit, sleep: vi.fn(async () => {}) });
|
|
const result = await handler.runWithBoundedRetry({ run, correlation: { kind: "room", fromAgentId: "a1", roomId: "r1" } }, { mode: "programmatic", maxRetries: 3 });
|
|
expect(result.outcome).toBe("parked");
|
|
expect(run).toHaveBeenCalledTimes(1);
|
|
expect(runAudit.database).toHaveBeenCalledWith(expect.objectContaining({ type: "message-delivery:park" }));
|
|
});
|
|
|
|
it("mode off does not retry transient failures", async () => {
|
|
const runAudit = { database: vi.fn(), git: vi.fn(), filesystem: vi.fn() } as any;
|
|
const run = vi.fn().mockRejectedValue(Object.assign(new Error("SQLITE_BUSY"), { code: "SQLITE_BUSY" }));
|
|
const handler = new MessageDeliveryAutoRecoveryHandler({ runAudit, sleep: vi.fn(async () => {}) });
|
|
const result = await handler.runWithBoundedRetry({ run, correlation: { kind: "direct", fromAgentId: "a1", toId: "a2" } }, { mode: "off", maxRetries: 3 });
|
|
expect(result.outcome).toBe("parked");
|
|
expect(run).toHaveBeenCalledTimes(1);
|
|
expect(runAudit.database).toHaveBeenCalledWith(expect.objectContaining({ type: "message-delivery:park", metadata: expect.objectContaining({ mode: "off" }) }));
|
|
});
|
|
|
|
it("records direct and room correlation metadata", async () => {
|
|
const runAudit = { database: vi.fn(), git: vi.fn(), filesystem: vi.fn() } as any;
|
|
const handler = new MessageDeliveryAutoRecoveryHandler({ runAudit, sleep: vi.fn(async () => {}) });
|
|
|
|
await handler.runWithBoundedRetry({ run: vi.fn().mockRejectedValue(new Error("recipient missing")), correlation: { kind: "direct", fromAgentId: "a1", toId: "a2" } }, { mode: "programmatic", maxRetries: 3 });
|
|
await handler.runWithBoundedRetry({ run: vi.fn().mockRejectedValue(new Error("room missing")), correlation: { kind: "room", fromAgentId: "a1", roomId: "r1" } }, { mode: "programmatic", maxRetries: 3 });
|
|
|
|
expect(runAudit.database).toHaveBeenCalledWith(expect.objectContaining({ metadata: expect.objectContaining({ correlation: expect.objectContaining({ kind: "direct", toId: "a2" }) }) }));
|
|
expect(runAudit.database).toHaveBeenCalledWith(expect.objectContaining({ metadata: expect.objectContaining({ correlation: expect.objectContaining({ kind: "room", roomId: "r1" }) }) }));
|
|
});
|
|
});
|