docling-studio/frontend/src/features/analysis/api.test.ts

85 lines
2.5 KiB
TypeScript

import { describe, it, expect, vi, beforeEach } from 'vitest'
import {
createAnalysis,
fetchAnalyses,
fetchAnalysis,
deleteAnalysis,
fetchDocumentAnalyses,
} from './api'
vi.mock('../../shared/api/http', () => ({
apiFetch: vi.fn(),
}))
import { apiFetch } from '../../shared/api/http'
describe('analysis API', () => {
beforeEach(() => {
vi.clearAllMocks()
})
it('createAnalysis sends POST with documentId only', async () => {
const job = { id: '1', documentId: 'doc-1', status: 'PENDING' }
apiFetch.mockResolvedValue(job)
const result = await createAnalysis('doc-1')
expect(apiFetch).toHaveBeenCalledWith('/api/analyses', {
method: 'POST',
body: JSON.stringify({ documentId: 'doc-1' }),
})
expect(result).toEqual(job)
})
it('createAnalysis sends POST with pipeline options', async () => {
const job = { id: '2', documentId: 'doc-1', status: 'PENDING' }
apiFetch.mockResolvedValue(job)
const options = { do_ocr: false, table_mode: 'fast', do_code_enrichment: true }
const result = await createAnalysis('doc-1', options)
expect(apiFetch).toHaveBeenCalledWith('/api/analyses', {
method: 'POST',
body: JSON.stringify({ documentId: 'doc-1', pipelineOptions: options }),
})
expect(result).toEqual(job)
})
it('fetchAnalyses calls GET /api/analyses', async () => {
const jobs = [{ id: '1', status: 'COMPLETED' }]
apiFetch.mockResolvedValue(jobs)
const result = await fetchAnalyses()
expect(apiFetch).toHaveBeenCalledWith('/api/analyses')
expect(result).toEqual(jobs)
})
it('fetchAnalysis calls GET /api/analyses/:id', async () => {
const job = { id: '42', status: 'RUNNING' }
apiFetch.mockResolvedValue(job)
const result = await fetchAnalysis('42')
expect(apiFetch).toHaveBeenCalledWith('/api/analyses/42')
expect(result).toEqual(job)
})
it('deleteAnalysis calls DELETE /api/analyses/:id', async () => {
apiFetch.mockResolvedValue(null)
await deleteAnalysis('42')
expect(apiFetch).toHaveBeenCalledWith('/api/analyses/42', { method: 'DELETE' })
})
it('fetchDocumentAnalyses calls GET /api/analyses?documentId=:id', async () => {
const analyses = [{ id: '1', documentId: 'doc-42', status: 'COMPLETED' }]
apiFetch.mockResolvedValue(analyses)
const result = await fetchDocumentAnalyses('doc-42')
expect(apiFetch).toHaveBeenCalledWith('/api/analyses?documentId=doc-42')
expect(result).toEqual(analyses)
})
})