import { describe, it, expect, vi } from "vitest";
import { render, screen } from "@testing-library/react";
import { MantineProvider } from "@mantine/core";
import { IComment } from "@/features/comment/types/comment.types";

// matchMedia (read by MantineProvider) is stubbed globally in vitest.setup.ts.

// The comment mutation hooks reach out to react-query/network — stub them so the
// component renders in isolation. We only assert the AI-badge rendering branch.
vi.mock("@/features/comment/queries/comment-query", () => ({
  useDeleteCommentMutation: () => ({ mutateAsync: vi.fn() }),
  useResolveCommentMutation: () => ({ mutateAsync: vi.fn() }),
  useUpdateCommentMutation: () => ({ mutateAsync: vi.fn() }),
}));

// CommentEditor pulls in the full TipTap editor stack; replace it with a stub.
vi.mock("@/features/comment/components/comment-editor", () => ({
  default: () => <div data-testid="comment-editor" />,
}));

import CommentListItem from "./comment-list-item";

const baseComment = (over?: Partial<IComment>): IComment =>
  ({
    id: "c-1",
    content: JSON.stringify({ type: "doc", content: [] }),
    creatorId: "user-1",
    pageId: "page-1",
    workspaceId: "ws-1",
    createdAt: new Date(),
    creator: { id: "user-1", name: "Service Bot", avatarUrl: null } as any,
    ...over,
  }) as IComment;

function renderItem(comment: IComment) {
  return render(
    <MantineProvider>
      <CommentListItem comment={comment} pageId="page-1" canComment={true} />
    </MantineProvider>,
  );
}

describe("CommentListItem — AI badge", () => {
  it('renders the AI-agent badge when createdSource === "agent"', () => {
    renderItem(baseComment({ createdSource: "agent", aiChatId: null }));
    expect(screen.getByText("AI-agent")).toBeDefined();
    expect(screen.getByText("Service Bot")).toBeDefined();
  });

  it('does NOT render the badge for a normal user comment (createdSource "user")', () => {
    renderItem(baseComment({ createdSource: "user" }));
    expect(screen.queryByText("AI-agent")).toBeNull();
    expect(screen.getByText("Service Bot")).toBeDefined();
  });

  // The non-clickable (null aiChatId) branch is a property of AiAgentBadge itself
  // and is covered in ai-agent-badge.test.tsx; this integration suite only needs
  // the insertion gate (agent → badge, user → no badge) above (#143 review).
});