|
|
import { afterEach, describe, expect, it } from "vitest";
|
|
|
import fs from "node:fs";
|
|
|
import path from "node:path";
|
|
|
import {
|
|
|
isFlockNonRetryableFailure,
|
|
|
releaseFlockExclusiveLock,
|
|
|
withFlockExclusiveLock,
|
|
|
} from "@/lib/flock/exclusive-lock";
|
|
|
|
|
|
const LOCK_PATH = path.join(process.cwd(), ".rpa", "flock-rpa.lock");
|
|
|
|
|
|
afterEach(() => {
|
|
|
releaseFlockExclusiveLock("test");
|
|
|
try {
|
|
|
fs.unlinkSync(LOCK_PATH);
|
|
|
} catch {
|
|
|
/* ignore */
|
|
|
}
|
|
|
});
|
|
|
|
|
|
describe("isFlockNonRetryableFailure", () => {
|
|
|
it("识别上限与拒号", () => {
|
|
|
expect(
|
|
|
isFlockNonRetryableFailure(
|
|
|
"FLOCK_QUOTE_LIMIT：Thank you for your interest",
|
|
|
),
|
|
|
).toBe(true);
|
|
|
expect(
|
|
|
isFlockNonRetryableFailure("FLOCK_ACCOUNT_REJECTED：issue creating"),
|
|
|
).toBe(true);
|
|
|
expect(isFlockNonRetryableFailure("互斥锁占用")).toBe(true);
|
|
|
expect(isFlockNonRetryableFailure("网络超时")).toBe(false);
|
|
|
});
|
|
|
});
|
|
|
|
|
|
describe("withFlockExclusiveLock", () => {
|
|
|
it("串行持有锁", async () => {
|
|
|
const order: string[] = [];
|
|
|
await withFlockExclusiveLock("test", async () => {
|
|
|
order.push("a");
|
|
|
expect(fs.existsSync(LOCK_PATH)).toBe(true);
|
|
|
});
|
|
|
expect(fs.existsSync(LOCK_PATH)).toBe(false);
|
|
|
expect(order).toEqual(["a"]);
|
|
|
});
|
|
|
});
|
