import { describe, expect, it } from "vitest";

import {
  buildGitHubAttachmentFromSearchItem,
  buildLegacyGitHubAttachmentFromSearchItem,
} from "./review-attachments";

describe("buildGitHubAttachmentFromSearchItem", () => {
  it("builds a forge change request attachment for pull requests", () => {
    const attachment = buildGitHubAttachmentFromSearchItem({
      kind: "change_request",
      number: 123,
      title: "Fix race in worktree setup",
      url: "https://github.com/getpaseo/paseo/pull/123",
      state: "OPEN",
      body: "PR body",
      labels: ["bug"],
      baseRefName: "main",
      headRefName: "fix/worktree-race",
    });

    expect(attachment).toEqual({
      type: "forge_change_request",
      mimeType: "application/paseo-forge-change-request",
      forge: "github",
      number: 123,
      title: "Fix race in worktree setup",
      url: "https://github.com/getpaseo/paseo/pull/123",
      body: "PR body",
      baseRefName: "main",
      headRefName: "fix/worktree-race",
    });
  });

  it("builds a forge issue attachment for issues", () => {
    const attachment = buildGitHubAttachmentFromSearchItem({
      kind: "issue",
      number: 55,
      title: "Improve startup error details",
      url: "https://github.com/getpaseo/paseo/issues/55",
      state: "OPEN",
      body: "Issue body",
      labels: ["bug"],
    });

    expect(attachment).toEqual({
      type: "forge_issue",
      mimeType: "application/paseo-forge-issue",
      forge: "github",
      number: 55,
      title: "Improve startup error details",
      url: "https://github.com/getpaseo/paseo/issues/55",
      body: "Issue body",
    });
  });

  it("returns null when no item is selected", () => {
    expect(buildGitHubAttachmentFromSearchItem(null)).toBeNull();
  });
});

describe("buildLegacyGitHubAttachmentFromSearchItem", () => {
  it("builds a legacy GitHub PR attachment for old daemons", () => {
    const attachment = buildLegacyGitHubAttachmentFromSearchItem({
      kind: "change_request",
      number: 123,
      title: "Fix race in worktree setup",
      url: "https://github.com/getpaseo/paseo/pull/123",
      state: "OPEN",
      body: "PR body",
      labels: ["bug"],
      baseRefName: "main",
      headRefName: "fix/worktree-race",
    });

    expect(attachment).toEqual({
      type: "github_pr",
      mimeType: "application/github-pr",
      number: 123,
      title: "Fix race in worktree setup",
      url: "https://github.com/getpaseo/paseo/pull/123",
      body: "PR body",
      baseRefName: "main",
      headRefName: "fix/worktree-race",
    });
  });
});