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

@ -45,9 +45,9 @@ class SegmentsTest extends \MailPoetTest {
$this->subscriberRepository = ContainerWrapper::getInstance()->get(SubscribersRepository::class);
$this->subscriberSegmentRepository = ContainerWrapper::getInstance()->get(SubscriberSegmentRepository::class);
$this->segment1 = $this->createSegment('Segment 1');
$this->segment2 = $this->createSegment('Segment 2');
$this->segment3 = $this->createSegment('Segment 3');
$this->segment1 = $this->segmentRepository->createOrUpdate('Segment 1');
$this->segment2 = $this->segmentRepository->createOrUpdate('Segment 2');
$this->segment3 = $this->segmentRepository->createOrUpdate('Segment 3');
}
public function testItCanGetASegment(): void {
@ -201,13 +201,6 @@ class SegmentsTest extends \MailPoetTest {
return $subscriberSegment;
}
private function createSegment(string $name): SegmentEntity {
$segment = new SegmentEntity($name, SegmentEntity::TYPE_DEFAULT, '');
$this->entityManager->persist($segment);
$this->entityManager->flush();
return $segment;
}
private function createSubsriber(string $email): SubscriberEntity {
$subscriber = new SubscriberEntity();
$subscriber->setEmail($email);