Migrated QrCodeCacheMiddleware to psr-15 middleware

This commit is contained in:
Alejandro Celaya 2017-03-24 23:34:17 +01:00
parent 734dac9456
commit 6c87436a96
2 changed files with 25 additions and 41 deletions

View File

@ -3,9 +3,11 @@ namespace Shlinkio\Shlink\Core\Middleware;
use Acelaya\ZsmAnnotatedServices\Annotation\Inject; use Acelaya\ZsmAnnotatedServices\Annotation\Inject;
use Doctrine\Common\Cache\Cache; use Doctrine\Common\Cache\Cache;
use Interop\Http\ServerMiddleware\DelegateInterface;
use Interop\Http\ServerMiddleware\MiddlewareInterface;
use Psr\Http\Message\ResponseInterface as Response; use Psr\Http\Message\ResponseInterface as Response;
use Psr\Http\Message\ServerRequestInterface as Request; use Psr\Http\Message\ServerRequestInterface as Request;
use Zend\Stratigility\MiddlewareInterface; use Zend\Diactoros\Response as DiactResp;
class QrCodeCacheMiddleware implements MiddlewareInterface class QrCodeCacheMiddleware implements MiddlewareInterface
{ {
@ -26,44 +28,29 @@ class QrCodeCacheMiddleware implements MiddlewareInterface
} }
/** /**
* Process an incoming request and/or response. * Process an incoming server request and return a response, optionally delegating
* * to the next middleware component to create the response.
* Accepts a server-side request and a response instance, and does
* something with them.
*
* If the response is not complete and/or further processing would not
* interfere with the work done in the middleware, or if the middleware
* wants to delegate to another process, it can use the `$out` callable
* if present.
*
* If the middleware does not return a value, execution of the current
* request is considered complete, and the response instance provided will
* be considered the response to return.
*
* Alternately, the middleware may return a response instance.
*
* Often, middleware will `return $out();`, with the assumption that a
* later middleware will return a response.
* *
* @param Request $request * @param Request $request
* @param Response $response * @param DelegateInterface $delegate
* @param null|callable $out *
* @return null|Response * @return Response
*/ */
public function __invoke(Request $request, Response $response, callable $out = null) public function process(Request $request, DelegateInterface $delegate)
{ {
$cacheKey = $request->getUri()->getPath(); $cacheKey = $request->getUri()->getPath();
// If this QR code is already cached, just return it // If this QR code is already cached, just return it
if ($this->cache->contains($cacheKey)) { if ($this->cache->contains($cacheKey)) {
$qrData = $this->cache->fetch($cacheKey); $qrData = $this->cache->fetch($cacheKey);
$response = new DiactResp();
$response->getBody()->write($qrData['body']); $response->getBody()->write($qrData['body']);
return $response->withHeader('Content-Type', $qrData['content-type']); return $response->withHeader('Content-Type', $qrData['content-type']);
} }
// If not, call the next middleware and cache it // If not, call the next middleware and cache it
/** @var Response $resp */ /** @var Response $resp */
$resp = $out($request, $response); $resp = $delegate->process($request);
$this->cache->save($cacheKey, [ $this->cache->save($cacheKey, [
'body' => $resp->getBody()->__toString(), 'body' => $resp->getBody()->__toString(),
'content-type' => $resp->getHeaderLine('Content-Type'), 'content-type' => $resp->getHeaderLine('Content-Type'),

View File

@ -4,7 +4,9 @@ namespace ShlinkioTest\Shlink\Core\Middleware;
use Doctrine\Common\Cache\ArrayCache; use Doctrine\Common\Cache\ArrayCache;
use Doctrine\Common\Cache\Cache; use Doctrine\Common\Cache\Cache;
use PHPUnit\Framework\TestCase; use PHPUnit\Framework\TestCase;
use Prophecy\Argument;
use Shlinkio\Shlink\Core\Middleware\QrCodeCacheMiddleware; use Shlinkio\Shlink\Core\Middleware\QrCodeCacheMiddleware;
use ShlinkioTest\Shlink\Common\Util\TestUtils;
use Zend\Diactoros\Response; use Zend\Diactoros\Response;
use Zend\Diactoros\ServerRequestFactory; use Zend\Diactoros\ServerRequestFactory;
use Zend\Diactoros\Uri; use Zend\Diactoros\Uri;
@ -29,18 +31,15 @@ class QrCodeCacheMiddlewareTest extends TestCase
/** /**
* @test * @test
*/ */
public function noCachedPathFallbacksToNextMiddleware() public function noCachedPathFallsBackToNextMiddleware()
{ {
$isCalled = false; $delegate = TestUtils::createDelegateMock();
$this->middleware->__invoke( $this->middleware->process(ServerRequestFactory::fromGlobals()->withUri(
ServerRequestFactory::fromGlobals(), new Uri('/foo/bar')
new Response(), ), $delegate->reveal());
function ($req, $resp) use (&$isCalled) {
$isCalled = true; $this->assertTrue($this->cache->contains('/foo/bar'));
return $resp; $delegate->process(Argument::any())->shouldHaveBeenCalledTimes(1);
}
);
$this->assertTrue($isCalled);
} }
/** /**
@ -51,19 +50,17 @@ class QrCodeCacheMiddlewareTest extends TestCase
$isCalled = false; $isCalled = false;
$uri = (new Uri())->withPath('/foo'); $uri = (new Uri())->withPath('/foo');
$this->cache->save('/foo', ['body' => 'the body', 'content-type' => 'image/png']); $this->cache->save('/foo', ['body' => 'the body', 'content-type' => 'image/png']);
$delegate = TestUtils::createDelegateMock();
$resp = $this->middleware->__invoke( $resp = $this->middleware->process(
ServerRequestFactory::fromGlobals()->withUri($uri), ServerRequestFactory::fromGlobals()->withUri($uri),
new Response(), $delegate->reveal()
function ($req, $resp) use (&$isCalled) {
$isCalled = true;
return $resp;
}
); );
$this->assertFalse($isCalled); $this->assertFalse($isCalled);
$resp->getBody()->rewind(); $resp->getBody()->rewind();
$this->assertEquals('the body', $resp->getBody()->getContents()); $this->assertEquals('the body', $resp->getBody()->getContents());
$this->assertEquals('image/png', $resp->getHeaderLine('Content-Type')); $this->assertEquals('image/png', $resp->getHeaderLine('Content-Type'));
$delegate->process(Argument::any())->shouldHaveBeenCalledTimes(0);
} }
} }