JsonFicatorPHP/tests/Integration/OpenAIIntegrationTest.php

<?php

declare(strict_types=1);

namespace JsonFicator\Tests\Integration;

use JsonFicator\JsonFicator;
use JsonFicator\Provider\OpenAIProvider;
use PHPUnit\Framework\TestCase;

/**
 * @group integration
 *
 * These tests require a valid OpenAI API key in the JSONFICATOR_OPENAI_API_KEY env variable.
 * Run them with: composer test:integration
 */
final class OpenAIIntegrationTest extends TestCase
{
    private static ?string $apiKey = null;

    public static function setUpBeforeClass(): void
    {
        self::$apiKey = $_SERVER['JSONFICATOR_OPENAI_API_KEY'] ?? getenv('JSONFICATOR_OPENAI_API_KEY') ?: null;
    }

    protected function setUp(): void
    {
        if (self::$apiKey === null || self::$apiKey === '') {
            $this->markTestSkipped('JSONFICATOR_OPENAI_API_KEY environment variable is not set.');
        }
    }

    public function testToJsonExtractsData(): void
    {
        $provider = new OpenAIProvider([
            'key' => self::$apiKey,
            'model' => 'gpt-4o-mini',
        ]);

        $jsonFicator = new JsonFicator($provider);

        $result = $jsonFicator->toJson('Колодки на ниссан лиф +799999999', [
            'carBrand' => 'string?',
            'carModel' => 'string?',
            'phoneNumber' => 'string?',
        ]);

        $data = $result->toArray();

        $this->assertArrayHasKey('carBrand', $data);
        $this->assertArrayHasKey('carModel', $data);
        $this->assertArrayHasKey('phoneNumber', $data);

        // Brand should be detected (Nissan).
        $this->assertNotNull($data['carBrand']);
        $this->assertIsString($data['carBrand']);
    }

    public function testToJsonDoesNotInventValues(): void
    {
        $provider = new OpenAIProvider([
            'key' => self::$apiKey,
            'model' => 'gpt-4o-mini',
        ]);

        $jsonFicator = new JsonFicator($provider);

        $result = $jsonFicator->toJson('Колодки на Nissan Leaf', [
            'brand' => 'string',
            'model' => 'string',
            'year' => 'int?',
            'phone' => 'string?',
        ]);

        $data = $result->toArray();

        $this->assertSame('Nissan', $data['brand']);
        $this->assertSame('Leaf', $data['model']);

        // Year and phone are not present in the text and are nullable.
        $this->assertNull($data['year']);
        $this->assertNull($data['phone']);
    }
}