<?php

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

test('it generates text successfully', function () {
    Http::fake([
        'api.openai.com/*' => Http::response([
            'choices' => [
                ['message' => ['content' => 'Once upon a time...']],
            ],
        ], 200),
    ]);

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

    $result = (new OpenAiDriver)->generate($provider, 'Write a story opener');

    expect($result->content)->toBe('Once upon a time...');
    expect($result->contentType)->toBe('text/plain');

    Http::assertSent(function ($request) {
        return $request->url() === 'https://api.openai.com/v1/chat/completions'
            && $request->hasHeader('Authorization', 'Bearer sk-test')
            && $request['messages'][0]['content'] === 'Write a story opener'
            && $request['model'] === 'gpt-4o-mini';
    });
});

test('it uses a model from options over the provider config over the default', function () {
    Http::fake([
        'api.openai.com/*' => Http::response(['choices' => [['message' => ['content' => 'hi']]]]),
    ]);

    $provider = AiProvider::factory()->create(['provider' => 'openai', 'config' => ['model' => 'gpt-4o']]);

    (new OpenAiDriver)->generate($provider, 'hi', ['model' => 'gpt-4-turbo']);

    Http::assertSent(fn ($request) => $request['model'] === 'gpt-4-turbo');
});

test('it throws when OpenAI returns an error', function () {
    Http::fake([
        'api.openai.com/*' => Http::response(['error' => ['message' => 'Invalid API key']], 401),
    ]);

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

    (new OpenAiDriver)->generate($provider, 'hi');
})->throws(AiGenerationException::class, 'Invalid API key');

test('it throws when the response has no message content', function () {
    Http::fake([
        'api.openai.com/*' => Http::response(['choices' => []]),
    ]);

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

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