shlink/module/Rest/test/Middleware/PathVersionMiddlewareTest.php

<?php
declare(strict_types=1);

namespace ShlinkioTest\Shlink\Rest\Middleware;

use PHPUnit\Framework\Assert;
use PHPUnit\Framework\TestCase;
use Prophecy\Argument;
use Psr\Http\Message\ServerRequestInterface as Request;
use Psr\Http\Server\RequestHandlerInterface;
use Shlinkio\Shlink\Rest\Middleware\PathVersionMiddleware;
use Zend\Diactoros\Response;
use Zend\Diactoros\ServerRequest;
use Zend\Diactoros\Uri;
use function array_shift;

class PathVersionMiddlewareTest extends TestCase
{
    /** @var PathVersionMiddleware */
    private $middleware;

    public function setUp()
    {
        $this->middleware = new PathVersionMiddleware();
    }

    /**
     * @test
     */
    public function whenVersionIsProvidedRequestRemainsUnchanged()
    {
        $request = (new ServerRequest())->withUri(new Uri('/v2/foo'));

        $delegate = $this->prophesize(RequestHandlerInterface::class);
        $process = $delegate->handle($request)->willReturn(new Response());

        $this->middleware->process($request, $delegate->reveal());

        $process->shouldHaveBeenCalled();
    }

    /**
     * @test
     */
    public function versionOneIsPrependedWhenNoVersionIsDefined()
    {
        $request = (new ServerRequest())->withUri(new Uri('/bar/baz'));

        $delegate = $this->prophesize(RequestHandlerInterface::class);
        $delegate->handle(Argument::type(Request::class))->will(function (array $args) use ($request) {
            $req = array_shift($args);

            Assert::assertNotSame($request, $req);
            Assert::assertEquals('/v1/bar/baz', $req->getUri()->getPath());
            return new Response();
        });


        $this->middleware->process($request, $delegate->reveal());
    }
}