docling-studio/frontend/src/features/analysis/store.test.js

import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
import { setActivePinia, createPinia } from 'pinia'
import { useAnalysisStore } from './store.js'

vi.mock('./api.js', () => ({
  fetchAnalyses: vi.fn(),
  fetchAnalysis: vi.fn(),
  createAnalysis: vi.fn(),
  deleteAnalysis: vi.fn(),
}))

import * as api from './api.js'

describe('useAnalysisStore', () => {
  beforeEach(() => {
    setActivePinia(createPinia())
    vi.clearAllMocks()
    vi.useFakeTimers()
  })

  afterEach(() => {
    vi.useRealTimers()
  })

  it('starts with empty state', () => {
    const store = useAnalysisStore()
    expect(store.analyses).toEqual([])
    expect(store.currentAnalysis).toBeNull()
    expect(store.running).toBe(false)
  })

  it('currentPages parses pagesJson from current analysis', () => {
    const store = useAnalysisStore()
    store.currentAnalysis = {
      pagesJson: JSON.stringify([{ pageNumber: 1, width: 612, height: 792 }]),
    }
    expect(store.currentPages).toEqual([{ pageNumber: 1, width: 612, height: 792 }])
  })

  it('currentPages returns [] when no current analysis', () => {
    const store = useAnalysisStore()
    expect(store.currentPages).toEqual([])
  })

  it('currentPages returns [] on invalid JSON', () => {
    const store = useAnalysisStore()
    store.currentAnalysis = { pagesJson: 'not json' }
    expect(store.currentPages).toEqual([])
  })

  it('load() fetches analyses', async () => {
    const data = [{ id: '1', status: 'COMPLETED' }]
    api.fetchAnalyses.mockResolvedValue(data)

    const store = useAnalysisStore()
    await store.load()

    expect(store.analyses).toEqual(data)
  })

  it('run() creates analysis, sets current, and starts polling', async () => {
    const job = { id: 'j1', status: 'PENDING', documentId: 'd1' }
    api.createAnalysis.mockResolvedValue(job)
    api.fetchAnalysis.mockResolvedValue({ ...job, status: 'COMPLETED' })

    const store = useAnalysisStore()
    const result = await store.run('d1')

    expect(result).toEqual(job)
    expect(store.currentAnalysis).toEqual(job)
    expect(store.analyses[0]).toEqual(job)
    expect(store.running).toBe(true)

    // Advance timer to trigger polling
    await vi.advanceTimersByTimeAsync(2000)

    expect(api.fetchAnalysis).toHaveBeenCalledWith('j1')
    expect(store.running).toBe(false) // COMPLETED stops polling

    store.stopPolling()
  })

  it('run() resets running on error', async () => {
    api.createAnalysis.mockRejectedValue(new Error('fail'))
    vi.spyOn(console, 'error').mockImplementation(() => {})

    const store = useAnalysisStore()
    await expect(store.run('d1')).rejects.toThrow('fail')

    expect(store.running).toBe(false)
  })

  it('select() fetches and sets current analysis', async () => {
    const job = { id: '42', status: 'COMPLETED' }
    api.fetchAnalysis.mockResolvedValue(job)

    const store = useAnalysisStore()
    await store.select('42')

    expect(store.currentAnalysis).toEqual(job)
  })

  it('remove() deletes and removes from list', async () => {
    api.deleteAnalysis.mockResolvedValue(null)

    const store = useAnalysisStore()
    store.analyses = [{ id: '1' }, { id: '2' }]
    store.currentAnalysis = { id: '1' }

    await store.remove('1')

    expect(store.analyses).toEqual([{ id: '2' }])
    expect(store.currentAnalysis).toBeNull()
  })

  it('remove() keeps currentAnalysis if different id', async () => {
    api.deleteAnalysis.mockResolvedValue(null)

    const store = useAnalysisStore()
    store.analyses = [{ id: '1' }, { id: '2' }]
    store.currentAnalysis = { id: '2' }

    await store.remove('1')

    expect(store.currentAnalysis).toEqual({ id: '2' })
  })

  it('polling stops on FAILED status', async () => {
    const job = { id: 'j1', status: 'PENDING', documentId: 'd1' }
    api.createAnalysis.mockResolvedValue(job)
    api.fetchAnalysis.mockResolvedValue({ ...job, status: 'FAILED', errorMessage: 'oops' })

    const store = useAnalysisStore()
    await store.run('d1')

    await vi.advanceTimersByTimeAsync(2000)

    expect(store.running).toBe(false)
    expect(store.currentAnalysis.status).toBe('FAILED')
  })

  it('polling stops on fetch error', async () => {
    const job = { id: 'j1', status: 'PENDING', documentId: 'd1' }
    api.createAnalysis.mockResolvedValue(job)
    api.fetchAnalysis.mockRejectedValue(new Error('network'))
    vi.spyOn(console, 'error').mockImplementation(() => {})

    const store = useAnalysisStore()
    await store.run('d1')

    await vi.advanceTimersByTimeAsync(2000)

    expect(store.running).toBe(false)
  })
})