@akiojin/gwt 4.10.0 → 4.12.0

package/src/cli/ui/__tests__/integration/navigation.test.tsx

@@ -1,514 +0,0 @@
-/**
- * @vitest-environment happy-dom
- */
-import { describe, it, expect, beforeEach, afterAll, vi } from "vitest";
-import type { Mock } from "vitest";
-import { render, waitFor } from "@testing-library/react";
-import { act } from "react-dom/test-utils";
-import React from "react";
-import { App } from "../../components/App.js";
-import { Window } from "happy-dom";
-import type { BranchInfo, BranchItem } from "../../types.js";
-import type { BranchListScreenProps } from "../../components/screens/BranchListScreen.js";
-import type { BranchActionSelectorScreenProps } from "../../screens/BranchActionSelectorScreen.js";
-
-const branchListProps: BranchListScreenProps[] = [];
-const branchActionProps: BranchActionSelectorScreenProps[] = [];
-
-vi.mock("../../../../git.ts", () => ({
-  __esModule: true,
-  getAllBranches: vi.fn(),
-  getRepositoryRoot: vi.fn(async () => "/repo"),
-  deleteBranch: vi.fn(async () => undefined),
-  fetchAllRemotes: vi.fn(async () => undefined),
-  collectUpstreamMap: vi.fn(async () => new Map()),
-  getBranchDivergenceStatuses: vi.fn(async () => []),
-}));
-
-vi.mock("../../../../config/index.js", () => ({
-  loadSession: vi.fn(),
-  getLastToolUsageMap: vi.fn(),
-}));
-
-vi.mock(
-  "../../components/screens/BranchListScreen.js",
-  async (importOriginal) => {
-    const actual =
-      await importOriginal<
-        typeof import("../../components/screens/BranchListScreen.js")
-      >();
-    return {
-      ...actual,
-      BranchListScreen: (props: BranchListScreenProps) => {
-        branchListProps.push(props);
-        return React.createElement(actual.BranchListScreen, props);
-      },
-    };
-  },
-);
-
-vi.mock(
-  "../../screens/BranchActionSelectorScreen.js",
-  async (importOriginal) => {
-    const actual =
-      await importOriginal<
-        typeof import("../../screens/BranchActionSelectorScreen.js")
-      >();
-    return {
-      ...actual,
-      BranchActionSelectorScreen: (props: BranchActionSelectorScreenProps) => {
-        branchActionProps.push(props);
-        return React.createElement(actual.BranchActionSelectorScreen, props);
-      },
-    };
-  },
-);
-
-const { mockIsProtectedBranchName, mockSwitchToProtectedBranch } = vi.hoisted(
-  () => ({
-    mockIsProtectedBranchName: vi.fn(() => false),
-    mockSwitchToProtectedBranch: vi.fn(async () => "none" as const),
-  }),
-);
-
-vi.mock("../../../../worktree.ts", () => ({
-  __esModule: true,
-  listAdditionalWorktrees: vi.fn(),
-  createWorktree: vi.fn(async () => undefined),
-  generateWorktreePath: vi.fn(async () => "/repo/.git/worktree/test"),
-  getMergedPRWorktrees: vi.fn(async () => []),
-  removeWorktree: vi.fn(async () => undefined),
-  isProtectedBranchName: mockIsProtectedBranchName,
-  switchToProtectedBranch: mockSwitchToProtectedBranch,
-}));
-
-const aiToolScreenProps: unknown[] = [];
-
-vi.mock("../../components/screens/AIToolSelectorScreen.js", () => {
-  return {
-    AIToolSelectorScreen: (props: unknown) => {
-      aiToolScreenProps.push(props);
-      return React.createElement("div");
-    },
-  };
-});
-
-import {
-  getAllBranches,
-  getRepositoryRoot,
-  deleteBranch,
-  fetchAllRemotes,
-} from "../../../../git.ts";
-import {
-  listAdditionalWorktrees,
-  createWorktree,
-  generateWorktreePath,
-  getMergedPRWorktrees,
-  removeWorktree,
-} from "../../../../worktree.ts";
-import { loadSession, getLastToolUsageMap } from "../../../../config/index.js";
-
-const mockedGetAllBranches = getAllBranches as Mock;
-const mockedGetRepositoryRoot = getRepositoryRoot as Mock;
-const mockedDeleteBranch = deleteBranch as Mock;
-const mockedFetchAllRemotes = fetchAllRemotes as Mock;
-const mockedListAdditionalWorktrees = listAdditionalWorktrees as Mock;
-const mockedCreateWorktree = createWorktree as Mock;
-const mockedGenerateWorktreePath = generateWorktreePath as Mock;
-const mockedGetMergedPRWorktrees = getMergedPRWorktrees as Mock;
-const mockedRemoveWorktree = removeWorktree as Mock;
-const mockedIsProtectedBranchName = mockIsProtectedBranchName as Mock;
-const mockedSwitchToProtectedBranch = mockSwitchToProtectedBranch as Mock;
-const mockedLoadSession = loadSession as Mock;
-const mockedGetLastToolUsageMap = getLastToolUsageMap as Mock;
-
-describe("Navigation Integration Tests", () => {
-  beforeEach(() => {
-    // Setup happy-dom
-    const window = new Window();
-    globalThis.window = window as unknown as typeof globalThis.window;
-    globalThis.document =
-      window.document as unknown as typeof globalThis.document;
-
-    // Reset mocks
-    mockedGetAllBranches.mockReset();
-    mockedListAdditionalWorktrees.mockReset();
-    mockedGetRepositoryRoot.mockReset();
-    mockedDeleteBranch.mockReset();
-    mockedFetchAllRemotes.mockReset();
-    mockedCreateWorktree.mockReset();
-    mockedGenerateWorktreePath.mockReset();
-    mockedGetMergedPRWorktrees.mockReset();
-    mockedRemoveWorktree.mockReset();
-    mockedIsProtectedBranchName.mockReset();
-    mockedSwitchToProtectedBranch.mockReset();
-    mockedLoadSession.mockReset();
-    mockedGetLastToolUsageMap.mockReset();
-    branchListProps.length = 0;
-    branchActionProps.length = 0;
-    aiToolScreenProps.length = 0;
-    mockedGetRepositoryRoot.mockResolvedValue("/repo");
-    mockedSwitchToProtectedBranch.mockResolvedValue("local");
-    mockedLoadSession.mockResolvedValue({ history: [] });
-    mockedGetLastToolUsageMap.mockResolvedValue(new Map());
-  });
-
-  const mockBranches: BranchInfo[] = [
-    {
-      name: "main",
-      type: "local",
-      branchType: "main",
-      isCurrent: true,
-    },
-    {
-      name: "feature/test",
-      type: "local",
-      branchType: "feature",
-      isCurrent: false,
-    },
-  ];
-
-  it("should start with branch-list screen", async () => {
-    (getAllBranches as ReturnType<typeof vi.fn>).mockResolvedValue(
-      mockBranches,
-    );
-    (listAdditionalWorktrees as ReturnType<typeof vi.fn>).mockResolvedValue([]);
-
-    const onExit = vi.fn();
-    const { getByText } = render(<App onExit={onExit} />);
-
-    await waitFor(() => {
-      expect(getByText(/gwt - Branch Selection/i)).toBeDefined();
-      expect(getByText(/main/)).toBeDefined();
-    });
-  });
-
-  it("should support navigation between screens", async () => {
-    (getAllBranches as ReturnType<typeof vi.fn>).mockResolvedValue(
-      mockBranches,
-    );
-    (listAdditionalWorktrees as ReturnType<typeof vi.fn>).mockResolvedValue([]);
-
-    const onExit = vi.fn();
-    const { container } = render(<App onExit={onExit} />);
-
-    await waitFor(() => {
-      expect(container).toBeDefined();
-    });
-
-    // Test will verify screen navigation
-    expect(container).toBeDefined();
-  });
-
-  it("should maintain state across screen transitions", async () => {
-    (getAllBranches as ReturnType<typeof vi.fn>).mockResolvedValue(
-      mockBranches,
-    );
-    (listAdditionalWorktrees as ReturnType<typeof vi.fn>).mockResolvedValue([]);
-
-    const onExit = vi.fn();
-    const { container } = render(<App onExit={onExit} />);
-
-    await waitFor(() => {
-      expect(container).toBeDefined();
-    });
-
-    // Test will verify state persistence
-    expect(container).toBeDefined();
-  });
-
-  it("should handle back navigation correctly", async () => {
-    (getAllBranches as ReturnType<typeof vi.fn>).mockResolvedValue(
-      mockBranches,
-    );
-    (listAdditionalWorktrees as ReturnType<typeof vi.fn>).mockResolvedValue([]);
-
-    const onExit = vi.fn();
-    const { container } = render(<App onExit={onExit} />);
-
-    await waitFor(() => {
-      expect(container).toBeDefined();
-    });
-
-    // Test will verify back navigation
-    expect(container).toBeDefined();
-  });
-
-  it("should handle navigation history", async () => {
-    (getAllBranches as ReturnType<typeof vi.fn>).mockResolvedValue(
-      mockBranches,
-    );
-    (listAdditionalWorktrees as ReturnType<typeof vi.fn>).mockResolvedValue([]);
-
-    const onExit = vi.fn();
-    const { container } = render(<App onExit={onExit} />);
-
-    await waitFor(() => {
-      expect(container).toBeDefined();
-    });
-
-    // Test will verify navigation history
-    expect(container).toBeDefined();
-  });
-
-  it("should display correct screen on navigation", async () => {
-    (getAllBranches as ReturnType<typeof vi.fn>).mockResolvedValue(
-      mockBranches,
-    );
-    (listAdditionalWorktrees as ReturnType<typeof vi.fn>).mockResolvedValue([]);
-
-    const onExit = vi.fn();
-    const { container } = render(<App onExit={onExit} />);
-
-    await waitFor(() => {
-      expect(container).toBeDefined();
-    });
-
-    // Test will verify correct screen rendering
-    expect(container).toBeDefined();
-  });
-
-  it("should call onExit when branch is selected", async () => {
-    (getAllBranches as ReturnType<typeof vi.fn>).mockResolvedValue(
-      mockBranches,
-    );
-    (listAdditionalWorktrees as ReturnType<typeof vi.fn>).mockResolvedValue([]);
-
-    const onExit = vi.fn();
-    const { container } = render(<App onExit={onExit} />);
-
-    await waitFor(() => {
-      expect(container).toBeDefined();
-    });
-
-    // Test will verify onExit is called
-    expect(container).toBeDefined();
-  });
-});
-
-describe("Protected Branch Navigation (T103)", () => {
-  const baseBranches: BranchInfo[] = [
-    {
-      name: "main",
-      type: "local",
-      branchType: "main",
-      isCurrent: true,
-    },
-    {
-      name: "feature/test",
-      type: "local",
-      branchType: "feature",
-      isCurrent: false,
-    },
-  ];
-
-  beforeEach(() => {
-    const window = new Window();
-    globalThis.window = window as unknown as typeof globalThis.window;
-    globalThis.document =
-      window.document as unknown as typeof globalThis.document;
-    mockedGetAllBranches.mockReset();
-    mockedListAdditionalWorktrees.mockReset();
-    mockedGetRepositoryRoot.mockReset();
-    mockedDeleteBranch.mockReset();
-    mockedFetchAllRemotes.mockReset();
-    mockedCreateWorktree.mockReset();
-    mockedGenerateWorktreePath.mockReset();
-    mockedGetMergedPRWorktrees.mockReset();
-    mockedRemoveWorktree.mockReset();
-    mockedIsProtectedBranchName.mockReset();
-    mockedSwitchToProtectedBranch.mockReset();
-    mockedLoadSession.mockReset();
-    mockedGetLastToolUsageMap.mockReset();
-    mockedGetRepositoryRoot.mockResolvedValue("/repo");
-    branchListProps.length = 0;
-    branchActionProps.length = 0;
-    aiToolScreenProps.length = 0;
-    mockedIsProtectedBranchName.mockImplementation((name: string) =>
-      ["main", "develop", "origin/main", "origin/develop"].includes(name),
-    );
-    mockedSwitchToProtectedBranch.mockResolvedValue("local");
-    mockedGetRepositoryRoot.mockResolvedValue("/repo");
-    mockedLoadSession.mockResolvedValue({ history: [] });
-    mockedGetLastToolUsageMap.mockResolvedValue(new Map());
-  });
-
-  it("prefills AI tool selector with last used tool for the branch", async () => {
-    const branchesWithUsage: BranchInfo[] = [
-      {
-        name: "feature/with-usage",
-        type: "local",
-        branchType: "feature",
-        isCurrent: false,
-        lastToolUsage: {
-          branch: "feature/with-usage",
-          worktreePath: "/repo/.worktrees/feature-with-usage",
-          toolId: "codex-cli",
-          toolLabel: "Codex",
-          timestamp: Date.now(),
-        },
-      },
-    ];
-
-    mockedGetAllBranches.mockResolvedValue(branchesWithUsage);
-    mockedListAdditionalWorktrees.mockResolvedValue([]);
-
-    const onExit = vi.fn();
-    render(<App onExit={onExit} />);
-
-    await waitFor(() => {
-      const latestProps = branchListProps.find((p) => p.branches?.length);
-      expect(latestProps?.branches?.length ?? 0).toBeGreaterThan(0);
-    });
-
-    // Simulate selecting the branch from the list
-    const listProps =
-      branchListProps.find((p) => p.branches?.length) ?? branchListProps.at(-1);
-    expect(listProps?.branches?.length).toBeGreaterThan(0);
-    listProps.onSelect(listProps.branches[0]);
-
-    await waitFor(() => {
-      expect(branchActionProps.length).toBeGreaterThan(0);
-    });
-
-    // Use existing branch (non-protected) to navigate to AI tool selector
-    const actionProps = branchActionProps.at(-1);
-    actionProps.onUseExisting();
-
-    await waitFor(() => {
-      expect(aiToolScreenProps.length).toBeGreaterThan(0);
-    });
-
-    const props = aiToolScreenProps.at(-1) as { initialToolId?: string };
-    expect(props.initialToolId).toBe("codex-cli");
-  });
-
-  it("switches local protected branches via root workflow and navigates to AI tool", async () => {
-    mockedGetAllBranches.mockResolvedValue(baseBranches);
-    mockedListAdditionalWorktrees.mockResolvedValue([]);
-
-    const onExit = vi.fn();
-    render(<App onExit={onExit} />);
-
-    await waitFor(() => {
-      expect(branchListProps.length).toBeGreaterThan(0);
-    });
-
-    await waitFor(() => {
-      const latest = branchListProps.at(-1);
-      const names = (latest?.branches as BranchItem[] | undefined)?.map(
-        (item) => item.name,
-      );
-      expect(names).toBeDefined();
-      expect(names).toContain("main");
-    });
-
-    const latestProps = branchListProps.at(-1);
-    const protectedBranch = (latestProps?.branches as BranchItem[]).find(
-      (item) => item.name === "main",
-    );
-    expect(protectedBranch).toBeDefined();
-
-    await act(async () => {
-      latestProps?.onSelect(protectedBranch);
-      await Promise.resolve();
-    });
-
-    await waitFor(() => {
-      expect(branchActionProps.length).toBeGreaterThan(0);
-    });
-
-    const actionProps = branchActionProps.at(-1);
-    expect(actionProps?.mode).toBe("protected");
-    expect(actionProps?.infoMessage).toContain("is a root branch");
-
-    await act(async () => {
-      await actionProps?.onUseExisting();
-      await Promise.resolve();
-    });
-
-    expect(mockedSwitchToProtectedBranch).toHaveBeenCalledWith({
-      branchName: "main",
-      repoRoot: "/repo",
-      remoteRef: null,
-    });
-
-    await waitFor(() => {
-      expect(aiToolScreenProps.length).toBeGreaterThan(0);
-    });
-  });
-
-  it("creates tracking branch for remote protected selections before navigating to AI tool", async () => {
-    const remoteBranches: BranchInfo[] = [
-      {
-        name: "origin/develop",
-        type: "remote",
-        branchType: "develop",
-        isCurrent: false,
-      },
-      {
-        name: "feature/test",
-        type: "local",
-        branchType: "feature",
-        isCurrent: false,
-      },
-    ];
-    mockedGetAllBranches.mockResolvedValue(remoteBranches);
-    mockedListAdditionalWorktrees.mockResolvedValue([]);
-    mockedSwitchToProtectedBranch.mockResolvedValue("remote");
-
-    const onExit = vi.fn();
-    render(<App onExit={onExit} />);
-
-    await waitFor(() => {
-      expect(branchListProps.length).toBeGreaterThan(0);
-    });
-
-    await waitFor(() => {
-      const latest = branchListProps.at(-1);
-      const names = (latest?.branches as BranchItem[] | undefined)?.map(
-        (item) => item.name,
-      );
-      expect(names).toBeDefined();
-      expect(names).toContain("origin/develop");
-    });
-
-    const latestProps = branchListProps.at(-1);
-    const protectedBranch = (latestProps?.branches as BranchItem[]).find(
-      (item) => item.name === "origin/develop",
-    );
-    expect(protectedBranch).toBeDefined();
-
-    await act(async () => {
-      latestProps?.onSelect(protectedBranch);
-      await Promise.resolve();
-    });
-
-    await waitFor(() => {
-      expect(branchActionProps.length).toBeGreaterThan(0);
-    });
-
-    const actionProps = branchActionProps.at(-1);
-    expect(actionProps?.mode).toBe("protected");
-    expect(actionProps?.primaryLabel).toContain("root");
-
-    await act(async () => {
-      await actionProps?.onUseExisting();
-      await Promise.resolve();
-    });
-
-    expect(mockedSwitchToProtectedBranch).toHaveBeenCalledWith({
-      branchName: "develop",
-      repoRoot: "/repo",
-      remoteRef: "origin/develop",
-    });
-
-    await waitFor(() => {
-      expect(aiToolScreenProps.length).toBeGreaterThan(0);
-    });
-  });
-});
-
-afterAll(() => {
-  vi.restoreAllMocks();
-});