test(FN-4687): complete Step 5 — add worktrunk acquisition coverage

Fusion-Task-Id: FN-4687
Fusion-Task-Lineage: d646a4dc-e28e-4b1f-ac07-c4ab79e893d5
This commit is contained in:
Fusion (runfusion.ai)
2026-05-15 18:17:05 -07:00
committed by gsxdsm
parent af2435d3aa
commit ffc4174d21

View File

@@ -0,0 +1,176 @@
import { describe, it, expect, vi, beforeEach } from "vitest";
const { execMock } = vi.hoisted(() => {
const mock = vi.fn();
(mock as any)[Symbol.for("nodejs.util.promisify.custom")] = mock;
return { execMock: mock };
});
vi.mock("node:child_process", () => ({ exec: execMock }));
vi.mock("../worktree-pool.js", async () => {
const actual = await vi.importActual<any>("../worktree-pool.js");
return { ...actual, isUsableTaskWorktree: vi.fn().mockResolvedValue(true) };
});
vi.mock("../worktree-db-hydrate.js", () => ({
hydrateWorktreeDb: vi.fn().mockResolvedValue({ degraded: false, tasksCopied: 1, documentsCopied: 1 }),
}));
const task = {
id: "FN-1",
title: "Task",
description: "Desc",
branch: null,
worktree: null,
} as any;
const makeStore = () => ({
updateTask: vi.fn().mockResolvedValue(undefined),
logEntry: vi.fn().mockResolvedValue(undefined),
});
const makeAudit = () => {
const events: Array<{ type: string; target: string; metadata?: Record<string, unknown> }> = [];
return {
events,
audit: {
git: vi.fn(async (event) => {
events.push(event);
}),
},
};
};
beforeEach(() => {
execMock.mockReset();
});
describe("acquireTaskWorktree worktrunk wiring", () => {
it("uses native by default when worktrunk settings absent", async () => {
execMock.mockResolvedValue({ stdout: "", stderr: "" });
const { acquireTaskWorktree } = await import("../worktree-acquisition.js");
const result = await acquireTaskWorktree({
task,
rootDir: "/repo",
store: makeStore() as any,
settings: {},
});
expect(result).toMatchObject({ source: "fresh", branch: "fusion/fn-1" });
expect(execMock).toHaveBeenCalledTimes(1);
expect(execMock.mock.calls[0]?.[0]).toContain("git worktree add -b");
});
it("prefers explicit createWorktree override", async () => {
const createWorktree = vi.fn().mockResolvedValue({ path: "/tmp/new", branch: "fusion/fn-1" });
const { acquireTaskWorktree } = await import("../worktree-acquisition.js");
await acquireTaskWorktree({
task,
rootDir: "/repo",
store: makeStore() as any,
settings: { worktrunk: { enabled: true, binaryPath: "worktrunk" } } as any,
createWorktree,
});
expect(createWorktree).toHaveBeenCalledTimes(1);
expect(execMock).not.toHaveBeenCalled();
});
it("emits worktrunk + native create audits when worktrunk succeeds", async () => {
execMock.mockResolvedValue({ stdout: "", stderr: "" });
const { acquireTaskWorktree } = await import("../worktree-acquisition.js");
const { audit, events } = makeAudit();
await acquireTaskWorktree({
task,
rootDir: "/repo",
store: makeStore() as any,
settings: { worktrunk: { enabled: true, binaryPath: "worktrunk", onFailure: "fail" } } as any,
audit: audit as any,
});
expect(execMock).toHaveBeenCalledWith('"worktrunk" --help', expect.any(Object));
expect(execMock).toHaveBeenCalledTimes(1);
expect(events.filter((event) => event.type === "worktree:worktrunk-create")).toHaveLength(1);
expect(events.filter((event) => event.type === "worktree:create")).toHaveLength(1);
});
it("fails hard without fallback when onFailure=fail", async () => {
execMock.mockRejectedValue({ stderr: "nope", status: 9 });
const { acquireTaskWorktree } = await import("../worktree-acquisition.js");
const { audit, events } = makeAudit();
await expect(
acquireTaskWorktree({
task,
rootDir: "/repo",
store: makeStore() as any,
settings: { worktrunk: { enabled: true, binaryPath: "worktrunk", onFailure: "fail" } } as any,
audit: audit as any,
}),
).rejects.toMatchObject({ code: "worktrunk_operation_failed", operation: "create" });
expect(execMock).toHaveBeenCalledTimes(1);
expect(execMock.mock.calls[0]?.[0]).toBe('"worktrunk" --help');
expect(events.some((event) => event.type === "worktree:worktrunk-fallback")).toBe(false);
});
it("falls back to native when onFailure=fallback-native", async () => {
execMock
.mockRejectedValueOnce({ stderr: "broken", status: 3 })
.mockResolvedValueOnce({ stdout: "", stderr: "" });
const { acquireTaskWorktree } = await import("../worktree-acquisition.js");
const { audit, events } = makeAudit();
await acquireTaskWorktree({
task,
rootDir: "/repo",
store: makeStore() as any,
settings: { worktrunk: { enabled: true, binaryPath: "worktrunk", onFailure: "fallback-native" } } as any,
audit: audit as any,
logger: { log: vi.fn(), warn: vi.fn(), error: vi.fn() },
});
expect(execMock).toHaveBeenCalledTimes(2);
expect(execMock.mock.calls[0]?.[0]).toBe('"worktrunk" --help');
expect(execMock.mock.calls[1]?.[0]).toContain("git worktree add -b");
expect(events.filter((event) => event.type === "worktree:worktrunk-fallback")).toHaveLength(1);
});
it("fails with binary missing when enabled and binaryPath absent", async () => {
const { acquireTaskWorktree } = await import("../worktree-acquisition.js");
await expect(
acquireTaskWorktree({
task,
rootDir: "/repo",
store: makeStore() as any,
settings: { worktrunk: { enabled: true, onFailure: "fail" } } as any,
}),
).rejects.toMatchObject({ code: "worktrunk_binary_missing", operation: "create" });
});
it("uses custom backend when provided", async () => {
const create = vi.fn().mockResolvedValue({ path: "/tmp/custom", branch: "fusion/fn-1-custom" });
const { acquireTaskWorktree } = await import("../worktree-acquisition.js");
const result = await acquireTaskWorktree({
task,
rootDir: "/repo",
store: makeStore() as any,
settings: { worktrunk: { enabled: true, binaryPath: "worktrunk" } } as any,
backend: {
kind: "native",
create,
remove: vi.fn(),
sync: vi.fn().mockResolvedValue({ skipped: true as const }),
prune: vi.fn(),
},
});
expect(result.branch).toBe("fusion/fn-1-custom");
expect(create).toHaveBeenCalledTimes(1);
expect(execMock).not.toHaveBeenCalled();
});
});