shlink/module/Rest/test/Middleware/ErrorHandler/BackwardsCompatibleProblemDetailsHandlerTest.php

<?php

declare(strict_types=1);

namespace ShlinkioTest\Shlink\Rest\Middleware\ErrorHandler;

use Laminas\Diactoros\ServerRequestFactory;
use PHPUnit\Framework\Attributes\DataProvider;
use PHPUnit\Framework\Attributes\Test;
use PHPUnit\Framework\TestCase;
use Psr\Http\Message\ServerRequestInterface;
use Psr\Http\Server\RequestHandlerInterface;
use Shlinkio\Shlink\Core\Exception\ValidationException;
use Shlinkio\Shlink\Rest\Exception\BackwardsCompatibleProblemDetailsException;
use Shlinkio\Shlink\Rest\Middleware\ErrorHandler\BackwardsCompatibleProblemDetailsHandler;
use Throwable;

class BackwardsCompatibleProblemDetailsHandlerTest extends TestCase
{
    private BackwardsCompatibleProblemDetailsHandler $handler;

    protected function setUp(): void
    {
        $this->handler = new BackwardsCompatibleProblemDetailsHandler();
    }

    /**
     * @param class-string<Throwable> $expectedException
     */
    #[Test, DataProvider('provideExceptions')]
    public function expectedExceptionIsThrownBasedOnTheRequestVersion(
        ServerRequestInterface $request,
        Throwable $thrownException,
        string $expectedException,
    ): void {
        $handler = $this->createMock(RequestHandlerInterface::class);
        $handler->expects($this->once())->method('handle')->with($request)->willThrowException($thrownException);

        $this->expectException($expectedException);

        $this->handler->process($request, $handler);
    }

    public static function provideExceptions(): iterable
    {
        $baseRequest = ServerRequestFactory::fromGlobals();

        yield 'no version' => [
            $baseRequest,
            ValidationException::fromArray([]),
            BackwardsCompatibleProblemDetailsException::class,
        ];
        yield 'version 1' => [
            $baseRequest->withAttribute('version', '1'),
            ValidationException::fromArray([]),
            BackwardsCompatibleProblemDetailsException::class,
        ];
        yield 'version 2' => [
            $baseRequest->withAttribute('version', '2'),
            ValidationException::fromArray([]),
            BackwardsCompatibleProblemDetailsException::class,
        ];
        yield 'version 3' => [
            $baseRequest->withAttribute('version', '3'),
            ValidationException::fromArray([]),
            ValidationException::class,
        ];
        yield 'version 4' => [
            $baseRequest->withAttribute('version', '3'),
            ValidationException::fromArray([]),
            ValidationException::class,
        ];
    }
}