import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
import { act, renderHook, waitFor } from "@testing-library/react";
import { useWorkspaces } from "../useWorkspaces";
import * as api from "../../api";

vi.mock("../../api", () => ({
  fetchWorkspaces: vi.fn(),
}));

const mockFetchWorkspaces = vi.mocked(api.fetchWorkspaces);

describe("useWorkspaces", () => {
  beforeEach(() => {
    mockFetchWorkspaces.mockReset();
  });

  afterEach(() => {
    vi.clearAllMocks();
    vi.useRealTimers();
  });

  it("loads project and task workspaces", async () => {
    mockFetchWorkspaces.mockResolvedValueOnce({
      project: "/Users/test/repo",
      tasks: [{ id: "FN-123", title: "Feature", worktree: "/Users/test/.worktrees/kb-123" }],
    });

    const { result } = renderHook(() => useWorkspaces());

    await waitFor(() => expect(result.current.loading).toBe(false));

    expect(result.current.projectName).toBe("repo");
    expect(result.current.workspaces).toEqual([
      {
        id: "FN-123",
        label: "FN-123",
        title: "Feature",
        worktree: "/Users/test/.worktrees/kb-123",
        kind: "task",
      },
    ]);
  });

  it("polls for workspace updates", async () => {
    vi.useFakeTimers();
    mockFetchWorkspaces
      .mockResolvedValueOnce({ project: "/repo", tasks: [] })
      .mockResolvedValueOnce({
        project: "/repo",
        tasks: [{ id: "FN-200", title: "Later", worktree: "/repo/.worktrees/kb-200" }],
      });

    const { result, unmount } = renderHook(() => useWorkspaces());

    await act(async () => {
      await Promise.resolve();
      await Promise.resolve();
    });
    expect(result.current.loading).toBe(false);
    expect(result.current.workspaces).toEqual([]);

    await act(async () => {
      await vi.advanceTimersByTimeAsync(10000);
    });

    expect(result.current.workspaces).toHaveLength(1);
    expect(mockFetchWorkspaces).toHaveBeenCalledTimes(2);
    unmount();
  });

  it("surfaces fetch errors", async () => {
    mockFetchWorkspaces.mockRejectedValueOnce(new Error("Failed to load workspaces"));

    const { result } = renderHook(() => useWorkspaces());

    await waitFor(() => expect(result.current.loading).toBe(false));

    expect(result.current.error).toBe("Failed to load workspaces");
    expect(result.current.workspaces).toEqual([]);
  });
});