Created DropDefaultDomainFromQueryMiddlewareTest

module/Rest/test/Middleware/ShortUrl/DropDefaultDomainFromQueryMiddlewareTest.php

@@ -0,0 +1,54 @@
+<?php
+
+declare(strict_types=1);
+
+namespace ShlinkioTest\Shlink\Rest\Middleware\ShortUrl;
+
+use Laminas\Diactoros\Response;
+use Laminas\Diactoros\ServerRequestFactory;
+use PHPUnit\Framework\Assert;
+use PHPUnit\Framework\TestCase;
+use Prophecy\Argument;
+use Prophecy\Prophecy\ObjectProphecy;
+use Psr\Http\Message\ServerRequestInterface;
+use Psr\Http\Server\RequestHandlerInterface;
+use Shlinkio\Shlink\Rest\Middleware\ShortUrl\DropDefaultDomainFromQueryMiddleware;
+
+class DropDefaultDomainFromQueryMiddlewareTest extends TestCase
+{
+    private DropDefaultDomainFromQueryMiddleware $middleware;
+    private ObjectProphecy $next;
+
+    public function setUp(): void
+    {
+        $this->next = $this->prophesize(RequestHandlerInterface::class);
+        $this->middleware = new DropDefaultDomainFromQueryMiddleware('doma.in');
+    }
+
+    /**
+     * @test
+     * @dataProvider provideQueryParams
+     */
+    public function domainIsDroppedWhenDefaultOneIsProvided(array $providedQuery, array $expectedQuery): void
+    {
+        $req = ServerRequestFactory::fromGlobals()->withQueryParams($providedQuery);
+
+        $handle = $this->next->handle(Argument::that(function (ServerRequestInterface $request) use ($expectedQuery) {
+            Assert::assertEquals($expectedQuery, $request->getQueryParams());
+            return $request;
+        }))->willReturn(new Response());
+
+        $this->middleware->process($req, $this->next->reveal());
+
+        $handle->shouldHaveBeenCalledOnce();
+    }
+
+    public function provideQueryParams(): iterable
+    {
+        yield [[], []];
+        yield [['foo' => 'bar'], ['foo' => 'bar']];
+        yield [['foo' => 'bar', 'domain' => 'doma.in'], ['foo' => 'bar']];
+        yield [['foo' => 'bar', 'domain' => 'not_default'], ['foo' => 'bar', 'domain' => 'not_default']];
+        yield [['domain' => 'doma.in'], []];
+    }
+}