90 lines
3.3 KiB
TypeScript
90 lines
3.3 KiB
TypeScript
import { describe, expect, it, vi } from "vitest";
|
|
import type { RunAuditEventInput, TaskStore } from "@fusion/core";
|
|
import { createRunAuditor } from "../run-audit.js";
|
|
|
|
type WorktrunkLifecycleCase = {
|
|
type:
|
|
| "worktree:worktrunk-install"
|
|
| "worktree:worktrunk-create"
|
|
| "worktree:worktrunk-sync"
|
|
| "worktree:worktrunk-prune"
|
|
| "worktree:worktrunk-remove";
|
|
target: string;
|
|
metadata: Record<string, unknown>;
|
|
};
|
|
|
|
describe("run-audit worktrunk lifecycle events", () => {
|
|
it.each<WorktrunkLifecycleCase>([
|
|
{
|
|
type: "worktree:worktrunk-install",
|
|
target: "/usr/local/bin/worktrunk",
|
|
metadata: { op: "install", binaryPath: "/usr/local/bin/worktrunk", durationMs: 12, installSource: "cargo" },
|
|
},
|
|
{
|
|
type: "worktree:worktrunk-create",
|
|
target: "/repo/.worktrees/fn-1",
|
|
metadata: { op: "create", binaryPath: "/usr/local/bin/worktrunk", worktreePath: "/repo/.worktrees/fn-1", durationMs: 31 },
|
|
},
|
|
{
|
|
type: "worktree:worktrunk-sync",
|
|
target: "/repo/.worktrees/fn-1",
|
|
metadata: { op: "sync", binaryPath: "/usr/local/bin/worktrunk", worktreePath: "/repo/.worktrees/fn-1", durationMs: 44 },
|
|
},
|
|
{
|
|
type: "worktree:worktrunk-prune",
|
|
target: "worktrunk-prune",
|
|
metadata: { op: "prune", binaryPath: "/usr/local/bin/worktrunk", durationMs: 20, prunedCount: 3 },
|
|
},
|
|
{
|
|
type: "worktree:worktrunk-remove",
|
|
target: "/repo/.worktrees/fn-1",
|
|
metadata: { op: "remove", binaryPath: "/usr/local/bin/worktrunk", worktreePath: "/repo/.worktrees/fn-1", durationMs: 13 },
|
|
},
|
|
])("persists $type with metadata", async ({ type, target, metadata }) => {
|
|
const recordRunAuditEvent = vi.fn(async (_event: RunAuditEventInput) => undefined);
|
|
const store = { recordRunAuditEvent } as unknown as TaskStore;
|
|
const auditor = createRunAuditor(store, { runId: "run-1", agentId: "agent-1", taskId: "FN-1", phase: "execute" });
|
|
|
|
await auditor.git({ type, target, metadata });
|
|
|
|
expect(recordRunAuditEvent).toHaveBeenCalledTimes(1);
|
|
expect(recordRunAuditEvent).toHaveBeenCalledWith(expect.objectContaining({
|
|
domain: "git",
|
|
mutationType: type,
|
|
target,
|
|
taskId: "FN-1",
|
|
runId: "run-1",
|
|
metadata: expect.objectContaining(metadata),
|
|
}));
|
|
});
|
|
|
|
it("records long stderr previews without mutating payload content", async () => {
|
|
const recordRunAuditEvent = vi.fn(async (_event: RunAuditEventInput) => undefined);
|
|
const auditor = createRunAuditor({ recordRunAuditEvent } as unknown as TaskStore, {
|
|
runId: "run-1",
|
|
agentId: "agent-1",
|
|
taskId: "FN-1",
|
|
});
|
|
const longPreview = "x".repeat(5000);
|
|
|
|
await auditor.git({
|
|
type: "worktree:worktrunk-failure",
|
|
target: "FN-1",
|
|
metadata: { op: "failure", stderrPreview: longPreview },
|
|
});
|
|
|
|
expect(recordRunAuditEvent.mock.calls[0]?.[0]?.metadata?.stderrPreview).toHaveLength(5000);
|
|
});
|
|
|
|
it("propagates store write failures", async () => {
|
|
const store = {
|
|
recordRunAuditEvent: vi.fn(async () => {
|
|
throw new Error("boom");
|
|
}),
|
|
} as unknown as TaskStore;
|
|
const auditor = createRunAuditor(store, { runId: "run-1", agentId: "agent-1", taskId: "FN-1" });
|
|
|
|
await expect(auditor.git({ type: "worktree:worktrunk-create", target: "/repo/.worktrees/fn-1" })).rejects.toThrow("boom");
|
|
});
|
|
});
|