Cavemanon/piratepoet
8
Fork 0
Code Pull Requests Actions Releases 16 Activity

Refactor Newsletter to use Doctrine instead of the NewsletterSegment model

Browse Source 
[MAILPOET-4363]
This commit is contained in:
Rodrigo Primo
2022-08-27 14:37:13 -03:00
committed by Aschepikov
parent f4053b0583
commit fd0acbd6cf
3 changed files with 28 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
mailpoet/tests/integration/Cron/Workers/SendingQueue/Tasks/NewsletterTest.php
View File

@@ -15,7 +15,6 @@ use MailPoet\Entities\NewsletterLinkEntity;
use MailPoet\Entities\NewsletterPostEntity;
use MailPoet\Entities\NewsletterSegmentEntity;
use MailPoet\Entities\ScheduledTaskEntity;
use MailPoet\Entities\SegmentEntity;
use MailPoet\Entities\SendingQueueEntity;
use MailPoet\Entities\SubscriberEntity;
use MailPoet\Logging\LoggerFactory;
@@ -28,6 +27,7 @@ use MailPoet\Router\Router;
use MailPoet\Settings\SettingsRepository;
use MailPoet\Tasks\Sending;
use MailPoet\Tasks\Sending as SendingTask;
use MailPoet\Test\DataFactories\Segment as SegmentFactory;
use MailPoet\Test\DataFactories\Subscriber as SubscriberFactory;
use MailPoet\WP\Emoji;
use MailPoet\WP\Functions as WPFunctions;
@@ -365,17 +365,18 @@ class NewsletterTest extends \MailPoetTest {
public function testItGetsSegments() {
$newsletterEntity = $this->newslettersRepository->findOneById($this->newsletter->id);
$this->assertInstanceOf(NewsletterEntity::class, $newsletterEntity);
$segmentIds = [];
for ($i = 1; $i <= 3; $i++) {
$segment = $this->entityManager->getReference(SegmentEntity::class, $i);
$this->assertInstanceOf(SegmentEntity::class, $segment);
$segment = (new SegmentFactory())->create();
$segmentIds[] = $segment->getId();
$newsletterSegment = new NewsletterSegmentEntity($newsletterEntity, $segment);
$this->entityManager->persist($newsletterSegment);
}
$this->entityManager->flush();
expect($this->newsletterTask->getNewsletterSegments($this->newsletter))->equals(
[1,2,3]
expect($this->newsletterTask->getNewsletterSegments($newsletterEntity))->equals(
$segmentIds
);
}