mirror of https://github.com/openclaw/openclaw.git
83 lines
2.8 KiB
TypeScript
83 lines
2.8 KiB
TypeScript
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
|
|
vi.mock("../infra/retry.js", () => ({
|
|
retryAsync: vi.fn(async (run: () => Promise<unknown>) => await run()),
|
|
}));
|
|
|
|
vi.mock("./post-json.js", () => ({
|
|
postJson: vi.fn(),
|
|
}));
|
|
|
|
describe("postJsonWithRetry", () => {
|
|
let retryAsyncMock: ReturnType<typeof vi.mocked<typeof import("../infra/retry.js").retryAsync>>;
|
|
let postJsonMock: ReturnType<typeof vi.mocked<typeof import("./post-json.js").postJson>>;
|
|
let postJsonWithRetry: typeof import("./batch-http.js").postJsonWithRetry;
|
|
|
|
beforeEach(async () => {
|
|
vi.resetModules();
|
|
vi.clearAllMocks();
|
|
({ postJsonWithRetry } = await import("./batch-http.js"));
|
|
const retryModule = await import("../infra/retry.js");
|
|
const postJsonModule = await import("./post-json.js");
|
|
retryAsyncMock = vi.mocked(retryModule.retryAsync);
|
|
postJsonMock = vi.mocked(postJsonModule.postJson);
|
|
});
|
|
|
|
it("posts JSON and returns parsed response payload", async () => {
|
|
postJsonMock.mockImplementationOnce(async (params) => {
|
|
return await params.parse({ ok: true, ids: [1, 2] });
|
|
});
|
|
|
|
const result = await postJsonWithRetry<{ ok: boolean; ids: number[] }>({
|
|
url: "https://memory.example/v1/batch",
|
|
headers: { Authorization: "Bearer test" },
|
|
body: { chunks: ["a", "b"] },
|
|
errorPrefix: "memory batch failed",
|
|
});
|
|
|
|
expect(result).toEqual({ ok: true, ids: [1, 2] });
|
|
expect(postJsonMock).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
url: "https://memory.example/v1/batch",
|
|
headers: { Authorization: "Bearer test" },
|
|
body: { chunks: ["a", "b"] },
|
|
errorPrefix: "memory batch failed",
|
|
attachStatus: true,
|
|
}),
|
|
);
|
|
|
|
const retryOptions = retryAsyncMock.mock.calls[0]?.[1] as
|
|
| {
|
|
attempts: number;
|
|
minDelayMs: number;
|
|
maxDelayMs: number;
|
|
shouldRetry: (err: unknown) => boolean;
|
|
}
|
|
| undefined;
|
|
expect(retryOptions?.attempts).toBe(3);
|
|
expect(retryOptions?.minDelayMs).toBe(300);
|
|
expect(retryOptions?.maxDelayMs).toBe(2000);
|
|
expect(retryOptions?.shouldRetry({ status: 429 })).toBe(true);
|
|
expect(retryOptions?.shouldRetry({ status: 503 })).toBe(true);
|
|
expect(retryOptions?.shouldRetry({ status: 400 })).toBe(false);
|
|
});
|
|
|
|
it("attaches status to non-ok errors", async () => {
|
|
postJsonMock.mockRejectedValueOnce(
|
|
Object.assign(new Error("memory batch failed: 503 backend down"), { status: 503 }),
|
|
);
|
|
|
|
await expect(
|
|
postJsonWithRetry({
|
|
url: "https://memory.example/v1/batch",
|
|
headers: {},
|
|
body: { chunks: [] },
|
|
errorPrefix: "memory batch failed",
|
|
}),
|
|
).rejects.toMatchObject({
|
|
message: expect.stringContaining("memory batch failed: 503 backend down"),
|
|
status: 503,
|
|
});
|
|
});
|
|
});
|