RouteTest.php

<?php

declare(strict_types=1);

namespace Yiisoft\Router\Tests;

use Nyholm\Psr7\Response;
use Nyholm\Psr7\ServerRequest;
use PHPUnit\Framework\TestCase;
use Psr\Container\ContainerInterface;
use Psr\EventDispatcher\EventDispatcherInterface;
use Psr\Http\Message\ResponseInterface;
use Psr\Http\Message\ServerRequestInterface;
use Psr\Http\Server\RequestHandlerInterface;
use RuntimeException;
use Yiisoft\Http\Method;
use Yiisoft\Middleware\Dispatcher\MiddlewareDispatcher;
use Yiisoft\Middleware\Dispatcher\MiddlewareFactory;
use Yiisoft\Middleware\Dispatcher\WrapperFactory;
use Yiisoft\Router\Route;
use Yiisoft\Router\Tests\Support\AssertTrait;
use Yiisoft\Router\Tests\Support\Container;
use Yiisoft\Router\Tests\Support\TestMiddleware1;
use Yiisoft\Router\Tests\Support\TestMiddleware2;
use Yiisoft\Router\Tests\Support\TestController;
use Yiisoft\Router\Tests\Support\TestMiddleware3;

final class RouteTest extends TestCase
{
    use AssertTrait;

    public function testName(): void
    {
        $route = Route::get('/')->name('test.route');

        $this->assertSame('test.route', $route->getData('name'));
    }

    public function testNameDefault(): void
    {
        $route = Route::get('/');

        $this->assertSame('GET /', $route->getData('name'));
    }

    public function testMethods(): void
    {
        $route = Route::methods([Method::POST, Method::HEAD], '/');

        $this->assertSame([Method::POST, Method::HEAD], $route->getData('methods'));
    }

    public function testGetDataWithWrongKey(): void
    {
        $route = Route::get('');

        $this->expectException(\InvalidArgumentException::class);
        $this->expectExceptionMessage('Unknown data key: wrong');

        $route->getData('wrong');
    }

    public function testGetMethod(): void
    {
        $route = Route::get('/');

        $this->assertSame([Method::GET], $route->getData('methods'));
    }

    public function testPostMethod(): void
    {
        $route = Route::post('/');

        $this->assertSame([Method::POST], $route->getData('methods'));
    }

    public function testPutMethod(): void
    {
        $route = Route::put('/');

        $this->assertSame([Method::PUT], $route->getData('methods'));
    }

    public function testDeleteMethod(): void
    {
        $route = Route::delete('/');

        $this->assertSame([Method::DELETE], $route->getData('methods'));
    }

    public function testPatchMethod(): void
    {
        $route = Route::patch('/');

        $this->assertSame([Method::PATCH], $route->getData('methods'));
    }

    public function testHeadMethod(): void
    {
        $route = Route::head('/');

        $this->assertSame([Method::HEAD], $route->getData('methods'));
    }

    public function testOptionsMethod(): void
    {
        $route = Route::options('/');

        $this->assertSame([Method::OPTIONS], $route->getData('methods'));
    }

    public function testPattern(): void
    {
        $route = Route::get('/test')->pattern('/test2');

        $this->assertSame('/test2', $route->getData('pattern'));
    }

    public function testHost(): void
    {
        $route = Route::get('/')->host('https://yiiframework.com/');

        $this->assertSame('https://yiiframework.com', $route->getData('host'));
    }

    public function testHosts(): void
    {
        $route = Route::get('/')
            ->hosts(
                'https://yiiframework.com/',
                'yf.com',
                'yii.com',
                'yf.ru'
            );

        $this->assertSame(
            [
                'https://yiiframework.com',
                'yf.com',
                'yii.com',
                'yf.ru',
            ],
            $route->getData('hosts')
        );
    }

    public function testMultipleHosts(): void
    {
        $route = Route::get('/')
            ->host('https://yiiframework.com/');
        $multipleRoute = Route::get('/')
            ->hosts(
                'https://yiiframework.com/',
                'https://yiiframework.ru/'
            );

        $this->assertCount(1, $route->getData('hosts'));
        $this->assertCount(2, $multipleRoute->getData('hosts'));
    }

    public function testDefaults(): void
    {
        $route = Route::get('/{language}')->defaults([
            'language' => 'en',
            'age' => 42,
        ]);

        $this->assertSame([
            'language' => 'en',
            'age' => '42',
        ], $route->getData('defaults'));
    }

    public function testOverride(): void
    {
        $route = Route::get('/')->override();

        $this->assertTrue($route->getData('override'));
    }

    public function testToString(): void
    {
        $route = Route::methods([Method::GET, Method::POST], '/')
            ->name('test.route')
            ->host('yiiframework.com');

        $this->assertSame('[test.route] GET,POST yiiframework.com/', (string)$route);
    }

    public function testToStringSimple(): void
    {
        $route = Route::get('/');

        $this->assertSame('GET /', (string)$route);
    }

    public function testDispatcherInjecting(): void
    {
        $request = new ServerRequest('GET', '/');
        $container = $this->getContainer(
            [
                TestController::class => new TestController(),
            ]
        );
        $dispatcher = $this->getDispatcher($container);
        $route = Route::get('/')->action([TestController::class, 'index']);
        $route->injectDispatcher($dispatcher);
        $response = $route
            ->getData('dispatcherWithMiddlewares')
            ->dispatch($request, $this->getRequestHandler());
        $this->assertSame(200, $response->getStatusCode());
    }

