import { render, screen, waitFor } from '@testing-library/react';
import userEvent from '@testing-library/user-event';
import { describe, it, expect, vi } from 'vitest';
import {
  Menubar,
  MenubarMenu,
  MenubarTrigger,
  MenubarContent,
  MenubarItem,
  MenubarSeparator,
} from './menubar';
import React from 'react';

// Mocking Portal
vi.mock('@radix-ui/react-menubar', async () => {
  const actual = await vi.importActual('@radix-ui/react-menubar');
  return {
    ...actual,
    Portal: ({ children }: { children: React.ReactNode }) => (
      <div data-testid="portal">{children}</div>
    ),
  };
});

describe('Menubar', () => {
  it('renders correctly', () => {
    render(
      <Menubar>
        <MenubarMenu>
          <MenubarTrigger>File</MenubarTrigger>
          <MenubarContent>
            <MenubarItem>New Tab</MenubarItem>
            <MenubarSeparator />
            <MenubarItem>Share</MenubarItem>
          </MenubarContent>
        </MenubarMenu>
      </Menubar>
    );

    expect(screen.getByText('File')).toBeInTheDocument();
  });

  it('has correct data-slot attributes', () => {
    const { container } = render(
      <Menubar>
        <MenubarMenu>
          <MenubarTrigger>File</MenubarTrigger>
        </MenubarMenu>
      </Menubar>
    );
    expect(container.querySelector('[data-slot="menubar"]')).toBeInTheDocument();
    expect(container.querySelector('[data-slot="menubar-trigger"]')).toBeInTheDocument();
  });

  it('opens content when the trigger is clicked', async () => {
    const user = userEvent.setup();
    render(
      <Menubar>
        <MenubarMenu>
          <MenubarTrigger>File</MenubarTrigger>
          <MenubarContent>
            <MenubarItem>New Tab</MenubarItem>
            <MenubarSeparator />
            <MenubarItem>Share</MenubarItem>
          </MenubarContent>
        </MenubarMenu>
      </Menubar>
    );

    expect(screen.queryByText('New Tab')).not.toBeInTheDocument();

    await user.click(screen.getByText('File'));

    await waitFor(() => {
      expect(screen.getByText('New Tab')).toBeInTheDocument();
    });
  });

  it('closes when Escape is pressed', async () => {
    const user = userEvent.setup();
    render(
      <Menubar>
        <MenubarMenu>
          <MenubarTrigger>File</MenubarTrigger>
          <MenubarContent>
            <MenubarItem>New Tab</MenubarItem>
          </MenubarContent>
        </MenubarMenu>
      </Menubar>
    );

    await user.click(screen.getByText('File'));
    await waitFor(() => {
      expect(screen.getByText('New Tab')).toBeInTheDocument();
    });

    await user.keyboard('{Escape}');

    await waitFor(() => {
      expect(screen.queryByText('New Tab')).not.toBeInTheDocument();
    });
  });

  it('triggers onSelect and closes when a menu item is clicked', async () => {
    const user = userEvent.setup();
    const onSelect = vi.fn();
    render(
      <Menubar>
        <MenubarMenu>
          <MenubarTrigger>File</MenubarTrigger>
          <MenubarContent>
            <MenubarItem onSelect={onSelect}>New Tab</MenubarItem>
          </MenubarContent>
        </MenubarMenu>
      </Menubar>
    );

    await user.click(screen.getByText('File'));
    await waitFor(() => {
      expect(screen.getByText('New Tab')).toBeInTheDocument();
    });

    await user.click(screen.getByText('New Tab'));

    await waitFor(() => {
      expect(onSelect).toHaveBeenCalled();
    });
    await waitFor(() => {
      expect(screen.queryByText('New Tab')).not.toBeInTheDocument();
    });
  });
});