From b8143a5bb472213e5b202b4e747c0acf525cb1ab Mon Sep 17 00:00:00 2001
From: Alejandro Celaya <alejandrocelaya@gmail.com>
Date: Sun, 14 May 2023 13:04:45 +0200
Subject: [PATCH] Create VisitDeleterRepositoryTest

---
 .../Repository/VisitDeleterRepositoryTest.php | 63 +++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 module/Core/test-db/Visit/Repository/VisitDeleterRepositoryTest.php

diff --git a/module/Core/test-db/Visit/Repository/VisitDeleterRepositoryTest.php b/module/Core/test-db/Visit/Repository/VisitDeleterRepositoryTest.php
new file mode 100644
index 00000000..1598fc94
--- /dev/null
+++ b/module/Core/test-db/Visit/Repository/VisitDeleterRepositoryTest.php
@@ -0,0 +1,63 @@
+<?php
+
+declare(strict_types=1);
+
+namespace ShlinkioDbTest\Shlink\Core\Visit\Repository;
+
+use PHPUnit\Framework\Attributes\Test;
+use Shlinkio\Shlink\Core\ShortUrl\Entity\ShortUrl;
+use Shlinkio\Shlink\Core\ShortUrl\Model\ShortUrlCreation;
+use Shlinkio\Shlink\Core\ShortUrl\Model\Validation\ShortUrlInputFilter;
+use Shlinkio\Shlink\Core\ShortUrl\Resolver\PersistenceShortUrlRelationResolver;
+use Shlinkio\Shlink\Core\Visit\Entity\Visit;
+use Shlinkio\Shlink\Core\Visit\Model\Visitor;
+use Shlinkio\Shlink\Core\Visit\Repository\VisitDeleterRepository;
+use Shlinkio\Shlink\TestUtils\DbTest\DatabaseTestCase;
+
+class VisitDeleterRepositoryTest extends DatabaseTestCase
+{
+    private VisitDeleterRepository $repo;
+
+    protected function setUp(): void
+    {
+        $em = $this->getEntityManager();
+        $this->repo = new VisitDeleterRepository($em, $em->getClassMetadata(Visit::class));
+    }
+
+    #[Test]
+    public function deletesExpectedVisits(): void
+    {
+        $shortUrl1 = ShortUrl::withLongUrl('https://foo.com');
+        $this->getEntityManager()->persist($shortUrl1);
+        $this->getEntityManager()->persist(Visit::forValidShortUrl($shortUrl1, Visitor::emptyInstance()));
+        $this->getEntityManager()->persist(Visit::forValidShortUrl($shortUrl1, Visitor::emptyInstance()));
+
+        $shortUrl2 = ShortUrl::create(ShortUrlCreation::fromRawData([
+            ShortUrlInputFilter::LONG_URL => 'https://foo.com',
+            ShortUrlInputFilter::DOMAIN => 's.test',
+            ShortUrlInputFilter::CUSTOM_SLUG => 'foo',
+        ]), new PersistenceShortUrlRelationResolver($this->getEntityManager()));
+        $this->getEntityManager()->persist($shortUrl2);
+        $this->getEntityManager()->persist(Visit::forValidShortUrl($shortUrl2, Visitor::emptyInstance()));
+        $this->getEntityManager()->persist(Visit::forValidShortUrl($shortUrl2, Visitor::emptyInstance()));
+        $this->getEntityManager()->persist(Visit::forValidShortUrl($shortUrl2, Visitor::emptyInstance()));
+        $this->getEntityManager()->persist(Visit::forValidShortUrl($shortUrl2, Visitor::emptyInstance()));
+
+        $shortUrl3 = ShortUrl::create(ShortUrlCreation::fromRawData([
+            ShortUrlInputFilter::LONG_URL => 'https://foo.com',
+            ShortUrlInputFilter::CUSTOM_SLUG => 'foo',
+        ]), new PersistenceShortUrlRelationResolver($this->getEntityManager()));
+        $this->getEntityManager()->persist($shortUrl3);
+        $this->getEntityManager()->persist(Visit::forValidShortUrl($shortUrl3, Visitor::emptyInstance()));
+
+        $this->getEntityManager()->flush();
+
+        self::assertEquals(0, $this->repo->deleteShortUrlVisits(ShortUrl::withLongUrl('https://invalid')->setId('99')));
+        self::assertEquals(2, $this->repo->deleteShortUrlVisits($shortUrl1));
+        self::assertEquals(0, $this->repo->deleteShortUrlVisits($shortUrl1));
+        self::assertEquals(4, $this->repo->deleteShortUrlVisits($shortUrl2));
+        self::assertEquals(0, $this->repo->deleteShortUrlVisits($shortUrl2));
+        self::assertEquals(1, $this->repo->deleteShortUrlVisits($shortUrl3));
+        self::assertEquals(0, $this->repo->deleteShortUrlVisits($shortUrl3));
+    }
+}