import { render, screen } from '@testing-library/react';
import userEvent from '@testing-library/user-event';
import { vi, test, expect } from 'vitest';

vi.mock('../../utils/marksExport.js', async (orig) => ({ ...(await orig()), downloadMarksFile: vi.fn() }));
import MarksSheet from './MarksSheet.jsx';
import { downloadMarksFile } from '../../utils/marksExport.js';

const base = { open: true, onClose: () => {}, colorLabels: { '#bbf7d0': 'Promises' }, onSetLabel: () => {}, onJump: () => {}, onRemoveBookmark: () => {}, onRemoveHighlight: () => {} };
const highlights = [{ id: 'h', color: '#bbf7d0', book: 'JHN', bookName: 'John', bookIndex: 42, chapter: 3, verse: 16, startOffset: 0, endOffset: 5, text: 'For God' }];

test('highlights and bookmarks can be exported as Markdown, named by your colour names', async () => {
  const user = userEvent.setup();
  render(<MarksSheet {...base} bookmarks={[]} highlights={highlights} />);
  await user.click(screen.getByRole('button', { name: 'Markdown' }));
  const [blob, name] = downloadMarksFile.mock.calls[0];
  expect(name).toBe('highlights-and-bookmarks.md');
  expect(await blob.text()).toContain('## Promises (green)');
});

test('Word export is offered too, and nothing to export means no buttons', async () => {
  const user = userEvent.setup();
  const { rerender } = render(<MarksSheet {...base} bookmarks={[]} highlights={highlights} />);
  await user.click(screen.getByRole('button', { name: 'Word (.docx)' }));
  await vi.waitFor(() => expect(downloadMarksFile).toHaveBeenCalledTimes(2));
  expect(downloadMarksFile.mock.calls[1][1]).toBe('highlights-and-bookmarks.docx');
  rerender(<MarksSheet {...base} bookmarks={[]} highlights={[]} />);
  expect(screen.queryByRole('group', { name: /export your marks/i })).toBeNull();
});