mirrors/shlink
1
0
Fork 0
mirror of https://github.com/shlinkio/shlink.git synced 2025-02-25 11:49:05 +03:00
Code Issues Projects Releases Packages Wiki Activity

Used ShorturlIdentifier model whenever possible

Browse source
This commit is contained in:
Alejandro Celaya 2021-05-23 08:41:42 +02:00
parent cd19876419
commit 5e6d2881bc
11 changed files with 112 additions and 80 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
module/Core/src/Paginator/Adapter/VisitsPaginatorAdapter.php
View file

@ -33,8 +33,7 @@ class VisitsPaginatorAdapter extends AbstractCacheableCountPaginatorAdapter
public function getSlice($offset, $length): array // phpcs:ignore public function getSlice($offset, $length): array // phpcs:ignore
{ {
return $this->visitRepository->findVisitsByShortCode( return $this->visitRepository->findVisitsByShortCode(
$this->identifier->shortCode(), $this->identifier,
$this->identifier->domain(),
new VisitsListFiltering( new VisitsListFiltering(
$this->params->getDateRange(), $this->params->getDateRange(),
$this->params->excludeBots(), $this->params->excludeBots(),
@ -48,8 +47,7 @@ class VisitsPaginatorAdapter extends AbstractCacheableCountPaginatorAdapter
protected function doCount(): int protected function doCount(): int
{ {
return $this->visitRepository->countVisitsByShortCode( return $this->visitRepository->countVisitsByShortCode(
$this->identifier->shortCode(), $this->identifier,
$this->identifier->domain(),
new VisitsCountFiltering( new VisitsCountFiltering(
$this->params->getDateRange(), $this->params->getDateRange(),
$this->params->excludeBots(), $this->params->excludeBots(),

12
module/Core/src/Repository/ShortUrlRepository.php
View file

@ -174,9 +174,9 @@ class ShortUrlRepository extends EntitySpecificationRepository implements ShortU
return $query->getOneOrNullResult(); return $query->getOneOrNullResult();
} }
public function findOne(string $shortCode, ?string $domain = null, ?Specification $spec = null): ?ShortUrl public function findOne(ShortUrlIdentifier $identifier, ?Specification $spec = null): ?ShortUrl
{ {
$qb = $this->createFindOneQueryBuilder($shortCode, $domain, $spec); $qb = $this->createFindOneQueryBuilder($identifier, $spec);
$qb->select('s'); $qb->select('s');
return $qb->getQuery()->getOneOrNullResult(); return $qb->getQuery()->getOneOrNullResult();
@ -194,7 +194,7 @@ class ShortUrlRepository extends EntitySpecificationRepository implements ShortU
private function doShortCodeIsInUse(ShortUrlIdentifier $identifier, ?Specification $spec, ?int $lockMode): bool private function doShortCodeIsInUse(ShortUrlIdentifier $identifier, ?Specification $spec, ?int $lockMode): bool
{ {
$qb = $this->createFindOneQueryBuilder($identifier->shortCode(), $identifier->domain(), $spec); $qb = $this->createFindOneQueryBuilder($identifier, $spec);
$qb->select('s.id'); $qb->select('s.id');
$query = $qb->getQuery()->setLockMode($lockMode); $query = $qb->getQuery()->setLockMode($lockMode);
@ -202,16 +202,16 @@ class ShortUrlRepository extends EntitySpecificationRepository implements ShortU
return $query->getOneOrNullResult() !== null; return $query->getOneOrNullResult() !== null;
} }
private function createFindOneQueryBuilder(string $slug, ?string $domain, ?Specification $spec): QueryBuilder private function createFindOneQueryBuilder(ShortUrlIdentifier $identifier, ?Specification $spec): QueryBuilder
{ {
$qb = $this->getEntityManager()->createQueryBuilder(); $qb = $this->getEntityManager()->createQueryBuilder();
$qb->from(ShortUrl::class, 's') $qb->from(ShortUrl::class, 's')
->where($qb->expr()->isNotNull('s.shortCode')) ->where($qb->expr()->isNotNull('s.shortCode'))
->andWhere($qb->expr()->eq('s.shortCode', ':slug')) ->andWhere($qb->expr()->eq('s.shortCode', ':slug'))
->setParameter('slug', $slug) ->setParameter('slug', $identifier->shortCode())
->setMaxResults(1); ->setMaxResults(1);
$this->whereDomainIs($qb, $domain); $this->whereDomainIs($qb, $identifier->domain());
$this->applySpecification($qb, $spec, 's'); $this->applySpecification($qb, $spec, 's');

2
module/Core/src/Repository/ShortUrlRepositoryInterface.php
View file

@ -35,7 +35,7 @@ interface ShortUrlRepositoryInterface extends ObjectRepository, EntitySpecificat
public function findOneWithDomainFallback(string $shortCode, ?string $domain = null): ?ShortUrl; public function findOneWithDomainFallback(string $shortCode, ?string $domain = null): ?ShortUrl;
public function findOne(string $shortCode, ?string $domain = null, ?Specification $spec = null): ?ShortUrl; public function findOne(ShortUrlIdentifier $identifier, ?Specification $spec = null): ?ShortUrl;
public function shortCodeIsInUse(ShortUrlIdentifier $identifier, ?Specification $spec = null): bool; public function shortCodeIsInUse(ShortUrlIdentifier $identifier, ?Specification $spec = null): bool;

14
module/Core/src/Repository/VisitRepository.php
View file

@ -11,6 +11,7 @@ use Shlinkio\Shlink\Common\Util\DateRange;
use Shlinkio\Shlink\Core\Entity\ShortUrl; use Shlinkio\Shlink\Core\Entity\ShortUrl;
use Shlinkio\Shlink\Core\Entity\Visit; use Shlinkio\Shlink\Core\Entity\Visit;
use Shlinkio\Shlink\Core\Entity\VisitLocation; use Shlinkio\Shlink\Core\Entity\VisitLocation;
use Shlinkio\Shlink\Core\Model\ShortUrlIdentifier;
use Shlinkio\Shlink\Core\Visit\Persistence\VisitsCountFiltering; use Shlinkio\Shlink\Core\Visit\Persistence\VisitsCountFiltering;
use Shlinkio\Shlink\Core\Visit\Persistence\VisitsListFiltering; use Shlinkio\Shlink\Core\Visit\Persistence\VisitsListFiltering;
use Shlinkio\Shlink\Core\Visit\Spec\CountOfOrphanVisits; use Shlinkio\Shlink\Core\Visit\Spec\CountOfOrphanVisits;
@ -84,28 +85,27 @@ class VisitRepository extends EntitySpecificationRepository implements VisitRepo
/** /**
* @return Visit[] * @return Visit[]
*/ */
public function findVisitsByShortCode(string $shortCode, ?string $domain, VisitsListFiltering $filtering): array public function findVisitsByShortCode(ShortUrlIdentifier $identifier, VisitsListFiltering $filtering): array
{ {
$qb = $this->createVisitsByShortCodeQueryBuilder($shortCode, $domain, $filtering); $qb = $this->createVisitsByShortCodeQueryBuilder($identifier, $filtering);
return $this->resolveVisitsWithNativeQuery($qb, $filtering->limit(), $filtering->offset()); return $this->resolveVisitsWithNativeQuery($qb, $filtering->limit(), $filtering->offset());
} }
public function countVisitsByShortCode(string $shortCode, ?string $domain, VisitsCountFiltering $filtering): int public function countVisitsByShortCode(ShortUrlIdentifier $identifier, VisitsCountFiltering $filtering): int
{ {
$qb = $this->createVisitsByShortCodeQueryBuilder($shortCode, $domain, $filtering); $qb = $this->createVisitsByShortCodeQueryBuilder($identifier, $filtering);
$qb->select('COUNT(v.id)'); $qb->select('COUNT(v.id)');
return (int) $qb->getQuery()->getSingleScalarResult(); return (int) $qb->getQuery()->getSingleScalarResult();
} }
private function createVisitsByShortCodeQueryBuilder( private function createVisitsByShortCodeQueryBuilder(
string $shortCode, ShortUrlIdentifier $identifier,
?string $domain,
VisitsCountFiltering $filtering VisitsCountFiltering $filtering
): QueryBuilder { ): QueryBuilder {
/** @var ShortUrlRepositoryInterface $shortUrlRepo */ /** @var ShortUrlRepositoryInterface $shortUrlRepo */
$shortUrlRepo = $this->getEntityManager()->getRepository(ShortUrl::class); $shortUrlRepo = $this->getEntityManager()->getRepository(ShortUrl::class);
$shortUrl = $shortUrlRepo->findOne($shortCode, $domain, $filtering->spec()); $shortUrl = $shortUrlRepo->findOne($identifier, $filtering->spec());
$shortUrlId = $shortUrl !== null ? $shortUrl->getId() : -1; $shortUrlId = $shortUrl !== null ? $shortUrl->getId() : -1;
// Parameters in this query need to be part of the query itself, as we need to use it a sub-query later // Parameters in this query need to be part of the query itself, as we need to use it a sub-query later

5
module/Core/src/Repository/VisitRepositoryInterface.php
View file

@ -7,6 +7,7 @@ namespace Shlinkio\Shlink\Core\Repository;
use Doctrine\Persistence\ObjectRepository; use Doctrine\Persistence\ObjectRepository;
use Happyr\DoctrineSpecification\Repository\EntitySpecificationRepositoryInterface; use Happyr\DoctrineSpecification\Repository\EntitySpecificationRepositoryInterface;
use Shlinkio\Shlink\Core\Entity\Visit; use Shlinkio\Shlink\Core\Entity\Visit;
use Shlinkio\Shlink\Core\Model\ShortUrlIdentifier;
use Shlinkio\Shlink\Core\Visit\Persistence\VisitsCountFiltering; use Shlinkio\Shlink\Core\Visit\Persistence\VisitsCountFiltering;
use Shlinkio\Shlink\Core\Visit\Persistence\VisitsListFiltering; use Shlinkio\Shlink\Core\Visit\Persistence\VisitsListFiltering;
use Shlinkio\Shlink\Rest\Entity\ApiKey; use Shlinkio\Shlink\Rest\Entity\ApiKey;
@ -33,9 +34,9 @@ interface VisitRepositoryInterface extends ObjectRepository, EntitySpecification
/** /**
* @return Visit[] * @return Visit[]
*/ */
public function findVisitsByShortCode(string $shortCode, ?string $domain, VisitsListFiltering $filtering): array; public function findVisitsByShortCode(ShortUrlIdentifier $identifier, VisitsListFiltering $filtering): array;
public function countVisitsByShortCode(string $shortCode, ?string $domain, VisitsCountFiltering $filtering): int; public function countVisitsByShortCode(ShortUrlIdentifier $identifier, VisitsCountFiltering $filtering): int;
/** /**
* @return Visit[] * @return Visit[]

6
module/Core/src/Service/ShortUrl/ShortUrlResolver.php
View file

@ -27,11 +27,7 @@ class ShortUrlResolver implements ShortUrlResolverInterface
{ {
/** @var ShortUrlRepository $shortUrlRepo */ /** @var ShortUrlRepository $shortUrlRepo */
$shortUrlRepo = $this->em->getRepository(ShortUrl::class); $shortUrlRepo = $this->em->getRepository(ShortUrl::class);
$shortUrl = $shortUrlRepo->findOne( $shortUrl = $shortUrlRepo->findOne($identifier, $apiKey !== null ? $apiKey->spec() : null);
$identifier->shortCode(),
$identifier->domain(),
$apiKey !== null ? $apiKey->spec() : null,
);
if ($shortUrl === null) { if ($shortUrl === null) {
throw ShortUrlNotFoundException::fromNotFound($identifier); throw ShortUrlNotFoundException::fromNotFound($identifier);
} }

16
module/Core/test-db/Repository/ShortUrlRepositoryTest.php
View file

@ -210,12 +210,16 @@ class ShortUrlRepositoryTest extends DatabaseTestCase
$this->getEntityManager()->flush(); $this->getEntityManager()->flush();
self::assertNotNull($this->repo->findOne('my-cool-slug')); self::assertNotNull($this->repo->findOne(ShortUrlIdentifier::fromShortCodeAndDomain('my-cool-slug')));
self::assertNull($this->repo->findOne('my-cool-slug', 'doma.in')); self::assertNull($this->repo->findOne(ShortUrlIdentifier::fromShortCodeAndDomain('my-cool-slug', 'doma.in')));
self::assertNull($this->repo->findOne('slug-not-in-use')); self::assertNull($this->repo->findOne(ShortUrlIdentifier::fromShortCodeAndDomain('slug-not-in-use')));
self::assertNull($this->repo->findOne('another-slug')); self::assertNull($this->repo->findOne(ShortUrlIdentifier::fromShortCodeAndDomain('another-slug')));
self::assertNull($this->repo->findOne('another-slug', 'example.com')); self::assertNull($this->repo->findOne(
self::assertNotNull($this->repo->findOne('another-slug', 'doma.in')); ShortUrlIdentifier::fromShortCodeAndDomain('another-slug', 'example.com'),
));
self::assertNotNull($this->repo->findOne(
ShortUrlIdentifier::fromShortCodeAndDomain('another-slug', 'doma.in'),
));
} }
/** @test */ /** @test */

97
module/Core/test-db/Repository/VisitRepositoryTest.php
View file

@ -11,6 +11,7 @@ use Shlinkio\Shlink\Core\Entity\Domain;
use Shlinkio\Shlink\Core\Entity\ShortUrl; use Shlinkio\Shlink\Core\Entity\ShortUrl;
use Shlinkio\Shlink\Core\Entity\Visit; use Shlinkio\Shlink\Core\Entity\Visit;
use Shlinkio\Shlink\Core\Entity\VisitLocation; use Shlinkio\Shlink\Core\Entity\VisitLocation;
use Shlinkio\Shlink\Core\Model\ShortUrlIdentifier;
use Shlinkio\Shlink\Core\Model\ShortUrlMeta; use Shlinkio\Shlink\Core\Model\ShortUrlMeta;
use Shlinkio\Shlink\Core\Model\Visitor; use Shlinkio\Shlink\Core\Model\Visitor;
use Shlinkio\Shlink\Core\Repository\VisitRepository; use Shlinkio\Shlink\Core\Repository\VisitRepository;
@ -89,32 +90,46 @@ class VisitRepositoryTest extends DatabaseTestCase
{ {
[$shortCode, $domain] = $this->createShortUrlsAndVisits(); [$shortCode, $domain] = $this->createShortUrlsAndVisits();
self::assertCount(0, $this->repo->findVisitsByShortCode('invalid', null, new VisitsListFiltering())); self::assertCount(0, $this->repo->findVisitsByShortCode(
self::assertCount(6, $this->repo->findVisitsByShortCode($shortCode, null, new VisitsListFiltering())); ShortUrlIdentifier::fromShortCodeAndDomain('invalid'),
self::assertCount(4, $this->repo->findVisitsByShortCode($shortCode, null, new VisitsListFiltering(null, true))); new VisitsListFiltering(),
self::assertCount(3, $this->repo->findVisitsByShortCode($shortCode, $domain, new VisitsListFiltering())); ));
self::assertCount(2, $this->repo->findVisitsByShortCode($shortCode, null, new VisitsListFiltering( self::assertCount(6, $this->repo->findVisitsByShortCode(
DateRange::withStartAndEndDate(Chronos::parse('2016-01-02'), Chronos::parse('2016-01-03')), ShortUrlIdentifier::fromShortCodeAndDomain($shortCode),
))); new VisitsListFiltering(),
self::assertCount(4, $this->repo->findVisitsByShortCode($shortCode, null, new VisitsListFiltering( ));
DateRange::withStartDate(Chronos::parse('2016-01-03')), self::assertCount(4, $this->repo->findVisitsByShortCode(
))); ShortUrlIdentifier::fromShortCodeAndDomain($shortCode),
self::assertCount(1, $this->repo->findVisitsByShortCode($shortCode, $domain, new VisitsListFiltering( new VisitsListFiltering(null, true),
DateRange::withStartDate(Chronos::parse('2016-01-03')), ));
)));
self::assertCount(3, $this->repo->findVisitsByShortCode( self::assertCount(3, $this->repo->findVisitsByShortCode(
$shortCode, ShortUrlIdentifier::fromShortCodeAndDomain($shortCode, $domain),
null, new VisitsListFiltering(),
));
self::assertCount(2, $this->repo->findVisitsByShortCode(
ShortUrlIdentifier::fromShortCodeAndDomain($shortCode),
new VisitsListFiltering(
DateRange::withStartAndEndDate(Chronos::parse('2016-01-02'), Chronos::parse('2016-01-03')),
),
));
self::assertCount(4, $this->repo->findVisitsByShortCode(
ShortUrlIdentifier::fromShortCodeAndDomain($shortCode),
new VisitsListFiltering(DateRange::withStartDate(Chronos::parse('2016-01-03'))),
));
self::assertCount(1, $this->repo->findVisitsByShortCode(
ShortUrlIdentifier::fromShortCodeAndDomain($shortCode, $domain),
new VisitsListFiltering(DateRange::withStartDate(Chronos::parse('2016-01-03'))),
));
self::assertCount(3, $this->repo->findVisitsByShortCode(
ShortUrlIdentifier::fromShortCodeAndDomain($shortCode),
new VisitsListFiltering(null, false, null, 3, 2), new VisitsListFiltering(null, false, null, 3, 2),
)); ));
self::assertCount(2, $this->repo->findVisitsByShortCode( self::assertCount(2, $this->repo->findVisitsByShortCode(
$shortCode, ShortUrlIdentifier::fromShortCodeAndDomain($shortCode),
null,
new VisitsListFiltering(null, false, null, 5, 4), new VisitsListFiltering(null, false, null, 5, 4),
)); ));
self::assertCount(1, $this->repo->findVisitsByShortCode( self::assertCount(1, $this->repo->findVisitsByShortCode(
$shortCode, ShortUrlIdentifier::fromShortCodeAndDomain($shortCode, $domain),
$domain,
new VisitsListFiltering(null, false, null, 3, 2), new VisitsListFiltering(null, false, null, 3, 2),
)); ));
} }
@ -124,22 +139,36 @@ class VisitRepositoryTest extends DatabaseTestCase
{ {
[$shortCode, $domain] = $this->createShortUrlsAndVisits(); [$shortCode, $domain] = $this->createShortUrlsAndVisits();
self::assertEquals(0, $this->repo->countVisitsByShortCode('invalid', null, new VisitsCountFiltering())); self::assertEquals(0, $this->repo->countVisitsByShortCode(
self::assertEquals(6, $this->repo->countVisitsByShortCode($shortCode, null, new VisitsCountFiltering())); ShortUrlIdentifier::fromShortCodeAndDomain('invalid'),
self::assertEquals(4, $this->repo->countVisitsByShortCode($shortCode, null, new VisitsCountFiltering( new VisitsCountFiltering(),
null, ));
true, self::assertEquals(6, $this->repo->countVisitsByShortCode(
))); ShortUrlIdentifier::fromShortCodeAndDomain($shortCode),
self::assertEquals(3, $this->repo->countVisitsByShortCode($shortCode, $domain, new VisitsCountFiltering())); new VisitsCountFiltering(),
self::assertEquals(2, $this->repo->countVisitsByShortCode($shortCode, null, new VisitsCountFiltering( ));
self::assertEquals(4, $this->repo->countVisitsByShortCode(
ShortUrlIdentifier::fromShortCodeAndDomain($shortCode),
new VisitsCountFiltering(null, true),
));
self::assertEquals(3, $this->repo->countVisitsByShortCode(
ShortUrlIdentifier::fromShortCodeAndDomain($shortCode, $domain),
new VisitsCountFiltering(),
));
self::assertEquals(2, $this->repo->countVisitsByShortCode(
ShortUrlIdentifier::fromShortCodeAndDomain($shortCode),
new VisitsCountFiltering(
DateRange::withStartAndEndDate(Chronos::parse('2016-01-02'), Chronos::parse('2016-01-03')), DateRange::withStartAndEndDate(Chronos::parse('2016-01-02'), Chronos::parse('2016-01-03')),
))); ),
self::assertEquals(4, $this->repo->countVisitsByShortCode($shortCode, null, new VisitsCountFiltering( ));
DateRange::withStartDate(Chronos::parse('2016-01-03')), self::assertEquals(4, $this->repo->countVisitsByShortCode(
))); ShortUrlIdentifier::fromShortCodeAndDomain($shortCode),
self::assertEquals(1, $this->repo->countVisitsByShortCode($shortCode, $domain, new VisitsCountFiltering( new VisitsCountFiltering(DateRange::withStartDate(Chronos::parse('2016-01-03'))),
DateRange::withStartDate(Chronos::parse('2016-01-03')), ));
))); self::assertEquals(1, $this->repo->countVisitsByShortCode(
ShortUrlIdentifier::fromShortCodeAndDomain($shortCode, $domain),
new VisitsCountFiltering(DateRange::withStartDate(Chronos::parse('2016-01-03'))),
));
} }
/** @test */ /** @test */

6
module/Core/test/Paginator/Adapter/VisitsPaginatorAdapterTest.php
View file

@ -35,8 +35,7 @@ class VisitsPaginatorAdapterTest extends TestCase
$offset = 5; $offset = 5;
$adapter = $this->createAdapter(null); $adapter = $this->createAdapter(null);
$findVisits = $this->repo->findVisitsByShortCode( $findVisits = $this->repo->findVisitsByShortCode(
'', ShortUrlIdentifier::fromShortCodeAndDomain(''),
null,
new VisitsListFiltering(new DateRange(), false, null, $limit, $offset), new VisitsListFiltering(new DateRange(), false, null, $limit, $offset),
)->willReturn([]); )->willReturn([]);
@ -54,8 +53,7 @@ class VisitsPaginatorAdapterTest extends TestCase
$apiKey = ApiKey::create(); $apiKey = ApiKey::create();
$adapter = $this->createAdapter($apiKey); $adapter = $this->createAdapter($apiKey);
$countVisits = $this->repo->countVisitsByShortCode( $countVisits = $this->repo->countVisitsByShortCode(
'', ShortUrlIdentifier::fromShortCodeAndDomain(''),
null,
new VisitsCountFiltering(new DateRange(), false, $apiKey->spec()), new VisitsCountFiltering(new DateRange(), false, $apiKey->spec()),
)->willReturn(3); )->willReturn(3);

10
module/Core/test/Service/ShortUrl/ShortUrlResolverTest.php
View file

@ -46,12 +46,13 @@ class ShortUrlResolverTest extends TestCase
{ {
$shortUrl = ShortUrl::withLongUrl('expected_url'); $shortUrl = ShortUrl::withLongUrl('expected_url');
$shortCode = $shortUrl->getShortCode(); $shortCode = $shortUrl->getShortCode();
$identifier = ShortUrlIdentifier::fromShortCodeAndDomain($shortCode);
$repo = $this->prophesize(ShortUrlRepositoryInterface::class); $repo = $this->prophesize(ShortUrlRepositoryInterface::class);
$findOne = $repo->findOne($shortCode, null, $apiKey !== null ? $apiKey->spec() : null)->willReturn($shortUrl); $findOne = $repo->findOne($identifier, $apiKey !== null ? $apiKey->spec() : null)->willReturn($shortUrl);
$getRepo = $this->em->getRepository(ShortUrl::class)->willReturn($repo->reveal()); $getRepo = $this->em->getRepository(ShortUrl::class)->willReturn($repo->reveal());
$result = $this->urlResolver->resolveShortUrl(new ShortUrlIdentifier($shortCode), $apiKey); $result = $this->urlResolver->resolveShortUrl($identifier, $apiKey);
self::assertSame($shortUrl, $result); self::assertSame($shortUrl, $result);
$findOne->shouldHaveBeenCalledOnce(); $findOne->shouldHaveBeenCalledOnce();
@ -65,16 +66,17 @@ class ShortUrlResolverTest extends TestCase
public function exceptionIsThrownIfShortcodeIsNotFound(?ApiKey $apiKey): void public function exceptionIsThrownIfShortcodeIsNotFound(?ApiKey $apiKey): void
{ {
$shortCode = 'abc123'; $shortCode = 'abc123';
$identifier = ShortUrlIdentifier::fromShortCodeAndDomain($shortCode);
$repo = $this->prophesize(ShortUrlRepositoryInterface::class); $repo = $this->prophesize(ShortUrlRepositoryInterface::class);
$findOne = $repo->findOne($shortCode, null, $apiKey !== null ? $apiKey->spec() : null)->willReturn(null); $findOne = $repo->findOne($identifier, $apiKey !== null ? $apiKey->spec() : null)->willReturn(null);
$getRepo = $this->em->getRepository(ShortUrl::class)->willReturn($repo->reveal(), $apiKey); $getRepo = $this->em->getRepository(ShortUrl::class)->willReturn($repo->reveal(), $apiKey);
$this->expectException(ShortUrlNotFoundException::class); $this->expectException(ShortUrlNotFoundException::class);
$findOne->shouldBeCalledOnce(); $findOne->shouldBeCalledOnce();
$getRepo->shouldBeCalledOnce(); $getRepo->shouldBeCalledOnce();
$this->urlResolver->resolveShortUrl(new ShortUrlIdentifier($shortCode), $apiKey); $this->urlResolver->resolveShortUrl($identifier, $apiKey);
} }
/** @test */ /** @test */

16
module/Core/test/Visit/VisitsStatsHelperTest.php
View file

@ -79,20 +79,22 @@ class VisitsStatsHelperTest extends TestCase
public function infoReturnsVisitsForCertainShortCode(?ApiKey $apiKey): void public function infoReturnsVisitsForCertainShortCode(?ApiKey $apiKey): void
{ {
$shortCode = '123ABC'; $shortCode = '123ABC';
$identifier = ShortUrlIdentifier::fromShortCodeAndDomain($shortCode);
$spec = $apiKey === null ? null : $apiKey->spec(); $spec = $apiKey === null ? null : $apiKey->spec();
$repo = $this->prophesize(ShortUrlRepositoryInterface::class); $repo = $this->prophesize(ShortUrlRepositoryInterface::class);
$count = $repo->shortCodeIsInUse(ShortUrlIdentifier::fromShortCodeAndDomain($shortCode), $spec)->willReturn( $count = $repo->shortCodeIsInUse($identifier, $spec)->willReturn(
true, true,
); );
$this->em->getRepository(ShortUrl::class)->willReturn($repo->reveal())->shouldBeCalledOnce(); $this->em->getRepository(ShortUrl::class)->willReturn($repo->reveal())->shouldBeCalledOnce();
$list = map(range(0, 1), fn () => Visit::forValidShortUrl(ShortUrl::createEmpty(), Visitor::emptyInstance())); $list = map(range(0, 1), fn () => Visit::forValidShortUrl(ShortUrl::createEmpty(), Visitor::emptyInstance()));
$repo2 = $this->prophesize(VisitRepository::class); $repo2 = $this->prophesize(VisitRepository::class);
$repo2->findVisitsByShortCode($shortCode, null, Argument::type(VisitsListFiltering::class))->willReturn($list); $repo2->findVisitsByShortCode($identifier, Argument::type(VisitsListFiltering::class))->willReturn($list);
$repo2->countVisitsByShortCode($shortCode, null, Argument::type(VisitsCountFiltering::class))->willReturn(1); $repo2->countVisitsByShortCode($identifier, Argument::type(VisitsCountFiltering::class))->willReturn(1);
$this->em->getRepository(Visit::class)->willReturn($repo2->reveal())->shouldBeCalledOnce(); $this->em->getRepository(Visit::class)->willReturn($repo2->reveal())->shouldBeCalledOnce();
$paginator = $this->helper->visitsForShortUrl(new ShortUrlIdentifier($shortCode), new VisitsParams(), $apiKey); $paginator = $this->helper->visitsForShortUrl($identifier, new VisitsParams(), $apiKey);
self::assertEquals($list, ArrayUtils::iteratorToArray($paginator->getCurrentPageResults())); self::assertEquals($list, ArrayUtils::iteratorToArray($paginator->getCurrentPageResults()));
$count->shouldHaveBeenCalledOnce(); $count->shouldHaveBeenCalledOnce();
@ -102,8 +104,10 @@ class VisitsStatsHelperTest extends TestCase
public function throwsExceptionWhenRequestingVisitsForInvalidShortCode(): void public function throwsExceptionWhenRequestingVisitsForInvalidShortCode(): void
{ {
$shortCode = '123ABC'; $shortCode = '123ABC';
$identifier = ShortUrlIdentifier::fromShortCodeAndDomain($shortCode);
$repo = $this->prophesize(ShortUrlRepositoryInterface::class); $repo = $this->prophesize(ShortUrlRepositoryInterface::class);
$count = $repo->shortCodeIsInUse(ShortUrlIdentifier::fromShortCodeAndDomain($shortCode), null)->willReturn( $count = $repo->shortCodeIsInUse($identifier, null)->willReturn(
false, false,
); );
$this->em->getRepository(ShortUrl::class)->willReturn($repo->reveal())->shouldBeCalledOnce(); $this->em->getRepository(ShortUrl::class)->willReturn($repo->reveal())->shouldBeCalledOnce();
@ -111,7 +115,7 @@ class VisitsStatsHelperTest extends TestCase
$this->expectException(ShortUrlNotFoundException::class); $this->expectException(ShortUrlNotFoundException::class);
$count->shouldBeCalledOnce(); $count->shouldBeCalledOnce();
$this->helper->visitsForShortUrl(new ShortUrlIdentifier($shortCode), new VisitsParams()); $this->helper->visitsForShortUrl($identifier, new VisitsParams());
} }
/** @test */ /** @test */