Function createMockVideoAdapter

packages/ai/tests/stream-generation.test.ts:159–210

(options?: {
    pollsBeforeComplete?: number
    failOnPoll?: number
  })

describe('generateVideo({ stream: true })', () => {
  function createMockVideoAdapter(options?: {
    pollsBeforeComplete?: number
    failOnPoll?: number
  }) {
    const pollsBeforeComplete = options?.pollsBeforeComplete ?? 2
    const failOnPoll = options?.failOnPoll
    let pollCount = 0

    return {
      kind: 'video' as const,
      name: 'test-video',
      model: 'test-model',
      '~types': {} as any,

      availableDurations: () => ({ kind: 'none' as const }),
      snapDuration: () => undefined,

      createVideoJob: vi.fn(async () => ({
        jobId: 'job-123',
        model: 'test-model',
      })),

      getVideoStatus: vi.fn(async (): Promise<VideoStatusResult> => {
        pollCount++
        if (failOnPoll && pollCount >= failOnPoll) {
          return {
            jobId: 'job-123',
            status: 'failed',
            error: 'Video processing error',
          }
        }
        if (pollCount >= pollsBeforeComplete) {
          return {
            jobId: 'job-123',
            status: 'completed',
            progress: 100,
          }
        }
        return {
          jobId: 'job-123',
          status: 'processing',
          progress: Math.round((pollCount / pollsBeforeComplete) * 100),
        }
      }),

      getVideoUrl: vi.fn(async () => ({
        jobId: 'job-123',
        url: 'https://example.com/video.mp4',
        expiresAt: new Date('2030-01-01'),
      })),
    }
  }

  it('should emit job lifecycle events until completion', async () => {
    const adapter = createMockVideoAdapter({ pollsBeforeComplete: 2 })