33601dde24
* feat(agent): add standalone agent runtime * fix(agent): add Bun and DOM types to agent tsconfig * refactor: wrap backup error in a tagged effect error * feat(controller): add agent manager and session handling * feat(backups): execute backups through the agent * fix(agent): harden disconnect and send-failure handling * fix: rebase conflicts * test: simplify mocks * refactor: split agent runtime state * fix(backup): keep old path when agent is disabled * fix: pr feedbacks
47 lines
1.6 KiB
TypeScript
47 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 { 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().mockResolvedValue(true);
|
|
const cancelBackup = vi.fn().mockResolvedValue(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",
|
|
});
|
|
});
|