    public function testMiddlewareAfterAction(): void
    {
        $route = Route::get('/')->action([TestController::class, 'index']);

        $this->expectException(RuntimeException::class);
        $this->expectExceptionMessage('middleware() can not be used after action().');
        $route->middleware(static fn () => new Response());
    }

    public function testPrependMiddlewareBeforeAction(): void
    {
        $route = Route::get('/');

        $this->expectException(RuntimeException::class);
        $this->expectExceptionMessage('prependMiddleware() can not be used before action().');
        $route->prependMiddleware(static fn () => new Response());
    }

    public function testDisabledMiddlewareDefinitions(): void
    {
        $request = new ServerRequest('GET', '/');

        $injectDispatcher = $this->getDispatcher(
            $this->getContainer([
                TestMiddleware1::class => new TestMiddleware1(),
                TestMiddleware2::class => new TestMiddleware2(),
                TestMiddleware3::class => new TestMiddleware3(),
                TestController::class => new TestController(),
            ])
        );

        $route = Route::get('/')
            ->middleware(TestMiddleware1::class, TestMiddleware2::class, TestMiddleware3::class)
            ->action([TestController::class, 'index'])
            ->disableMiddleware(TestMiddleware1::class, TestMiddleware3::class);
        $route->injectDispatcher($injectDispatcher);

        $dispatcher = $route->getData('dispatcherWithMiddlewares');

        $response = $dispatcher->dispatch($request, $this->getRequestHandler());
        $this->assertSame(200, $response->getStatusCode());
        $this->assertSame('2', (string) $response->getBody());
    }

    public function testPrependMiddlewareDefinitions(): void
    {
        $request = new ServerRequest('GET', '/');

        $injectDispatcher = $this->getDispatcher(
            $this->getContainer([
                TestMiddleware1::class => new TestMiddleware1(),
                TestMiddleware2::class => new TestMiddleware2(),
                TestMiddleware3::class => new TestMiddleware3(),
                TestController::class => new TestController(),
            ])
        );

        $route = Route::get('/')
            ->middleware(TestMiddleware3::class)
            ->action([TestController::class, 'index'])
            ->prependMiddleware(TestMiddleware1::class, TestMiddleware2::class);
        $route->injectDispatcher($injectDispatcher);

        $dispatcher = $route->getData('dispatcherWithMiddlewares');

        $response = $dispatcher->dispatch($request, $this->getRequestHandler());
        $this->assertSame(200, $response->getStatusCode());
        $this->assertSame('123', (string) $response->getBody());
    }

    public function testGetDispatcherWithoutDispatcher(): void
    {
        $route = Route::get('/')->name('test');

        $this->expectException(RuntimeException::class);
        $this->expectExceptionMessage('There is no dispatcher in the route test.');
        $route->getData('dispatcherWithMiddlewares');
    }

    public function testGetDispatcherWithMiddlewares(): void
    {
        $request = new ServerRequest('GET', '/');

        $injectDispatcher = $this
            ->getDispatcher(
                $this->getContainer([
                    TestMiddleware1::class => new TestMiddleware1(),
                    TestMiddleware2::class => new TestMiddleware2(),
                    TestController::class => new TestController(),
                ])
            )
            ->withMiddlewares([
                TestMiddleware1::class,
                TestMiddleware2::class,
                [TestController::class, 'index'],
            ]);

        $route = Route::get('/');
        $route->injectDispatcher($injectDispatcher);

        $dispatcher = $route->getData('dispatcherWithMiddlewares');

        $response = $dispatcher->dispatch($request, $this->getRequestHandler());
        $this->assertSame(200, $response->getStatusCode());
        $this->assertSame('12', (string) $response->getBody());
    }

    public function testDebugInfo(): void
    {
        $route = Route::get('/')
            ->name('test')
            ->host('example.com')
            ->defaults(['age' => 42])
            ->override()
            ->middleware(TestMiddleware1::class)
            ->disableMiddleware(TestMiddleware2::class);

        $expected = <<<EOL
Yiisoft\Router\Route Object
(
    [name] => test
    [methods] => Array
        (
            [0] => GET
        )

    [pattern] => /
    [hosts] => Array
        (
            [0] => example.com
        )

    [defaults] => Array
        (
            [age] => 42
        )

    [override] => 1
    [actionAdded] =>
    [middlewareDefinitions] => Array
        (
            [0] => Yiisoft\Router\Tests\Support\TestMiddleware1
        )

    [disabledMiddlewareDefinitions] => Array
        (
            [0] => Yiisoft\Router\Tests\Support\TestMiddleware2
        )

    [middlewareDispatcher] =>
)

EOL;

        $this->assertSameStringsIgnoringLineEndingsAndSpaces($expected, print_r($route, true));
    }

    private function getRequestHandler(): RequestHandlerInterface
    {
        return new class () implements RequestHandlerInterface {
            public function handle(ServerRequestInterface $request): ResponseInterface
            {
                return new Response(404);
            }
        };
    }

    private function getDispatcher(ContainerInterface $container = null): MiddlewareDispatcher
    {
        if ($container === null) {
            return new MiddlewareDispatcher(
                new MiddlewareFactory($this->getContainer(), new WrapperFactory($this->getContainer())),
                $this->createMock(EventDispatcherInterface::class)
            );
        }

        return new MiddlewareDispatcher(
            new MiddlewareFactory($container, new WrapperFactory($container)),
            $this->createMock(EventDispatcherInterface::class)
        );
    }

    private function getContainer(array $instances = []): ContainerInterface
    {
        return new Container($instances);
    }
}