Use method createOrUpdate from SegmentRepository in tests

[MAILPOET-3167]
This commit is contained in:
Jan Lysý
2020-12-03 15:44:52 +01:00
committed by Veljko V
parent 79abec8f13
commit 83cb272555
7 changed files with 53 additions and 74 deletions

View File

@ -5,20 +5,25 @@ namespace MailPoet\Entities;
use MailPoet\Newsletter\NewslettersRepository;
use MailPoet\Newsletter\Options\NewsletterOptionsRepository;
use MailPoet\Newsletter\Segment\NewsletterSegmentRepository;
use MailPoet\Segments\SegmentsRepository;
class NewsletterEntityTest extends \MailPoetTest {
/** @var NewslettersRepository */
private $newsletterRepository;
/** @var SegmentsRepository */
private $segmentRepository;
public function _before() {
$this->cleanup();
$this->newsletterRepository = $this->diContainer->get(NewslettersRepository::class);
$this->segmentRepository = $this->diContainer->get(SegmentsRepository::class);
}
public function testItRemovesOrphanedSegmentRelations() {
$newsletter = $this->createNewsletter();
$segment = $this->createSegment();
$segment = $this->segmentRepository->createOrUpdate('Segment', 'Segment description');
$newsletterSegment = new NewsletterSegmentEntity($newsletter, $segment);
$this->entityManager->persist($newsletterSegment);
$this->entityManager->flush();
@ -86,12 +91,6 @@ class NewsletterEntityTest extends \MailPoetTest {
return $newsletter;
}
private function createSegment(): SegmentEntity {
$segment = new SegmentEntity('Segment', SegmentEntity::TYPE_DEFAULT, 'Segment description');
$this->entityManager->persist($segment);
return $segment;
}
private function createOptionField(string $name): NewsletterOptionFieldEntity {
$newsletterOptionField = new NewsletterOptionFieldEntity();
$newsletterOptionField->setName($name);