Spaces:
Sleeping
Sleeping
File size: 3,579 Bytes
cb75c6a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 | import { render, screen, fireEvent, waitFor } from '@testing-library/react';
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
import App from './App';
describe('App Integration', () => {
// Mock fetch
const mockFetch = vi.fn();
global.fetch = mockFetch;
beforeEach(() => {
mockFetch.mockClear();
// Default mocks
mockFetch.mockResolvedValue({
json: () => Promise.resolve({ success: true, messages: [], sessions: [] }),
ok: true
});
});
afterEach(() => {
vi.restoreAllMocks();
});
it('renders app structure', async () => {
render(<App />);
expect(screen.getByText('Medical AI Assistant')).toBeInTheDocument();
expect(screen.getByText('MediGenius')).toBeInTheDocument();
expect(screen.getByPlaceholderText('Ask your medical question...')).toBeInTheDocument();
});
it('loads sessions on mount', async () => {
const mockSessions = [{ session_id: '1', preview: 'Flu symptoms', last_active: '2023' }];
mockFetch.mockImplementation((url) => {
if (url === '/api/v1/sessions') {
return Promise.resolve({
json: () => Promise.resolve({ success: true, sessions: mockSessions }),
ok: true
});
}
return Promise.resolve({ json: () => Promise.resolve({ success: true }), ok: true });
});
render(<App />);
await waitFor(() => {
expect(screen.getByText('Flu symptoms')).toBeInTheDocument();
});
});
it('sends a message and displays response', async () => {
// Setup mocks
mockFetch.mockImplementation((url, options) => {
if (url === '/api/v1/chat') {
return Promise.resolve({
json: () => Promise.resolve({
success: true,
response: 'I can help with that.',
source: 'test-source',
timestamp: 'now'
}),
ok: true
});
}
return Promise.resolve({ json: () => Promise.resolve({ success: true, sessions: [] }), ok: true });
});
render(<App />);
const input = screen.getByPlaceholderText('Ask your medical question...');
fireEvent.change(input, { target: { value: 'Headache' } });
const sendBtn = screen.getByRole('button', { name: /Send message/i });
fireEvent.click(sendBtn);
expect(screen.getByText('Headache')).toBeInTheDocument(); // Optimistic UI
await waitFor(() => {
expect(screen.getByText('I can help with that.')).toBeInTheDocument();
});
});
it('creates new chat', async () => {
mockFetch.mockImplementation((url) => {
if (url === '/api/v1/new-chat') {
return Promise.resolve({
json: () => Promise.resolve({ success: true, session_id: 'new-session' }),
ok: true
});
}
return Promise.resolve({ json: () => Promise.resolve({ success: true, sessions: [] }), ok: true });
});
render(<App />);
const newChatBtn = screen.getByText('New Chat');
fireEvent.click(newChatBtn);
await waitFor(() => {
// Check if new chat API was called
expect(mockFetch).toHaveBeenCalledWith('/api/v1/new-chat', expect.any(Object));
});
});
});
|