Testing AI-Powered Features: A Practical Guide

describe('validatePrompt', () => {
  it('accepts valid prompts', () => {
    expect(validatePrompt('A sunset').valid).toBe(true);
  });

  it('rejects empty prompts', () => {
    const result = validatePrompt('');
    expect(result.valid).toBe(false);
    expect(result.errors).toContain('Prompt cannot be empty');
  });
});

const mockGenerateImage = vi.fn();
vi.mock('./abstrakt-client', () => ({ run: mockGenerateImage }));

describe('ImageService', () => {
  it('generates an image', async () => {
    mockGenerateImage.mockResolvedValue({
      result: { images: [{ url: 'https://example.com/image.png' }] }
    });
    
    const result = await service.generate('A sunset');
    expect(result.url).toBe('https://example.com/image.png');
  });

  it('retries on rate limit', async () => {
    mockGenerateImage
      .mockRejectedValueOnce({ code: 'RATE_LIMITED' })
      .mockResolvedValueOnce({ result: { images: [{ url: '...' }] } });
    
    await service.generate('A sunset');
    expect(mockGenerateImage).toHaveBeenCalledTimes(2);
  });
});

describe('Webhook Handler', () => {
  it('accepts valid signatures', async () => {
    const payload = { event: 'job.completed' };
    const signature = signPayload(payload);
    
    const response = await request(app)
      .post('/api/webhooks/abstrakt')
      .set('x-abstrakt-signature', signature)
      .send(payload);
    
    expect(response.status).toBe(200);
  });
});

test('generates an image', async ({ page }) => {
  await page.goto('/generate');
  await page.fill('[data-testid="prompt"]', 'A sunset');
  await page.click('[data-testid="generate"]');
  
  await expect(page.locator('[data-testid="image"]')).toBeVisible({
    timeout: 60000
  });
});

jobs:
  test:
    steps:
      - run: npm run test:unit
      - run: npm run test:integration
        env:
          ABSTRAKT_API_KEY: ${{ secrets.TEST_KEY }}