forked from farhoodlabs/paperclip
Dotta
31 lines
993 B
TypeScript
31 lines
993 B
TypeScript
import { randomUUID } from "node:crypto";
|
|
import { afterEach, describe, expect, it, vi } from "vitest";
|
|
import { withAgentStartLock } from "../services/agent-start-lock.ts";
|
|
|
|
describe("heartbeat agent start lock", () => {
|
|
afterEach(() => {
|
|
vi.useRealTimers();
|
|
});
|
|
|
|
it("does not let a stale start lock freeze later queued-run starts", async () => {
|
|
vi.useFakeTimers();
|
|
|
|
const agentId = randomUUID();
|
|
const firstStart = vi.fn(() => new Promise<void>(() => undefined));
|
|
const secondStart = vi.fn(async () => "started");
|
|
|
|
void withAgentStartLock(agentId, firstStart);
|
|
await Promise.resolve();
|
|
expect(firstStart).toHaveBeenCalledTimes(1);
|
|
|
|
const secondStartResult = withAgentStartLock(agentId, secondStart);
|
|
await Promise.resolve();
|
|
expect(secondStart).not.toHaveBeenCalled();
|
|
|
|
await vi.advanceTimersByTimeAsync(30_000);
|
|
|
|
await expect(secondStartResult).resolves.toBe("started");
|
|
expect(secondStart).toHaveBeenCalledTimes(1);
|
|
});
|
|
});
|