Function createMockImageAdapter

packages/ai/tests/stream-generation.test.ts:26–44

(
    overrides?: Partial<{
      generateImages: (...args: Array<any>) => Promise<any>
    }>,
  )

describe('generateImage({ stream: true })', () => {
  function createMockImageAdapter(
    overrides?: Partial<{
      generateImages: (...args: Array<any>) => Promise<any>
    }>,
  ) {
    return {
      kind: 'image' as const,
      name: 'test-image',
      model: 'test-model',
      '~types': {} as any,
      generateImages:
        overrides?.generateImages ??
        vi.fn(async () => ({
          id: 'img-1',
          model: 'test-model',
          images: [{ url: 'https://example.com/image.png' }],
        })),
    }
  }

  it('should emit RUN_STARTED, CUSTOM result, and RUN_FINISHED', async () => {
    const adapter = createMockImageAdapter()