<?php

use App\Models\AiProvider;
use App\Support\AiProviders\AiGenerationException;
use App\Support\AiProviders\StabilityAiDriver;
use Illuminate\Support\Facades\Http;

test('it generates an image successfully', function () {
    $bytes = 'fake-png-bytes';

    Http::fake([
        'api.stability.ai/*' => Http::response([
            'image' => base64_encode($bytes),
            'seed' => 42,
            'finish_reason' => 'SUCCESS',
        ], 200),
    ]);

    $provider = AiProvider::factory()->create(['provider' => 'stability_ai', 'api_key' => 'sk-test']);

    $result = (new StabilityAiDriver)->generate($provider, 'a red bicycle');

    expect($result->content)->toBe($bytes);
    expect($result->contentType)->toBe('image/png');
    expect($result->raw['seed'])->toBe(42);

    Http::assertSent(function ($request) {
        return $request->url() === 'https://api.stability.ai/v2beta/stable-image/generate/core'
            && $request->hasHeader('Authorization', 'Bearer sk-test');
    });
});

test('it respects an output_format override', function () {
    Http::fake([
        'api.stability.ai/*' => Http::response(['image' => base64_encode('bytes')]),
    ]);

    $provider = AiProvider::factory()->create(['provider' => 'stability_ai']);

    $result = (new StabilityAiDriver)->generate($provider, 'prompt', ['output_format' => 'webp']);

    expect($result->contentType)->toBe('image/webp');
});

test('it throws when StabilityAI returns an error', function () {
    Http::fake([
        'api.stability.ai/*' => Http::response(['errors' => ['invalid prompt']], 400),
    ]);

    $provider = AiProvider::factory()->create(['provider' => 'stability_ai']);

    (new StabilityAiDriver)->generate($provider, 'prompt');
})->throws(AiGenerationException::class, 'invalid prompt');

test('it throws when the response has no image data', function () {
    Http::fake([
        'api.stability.ai/*' => Http::response(['finish_reason' => 'CONTENT_FILTERED']),
    ]);

    $provider = AiProvider::factory()->create(['provider' => 'stability_ai']);

    (new StabilityAiDriver)->generate($provider, 'prompt');
})->throws(AiGenerationException::class);