Aether/frontend/src/api/__tests__/dashboard-body-loading.spec.ts

import { beforeEach, describe, expect, it, vi } from 'vitest'

const { getMock } = vi.hoisted(() => ({ getMock: vi.fn() }))
vi.mock('@/api/client', () => ({ default: { get: getMock } }))

import { dashboardApi } from '@/api/dashboard'
import { cache } from '@/utils/cache'

beforeEach(() => {
  cache.clear()
  getMock.mockReset()
  getMock.mockResolvedValue({ data: { id: 'usage-1' } })
})

describe('dashboard body loading', () => {
  it('requests opaque body bytes, outside the JSON detail cache', async () => {
    const bytes = new ArrayBuffer(20)
    const controller = new AbortController()
    getMock.mockResolvedValue({ data: bytes, headers: { 'x-aether-body-encoding': 'gzip', 'x-aether-usage-id': 'usage-1', 'x-aether-body-field': 'request_body' } })
    await expect(dashboardApi.getRequestBody('usage-1', 'request_body', controller.signal)).resolves.toEqual({ bytes, encoding: 'gzip' })
    expect(getMock).toHaveBeenCalledWith('/api/admin/usage/usage-1', { params: { include_bodies: true, body_field: 'request_body', body_format: 'raw' }, responseType: 'arraybuffer', signal: controller.signal })
  })

  it.each([
    { 'x-aether-body-encoding': 'br', 'x-aether-usage-id': 'usage-1', 'x-aether-body-field': 'request_body' },
    { 'x-aether-body-encoding': 'gzip', 'x-aether-usage-id': 'usage-other', 'x-aether-body-field': 'request_body' },
    { 'x-aether-body-encoding': 'gzip', 'x-aether-usage-id': 'usage-1', 'x-aether-body-field': 'response_body' },
    {},
  ])('rejects mismatched or invalid body response headers', async headers => {
    getMock.mockResolvedValue({ data: new ArrayBuffer(10), headers })
    await expect(dashboardApi.getRequestBody('usage-1', 'request_body')).rejects.toThrow('Invalid body response')
  })

  it('defaults to shallow details and isolates cached records', async () => {
    await dashboardApi.getRequestDetail('usage-1')
    expect(getMock).toHaveBeenLastCalledWith('/api/admin/usage/usage-1', { params: { include_bodies: false } })
    await dashboardApi.getRequestDetail('usage-1', { cacheTtlMs: 5000 })
    await dashboardApi.getRequestDetail('usage-2', { cacheTtlMs: 5000 })
    await dashboardApi.getRequestDetail('usage-1', { cacheTtlMs: 5000 })
    expect(getMock).toHaveBeenCalledTimes(3)
    expect(getMock).toHaveBeenLastCalledWith('/api/admin/usage/usage-2', { params: { include_bodies: false } })
  })

  it('does not share abortable requests with another caller or a cancelled request', async () => {
    const first = new AbortController()
    const second = new AbortController()
    const firstRequest = dashboardApi.getRequestDetail('usage-1', { signal: first.signal })
    first.abort()
    const secondRequest = dashboardApi.getRequestDetail('usage-1', { signal: second.signal })
    await Promise.all([firstRequest, secondRequest])
    expect(getMock).toHaveBeenCalledTimes(2)
    expect(getMock).toHaveBeenLastCalledWith('/api/admin/usage/usage-1', {
      params: { include_bodies: false }, signal: second.signal,
    })
  })
})