shlink/module/Core/test/Action/PreviewActionTest.php

103 lines
3.4 KiB
PHP
Raw Normal View History

2016-08-18 18:23:40 +03:00
<?php
2017-10-12 11:13:20 +03:00
declare(strict_types=1);
2016-08-18 18:23:40 +03:00
namespace ShlinkioTest\Shlink\Core\Action;
use finfo;
2017-03-24 22:34:18 +03:00
use PHPUnit\Framework\TestCase;
use Prophecy\Argument;
use Prophecy\Prophecy\MethodProphecy;
2016-08-18 18:23:40 +03:00
use Prophecy\Prophecy\ObjectProphecy;
2018-03-26 20:02:41 +03:00
use Psr\Http\Server\RequestHandlerInterface;
2016-08-18 18:23:40 +03:00
use Shlinkio\Shlink\Common\Service\PreviewGenerator;
use Shlinkio\Shlink\Core\Action\PreviewAction;
use Shlinkio\Shlink\Core\Entity\ShortUrl;
use Shlinkio\Shlink\Core\Exception\EntityDoesNotExistException;
2016-08-18 18:23:40 +03:00
use Shlinkio\Shlink\Core\Exception\InvalidShortCodeException;
use Shlinkio\Shlink\Core\Service\UrlShortener;
use ShlinkioTest\Shlink\Common\Util\TestUtils;
use Zend\Diactoros\Response;
2016-08-18 18:23:40 +03:00
use Zend\Diactoros\ServerRequestFactory;
class PreviewActionTest extends TestCase
{
/**
* @var PreviewAction
*/
protected $action;
/**
* @var ObjectProphecy
*/
private $previewGenerator;
/**
* @var ObjectProphecy
*/
private $urlShortener;
public function setUp()
{
$this->previewGenerator = $this->prophesize(PreviewGenerator::class);
$this->urlShortener = $this->prophesize(UrlShortener::class);
$this->action = new PreviewAction($this->previewGenerator->reveal(), $this->urlShortener->reveal());
}
/**
* @test
*/
public function invalidShortCodeFallsBackToNextMiddleware()
2016-08-18 18:23:40 +03:00
{
$shortCode = 'abc123';
$this->urlShortener->shortCodeToUrl($shortCode)->willThrow(EntityDoesNotExistException::class)
->shouldBeCalledTimes(1);
2018-03-26 20:02:41 +03:00
$delegate = $this->prophesize(RequestHandlerInterface::class);
$delegate->handle(Argument::cetera())->shouldBeCalledTimes(1)
->willReturn(new Response());
2016-08-18 18:23:40 +03:00
$this->action->process(
2016-08-18 18:23:40 +03:00
ServerRequestFactory::fromGlobals()->withAttribute('shortCode', $shortCode),
$delegate->reveal()
2016-08-18 18:23:40 +03:00
);
}
/**
* @test
*/
public function correctShortCodeReturnsImageResponse()
{
$shortCode = 'abc123';
$url = 'foobar.com';
$shortUrl = (new ShortUrl())->setLongUrl($url);
2016-08-18 18:23:40 +03:00
$path = __FILE__;
$this->urlShortener->shortCodeToUrl($shortCode)->willReturn($shortUrl)->shouldBeCalledTimes(1);
2016-08-18 18:23:40 +03:00
$this->previewGenerator->generatePreview($url)->willReturn($path)->shouldBeCalledTimes(1);
$resp = $this->action->process(
2016-08-18 18:23:40 +03:00
ServerRequestFactory::fromGlobals()->withAttribute('shortCode', $shortCode),
2018-03-26 20:02:41 +03:00
TestUtils::createReqHandlerMock()->reveal()
2016-08-18 18:23:40 +03:00
);
$this->assertEquals(filesize($path), $resp->getHeaderLine('Content-length'));
$this->assertEquals((new finfo(FILEINFO_MIME))->file($path), $resp->getHeaderLine('Content-type'));
2016-08-18 18:23:40 +03:00
}
/**
* @test
*/
public function invalidShortCodeExceptionFallsBackToNextMiddleware()
2016-08-18 18:23:40 +03:00
{
$shortCode = 'abc123';
$this->urlShortener->shortCodeToUrl($shortCode)->willThrow(InvalidShortCodeException::class)
->shouldBeCalledTimes(1);
2018-03-26 20:02:41 +03:00
$delegate = $this->prophesize(RequestHandlerInterface::class);
/** @var MethodProphecy $process */
2018-03-26 20:02:41 +03:00
$process = $delegate->handle(Argument::any())->willReturn(new Response());
2016-08-18 18:23:40 +03:00
$this->action->process(
2016-08-18 18:23:40 +03:00
ServerRequestFactory::fromGlobals()->withAttribute('shortCode', $shortCode),
$delegate->reveal()
2016-08-18 18:23:40 +03:00
);
$process->shouldHaveBeenCalledTimes(1);
2016-08-18 18:23:40 +03:00
}
}