/** * Tests for the UploadModal component. * Tests file upload functionality with drag-and-drop support. */ import { describe, it, expect, vi, beforeEach } from 'vitest'; import { screen, fireEvent, waitFor } from '@testing-library/react'; import { render } from '../utils'; import { UploadModal } from '../../components/UploadModal'; import { http, HttpResponse } from 'msw'; import { server } from '../mocks/server'; describe('UploadModal', () => { const mockOnClose = vi.fn(); beforeEach(() => { vi.clearAllMocks(); server.use( http.post('/api/v1/archives/upload-bulk', async () => { return HttpResponse.json({ uploaded: 1, failed: 0, results: [{ id: 1, filename: 'test.3mf' }], errors: [], }); }) ); }); describe('rendering', () => { it('renders the modal with title', () => { render(); expect(screen.getByText('Upload 3MF Files')).toBeInTheDocument(); }); it('renders drag and drop zone', () => { render(); expect(screen.getByText('Drag & drop .3mf files here')).toBeInTheDocument(); }); it('renders Browse Files button', () => { render(); expect(screen.getByRole('button', { name: 'Browse Files' })).toBeInTheDocument(); }); it('renders Cancel button', () => { render(); expect(screen.getByRole('button', { name: 'Cancel' })).toBeInTheDocument(); }); it('renders Upload button (disabled initially)', () => { render(); const uploadButton = screen.getByRole('button', { name: /Upload/i }); expect(uploadButton).toBeDisabled(); }); }); describe('file handling with initialFiles', () => { it('shows initial files when provided', () => { const initialFiles = [ new File(['content'], 'model.3mf', { type: 'application/3mf' }), ]; render(); expect(screen.getByText('model.3mf')).toBeInTheDocument(); }); it('enables Upload button when files are present', () => { const initialFiles = [ new File(['content'], 'model.3mf', { type: 'application/3mf' }), ]; render(); const uploadButton = screen.getByRole('button', { name: /Upload/i }); expect(uploadButton).not.toBeDisabled(); }); it('shows file count in Upload button', () => { const initialFiles = [ new File(['content'], 'model1.3mf', { type: 'application/3mf' }), new File(['content'], 'model2.3mf', { type: 'application/3mf' }), ]; render(); expect(screen.getByRole('button', { name: /Upload \(2\)/i })).toBeInTheDocument(); }); it('filters out non-3mf files from initialFiles', () => { const initialFiles = [ new File(['content'], 'model.3mf', { type: 'application/3mf' }), new File(['content'], 'image.png', { type: 'image/png' }), new File(['content'], 'doc.txt', { type: 'text/plain' }), ]; render(); expect(screen.getByText('model.3mf')).toBeInTheDocument(); expect(screen.queryByText('image.png')).not.toBeInTheDocument(); expect(screen.queryByText('doc.txt')).not.toBeInTheDocument(); }); }); describe('file removal', () => { it('allows removing a file before upload', async () => { const initialFiles = [ new File(['content'], 'model.3mf', { type: 'application/3mf' }), ]; render(); expect(screen.getByText('model.3mf')).toBeInTheDocument(); // Find and click the remove button (X icon next to file) const fileItem = screen.getByText('model.3mf').closest('.flex'); const removeButton = fileItem?.querySelector('button'); if (removeButton) { fireEvent.click(removeButton); await waitFor(() => { expect(screen.queryByText('model.3mf')).not.toBeInTheDocument(); }); } }); }); describe('upload button behavior', () => { it('Upload button triggers upload mutation when clicked', async () => { const initialFiles = [ new File(['content'], 'test.3mf', { type: 'application/3mf' }), ]; render(); const uploadButton = screen.getByRole('button', { name: /Upload/i }); expect(uploadButton).not.toBeDisabled(); // Click should trigger upload (button text will change) fireEvent.click(uploadButton); // The button should show uploading state or become disabled await waitFor(() => { // Either showing "Uploading..." or a spinner is present const hasUploadingText = screen.queryByText(/Uploading/i) !== null; const hasSpinner = document.querySelector('.animate-spin') !== null; expect(hasUploadingText || hasSpinner).toBe(true); }); }); it('Upload button is disabled when no files are pending', async () => { render(); const uploadButton = screen.getByRole('button', { name: /Upload/i }); expect(uploadButton).toBeDisabled(); }); it('shows correct file count in Upload button', () => { const initialFiles = [ new File(['content'], 'file1.3mf', { type: 'application/3mf' }), new File(['content'], 'file2.3mf', { type: 'application/3mf' }), new File(['content'], 'file3.3mf', { type: 'application/3mf' }), ]; render(); expect(screen.getByRole('button', { name: /Upload \(3\)/i })).toBeInTheDocument(); }); }); describe('close behavior', () => { it('calls onClose when Cancel button is clicked', () => { render(); const cancelButton = screen.getByRole('button', { name: 'Cancel' }); fireEvent.click(cancelButton); expect(mockOnClose).toHaveBeenCalled(); }); it('calls onClose when X button is clicked', () => { render(); // Find the X button in the header const buttons = screen.getAllByRole('button'); const closeButton = buttons.find(btn => btn.querySelector('.lucide-x') ); if (closeButton) { fireEvent.click(closeButton); expect(mockOnClose).toHaveBeenCalled(); } }); it('calls onClose when Escape key is pressed', () => { render(); fireEvent.keyDown(window, { key: 'Escape' }); expect(mockOnClose).toHaveBeenCalled(); }); }); describe('drag and drop', () => { it('highlights drop zone on drag over', () => { render(); const dropZone = screen.getByText('Drag & drop .3mf files here').closest('div'); if (dropZone) { fireEvent.dragOver(dropZone, { dataTransfer: { files: [] }, }); // The drop zone should have the highlight class expect(dropZone.className).toContain('border-bambu-green'); } }); it('removes highlight on drag leave', () => { render(); const dropZone = screen.getByText('Drag & drop .3mf files here').closest('div'); if (dropZone) { fireEvent.dragOver(dropZone, { dataTransfer: { files: [] } }); fireEvent.dragLeave(dropZone, { dataTransfer: { files: [] } }); // The drop zone should not have the highlight class expect(dropZone.className).not.toContain('bg-bambu-green'); } }); }); describe('file size display', () => { it('shows file size in MB', () => { const file = new File(['x'.repeat(1048576)], 'large.3mf', { type: 'application/3mf' }); // 1 MB render(); expect(screen.getByText('1.0 MB')).toBeInTheDocument(); }); }); });