e65a135676
* feat(agents): create agent registry and service * fix: mark agent offline only if the session was removed properly * refactor: centralize agent backup lifecycle state * refactor: simplify session management * refactor: move effect / async boundary in one place * chore: regen migration * refactor: improve error handling * chore: pr feedback
48 lines
1.6 KiB
TypeScript
48 lines
1.6 KiB
TypeScript
import { afterEach, expect, test, vi } from "vitest";
|
|
import waitForExpect from "wait-for-expect";
|
|
import { fromAny, fromPartial } from "@total-typescript/shoehorn";
|
|
import { Effect } from "effect";
|
|
import { agentManager, type ProcessWithAgentRuntime } from "../agents-manager";
|
|
import type { AgentManagerRuntime } from "../controller/server";
|
|
import type { BackupRunPayload } from "@zerobyte/contracts/agent-protocol";
|
|
|
|
const setAgentRuntime = (agentManagerRuntime: Partial<AgentManagerRuntime> | null) => {
|
|
(process as ProcessWithAgentRuntime).__zerobyteAgentRuntime = {
|
|
agentManager: fromAny(agentManagerRuntime),
|
|
localAgent: null,
|
|
isStoppingLocalAgent: false,
|
|
localAgentRestartTimeout: null,
|
|
};
|
|
};
|
|
|
|
afterEach(() => {
|
|
delete (process as ProcessWithAgentRuntime).__zerobyteAgentRuntime;
|
|
vi.restoreAllMocks();
|
|
});
|
|
|
|
test("cancelBackup resolves a running backup when the cancel command cannot be delivered", async () => {
|
|
const sendBackup = vi.fn(() => Effect.succeed(true));
|
|
const cancelBackup = vi.fn(() => Effect.succeed(false));
|
|
setAgentRuntime({ sendBackup, cancelBackup });
|
|
|
|
const resultPromise = agentManager.runBackup("local", {
|
|
scheduleId: 42,
|
|
payload: fromPartial<BackupRunPayload>({
|
|
jobId: "job-1",
|
|
scheduleId: "schedule-1",
|
|
}),
|
|
signal: new AbortController().signal,
|
|
onProgress: vi.fn(),
|
|
});
|
|
|
|
await waitForExpect(() => {
|
|
expect(sendBackup).toHaveBeenCalledTimes(1);
|
|
});
|
|
|
|
await expect(agentManager.cancelBackup("local", 42)).resolves.toBe(true);
|
|
await expect(resultPromise).resolves.toEqual({ status: "cancelled" });
|
|
expect(cancelBackup).toHaveBeenCalledWith("local", {
|
|
jobId: "job-1",
|
|
scheduleId: "schedule-1",
|
|
});
|
|
});
|