Replace NewsletterLink with NewsletterLinkEntity in the class TrackTest

[MAILPOET-3816]
This commit is contained in:
Rodrigo Primo
2021-10-27 14:09:54 -03:00
committed by Veljko V
parent d1b13dbd70
commit 75da3dca45

View File

@@ -3,19 +3,22 @@
namespace MailPoet\Test\Router\Endpoints; namespace MailPoet\Test\Router\Endpoints;
use Codeception\Stub; use Codeception\Stub;
use MailPoet\Cron\Workers\StatsNotifications\NewsletterLinkRepository;
use MailPoet\Entities\NewsletterEntity; use MailPoet\Entities\NewsletterEntity;
use MailPoet\Entities\NewsletterLinkEntity; use MailPoet\Entities\NewsletterLinkEntity;
use MailPoet\Entities\ScheduledTaskEntity; use MailPoet\Entities\ScheduledTaskEntity;
use MailPoet\Entities\ScheduledTaskSubscriberEntity; use MailPoet\Entities\ScheduledTaskSubscriberEntity;
use MailPoet\Entities\SendingQueueEntity; use MailPoet\Entities\SendingQueueEntity;
use MailPoet\Entities\SubscriberEntity; use MailPoet\Entities\SubscriberEntity;
use MailPoet\Models\Newsletter; use MailPoet\Models\ScheduledTask;
use MailPoet\Models\NewsletterLink;
use MailPoet\Models\SendingQueue; use MailPoet\Models\SendingQueue;
use MailPoet\Newsletter\Sending\SendingQueuesRepository; use MailPoet\Newsletter\Sending\SendingQueuesRepository;
use MailPoet\Router\Endpoints\Track; use MailPoet\Router\Endpoints\Track;
use MailPoet\Subscribers\LinkTokens; use MailPoet\Subscribers\LinkTokens;
use MailPoet\Tasks\Sending;
use MailPoet\Tasks\Sending as SendingTask; use MailPoet\Tasks\Sending as SendingTask;
use MailPoet\Test\DataFactories\Newsletter as NewsletterFactory;
use MailPoet\Test\DataFactories\NewsletterLink as NewsletterLinkFactory;
class TrackTest extends \MailPoetTest { class TrackTest extends \MailPoetTest {
public $track; public $track;
@@ -177,31 +180,45 @@ class TrackTest extends \MailPoetTest {
public function testItGetsProperHashWhenDuplicateHashesExist() { public function testItGetsProperHashWhenDuplicateHashesExist() {
// create another newsletter and queue // create another newsletter and queue
$newsletter = Newsletter::create(); $newsletterFactory = new NewsletterFactory();
$newsletter->type = 'type'; $newsletter = $newsletterFactory
$newsletter = $newsletter->save(); ->withSendingQueue()
$queue = SendingTask::create(); ->create();
$queue->newsletterId = $newsletter->id; $scheduledTaskEntity = $newsletter->getLatestQueue()->getTask();
$queue->setSubscribers([$this->subscriber->getId()]); $scheduledTaskEntity->setType(Sending::TASK_TYPE);
$queue->updateProcessedSubscribers([$this->subscriber->getId()]); $this->entityManager->persist($scheduledTaskEntity);
$queue->save(); $this->entityManager->flush();
$scheduledTask = ScheduledTask::where('id', $scheduledTaskEntity->getId())->findOne();
$sendingQueue = SendingQueue::where('newsletter_id', $newsletter->getId())->findOne();
$this->assertInstanceOf(ScheduledTask::class, $scheduledTask);
$this->assertInstanceOf(SendingQueue::class, $sendingQueue);
$sendingTask = SendingTask::create($scheduledTask, $sendingQueue);
$sendingTask->newsletterId = $newsletter->getId();
$sendingTask->setSubscribers([$this->subscriber->getId()]);
$sendingTask->updateProcessedSubscribers([$this->subscriber->getId()]);
$sendingTask->save();
$trackData = $this->trackData; $trackData = $this->trackData;
$trackData['queue_id'] = $queue->id; $trackData['queue_id'] = $sendingTask->id;
$trackData['newsletter_id'] = $newsletter->id; $trackData['newsletter_id'] = $newsletter->getId();
// create another link with the same hash but different queue ID // create another link with the same hash but different queue ID
$link = NewsletterLink::create(); $newsletterLinkFactory = new NewsletterLinkFactory($newsletter);
$link->hash = $this->link->getHash(); $link = $newsletterLinkFactory
$link->url = $this->link->getUrl(); ->withHash($this->link->getHash())
$link->newsletterId = $trackData['newsletter_id']; ->withUrl($this->link->getUrl())
$link->queueId = $trackData['queue_id']; ->create();
$link = $link->save();
// assert that 2 links with identical hash exist // assert that 2 links with identical hash exist
$newsletterLink = NewsletterLink::where('hash', $link->hash)->findMany(); $newsletterLinkRepository = $this->diContainer->get(NewsletterLinkRepository::class);
$newsletterLink = $newsletterLinkRepository->findBy(['hash' => $link->getHash()]);
expect($newsletterLink)->count(2); expect($newsletterLink)->count(2);
// assert that the fetched link ID belong to the newly created link // assert that the fetched link ID belong to the newly created link
$processedData = $this->track->_processTrackData($trackData); $processedData = $this->track->_processTrackData($trackData);
expect($processedData->link->getId())->equals($link->id); expect($processedData->link->getId())->equals($link->getId());
} }
public function _after() { public function _after() {