86 lines
2.5 KiB
TypeScript
86 lines
2.5 KiB
TypeScript
import { beforeEach, describe, expect, it, vi } from 'vitest'
|
|
|
|
const getProviderConfigMock = vi.hoisted(() => vi.fn(async () => ({
|
|
id: 'openai-compatible:oa-1',
|
|
apiKey: 'oa-key',
|
|
baseUrl: 'https://oa.test/v1',
|
|
})))
|
|
|
|
vi.mock('@/lib/api-config', () => ({
|
|
getProviderConfig: getProviderConfigMock,
|
|
}))
|
|
|
|
import { pollAsyncTask } from '@/lib/async-poll'
|
|
|
|
const PROVIDER_TOKEN = Buffer.from('openai-compatible:oa-1', 'utf8').toString('base64url')
|
|
|
|
/**
|
|
* pollOpenAIVideoTask now uses raw fetch (not OpenAI SDK),
|
|
* so we mock fetch instead of the SDK.
|
|
*/
|
|
describe('async poll OPENAI video status mapping', () => {
|
|
let fetchSpy: ReturnType<typeof vi.fn>
|
|
|
|
beforeEach(() => {
|
|
vi.clearAllMocks()
|
|
getProviderConfigMock.mockResolvedValue({
|
|
id: 'openai-compatible:oa-1',
|
|
apiKey: 'oa-key',
|
|
baseUrl: 'https://oa.test/v1',
|
|
})
|
|
fetchSpy = vi.fn()
|
|
globalThis.fetch = fetchSpy as unknown as typeof fetch
|
|
})
|
|
|
|
it('maps queued/in_progress to pending', async () => {
|
|
fetchSpy
|
|
.mockResolvedValueOnce({
|
|
ok: true,
|
|
json: async () => ({ status: 'queued' }),
|
|
})
|
|
.mockResolvedValueOnce({
|
|
ok: true,
|
|
json: async () => ({ status: 'in_progress' }),
|
|
})
|
|
|
|
const queued = await pollAsyncTask(`OPENAI:VIDEO:${PROVIDER_TOKEN}:vid_queued`, 'user-1')
|
|
const progress = await pollAsyncTask(`OPENAI:VIDEO:${PROVIDER_TOKEN}:vid_running`, 'user-1')
|
|
|
|
expect(queued).toEqual({ status: 'pending' })
|
|
expect(progress).toEqual({ status: 'pending' })
|
|
})
|
|
|
|
it('maps completed to downloadable url and auth headers', async () => {
|
|
fetchSpy.mockResolvedValueOnce({
|
|
ok: true,
|
|
json: async () => ({
|
|
id: 'vid_done',
|
|
status: 'completed',
|
|
}),
|
|
})
|
|
|
|
const result = await pollAsyncTask(`OPENAI:VIDEO:${PROVIDER_TOKEN}:vid_done`, 'user-1')
|
|
|
|
expect(result.status).toBe('completed')
|
|
expect(result.resultUrl).toBe('https://oa.test/v1/videos/vid_done/content')
|
|
expect(result.videoUrl).toBe('https://oa.test/v1/videos/vid_done/content')
|
|
expect(result.downloadHeaders).toEqual({
|
|
Authorization: 'Bearer oa-key',
|
|
})
|
|
})
|
|
|
|
it('maps failed to failed with provider error message', async () => {
|
|
fetchSpy.mockResolvedValueOnce({
|
|
ok: true,
|
|
json: async () => ({
|
|
id: 'vid_failed',
|
|
status: 'failed',
|
|
error: { message: 'generation failed' },
|
|
}),
|
|
})
|
|
|
|
const result = await pollAsyncTask(`OPENAI:VIDEO:${PROVIDER_TOKEN}:vid_failed`, 'user-1')
|
|
expect(result).toEqual({ status: 'failed', error: 'generation failed' })
|
|
})
|
|
})
|