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

Skip used segments when bulk trashing segments

Browse Source 
[MAILPOET-3463]
This commit is contained in:
Rostislav Wolny
2021-03-23 14:47:04 +01:00
committed by Veljko V
parent b17236b88c
commit a264951904
2 changed files with 55 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
tests/integration/Segments/SegmentsRepositoryTest.php
View File

@@ -4,6 +4,8 @@ namespace MailPoet\Segments;
use MailPoet\Entities\DynamicSegmentFilterData;
use MailPoet\Entities\DynamicSegmentFilterEntity;
use MailPoet\Entities\NewsletterEntity;
use MailPoet\Entities\NewsletterSegmentEntity;
use MailPoet\Entities\SegmentEntity;
class SegmentsRepositoryTest extends \MailPoetTest {
@@ -40,6 +42,32 @@ class SegmentsRepositoryTest extends \MailPoetTest {
expect($segment2->getDeletedAt())->isInstanceOf(\DateTimeInterface::class);
}
public function testItSkipTrashingForActivelyUsedDefaultSegments() {
$segment1 = $this->createDefaultSegment('Segment 1');
$segment2 = $this->createDefaultSegment('Segment 2');
$this->addActiveNewsletterToSegment($segment1);
$this->entityManager->flush();
$result = $this->segmentsRepository->bulkTrash([$segment1->getId(), $segment2->getId()]);
$this->entityManager->refresh($segment1);
$this->entityManager->refresh($segment2);
expect($result)->equals(1);
expect($segment1->getDeletedAt())->null();
expect($segment2->getDeletedAt())->isInstanceOf(\DateTimeInterface::class);
}
public function testItSkipTrashingForActivelyUsedDynamicSegments() {
$segment1 = $this->createDynamicSegmentEntityForEditorUsers();
$segment2 = $this->createDynamicSegmentEntityForEditorUsers();
$this->addActiveNewsletterToSegment($segment2);
$this->entityManager->flush();
$result = $this->segmentsRepository->bulkTrash([$segment1->getId(), $segment2->getId()], SegmentEntity::TYPE_DYNAMIC);
$this->entityManager->refresh($segment1);
$this->entityManager->refresh($segment2);
expect($result)->equals(1);
expect($segment1->getDeletedAt())->isInstanceOf(\DateTimeInterface::class);
expect($segment2->getDeletedAt())->null();
}
private function createDefaultSegment(string $name): SegmentEntity {
$segment = new SegmentEntity($name, SegmentEntity::TYPE_DEFAULT, 'description');
$this->entityManager->persist($segment);
@@ -58,9 +86,21 @@ class SegmentsRepositoryTest extends \MailPoetTest {
return $segment;
}
private function addActiveNewsletterToSegment(SegmentEntity $segmentEntity) {
$newsletter = new NewsletterEntity();
$newsletter->setSubject('Subject');
$newsletter->setType(NewsletterEntity::TYPE_WELCOME);
$newsletter->setStatus(NewsletterEntity::STATUS_ACTIVE);
$newsletterSegment = new NewsletterSegmentEntity($newsletter, $segmentEntity);
$this->entityManager->persist($newsletter);
$this->entityManager->persist($newsletterSegment);
}
private function cleanup() {
$this->truncateEntity(SegmentEntity::class);
$this->truncateEntity(DynamicSegmentFilterEntity::class);
$this->truncateEntity(NewsletterEntity::class);
$this->truncateEntity(NewsletterSegmentEntity::class);
}
public function _after() {