PaperClipAI/ui/src/components/WorkspaceBaseRefRecoveryNotice.test.tsx

// @vitest-environment jsdom
import { act } from "react";
import { createRoot, type Root } from "react-dom/client";
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import { WorkspaceBaseRefRecoveryNotice, type WorkspaceBaseRefRecoveryNoticeProps } from "./WorkspaceBaseRefRecoveryNotice";

const evidence = { requestedRef: "main", repository: "paperclipai/paperclip", defaultBranch: "master", agentName: "Felix", failureKind: "missing_branch" as const };
describe("workspace branch repair", () => {
  let container: HTMLDivElement;
  let root: Root;
  beforeEach(() => { container = document.createElement("div"); document.body.append(container); root = createRoot(container); });
  afterEach(async () => { await act(async () => root.unmount()); container.remove(); });
  const button = (name: string) => Array.from(container.querySelectorAll("button")).find(b => b.textContent === name)!;
  const render = async (props: Partial<WorkspaceBaseRefRecoveryNoticeProps> = {}) => act(async () => { root.render(<WorkspaceBaseRefRecoveryNotice {...evidence} onRepair={async () => {}} {...props} />); });

  it("shows the missing branch and verified default without a secrets diagnosis", async () => {
    await render();
    expect(container.textContent).toContain("Branch main was not found");
    expect(container.textContent).toContain("Felix hasn’t started");
    expect(container.textContent).not.toContain("secret");
    expect(button("Use master & retry").disabled).toBe(false);
    expect(container.textContent).toContain("Changes only this task");
  });
  it("waits for a repair receipt and prevents duplicate submissions", async () => {
    let finish!: () => void;
    const repair = vi.fn(() => new Promise<void>(resolve => { finish = resolve; }));
    await render({ onRepair: repair });
    await act(async () => { button("Use master & retry").click(); button("Use master & retry")?.click(); });
    expect(repair).toHaveBeenCalledExactlyOnceWith("master");
    expect(button("Saving & requesting retry…").disabled).toBe(true);
    expect(container.textContent).not.toContain("Branch updated");
    await act(async () => finish());
    expect(container.textContent).toContain("Branch updated · retry requested");
    expect(button("Use master & retry")).toBeUndefined();
  });
  it("keeps a custom branch after failure and permits correction", async () => {
    const repair = vi.fn().mockRejectedValue(new Error("Branch was not found."));
    await render({ onRepair: repair });
    await act(async () => button("Choose another branch").click());
    const input = container.querySelector("input")!;
    await act(async () => {
      Object.getOwnPropertyDescriptor(HTMLInputElement.prototype, "value")!.set!.call(input, "release/next");
      input.dispatchEvent(new Event("input", { bubbles: true }));
    });
    await act(async () => container.querySelector("form")!.dispatchEvent(new Event("submit", { bubbles: true, cancelable: true })));
    expect(repair).toHaveBeenCalledExactlyOnceWith("release/next");
    expect(input.value).toBe("release/next");
    expect(container.querySelector('[role="alert"]')?.textContent).toContain("Branch was not found");
    expect(button("Save branch & retry").disabled).toBe(false);
    expect(container.textContent).not.toContain("Branch updated");
  });
  it("rejects invalid branch syntax before sending a repair", async () => {
    const repair = vi.fn();
    await render({ onRepair: repair, defaultBranch: "bad..branch" });
    await act(async () => button("Use bad..branch & retry").click());
    expect(repair).not.toHaveBeenCalled();
    expect(container.querySelector('[role="alert"]')?.textContent).toContain("Enter a valid branch name");
  });
  it("does not guess a default branch or call an unresolved ref missing", async () => {
    await render({ defaultBranch: null, failureKind: "unresolved_ref" });
    expect(button("Use master & retry")).toBeUndefined();
    expect(container.textContent).not.toContain("has no branch");
    expect(container.querySelector("input")?.value).toBe("");
    expect(button("Save branch & retry").disabled).toBe(true);
  });
  it("does not submit a disabled repair", async () => {
    const repair = vi.fn();
    await render({ onRepair: repair, unavailableReason: "The task is paused." });
    expect(button("Use master & retry").disabled).toBe(true);
    await act(async () => button("Use master & retry").click());
    expect(repair).not.toHaveBeenCalled();
  });
});