251 lines
11 KiB
PHP
251 lines
11 KiB
PHP
<?php
|
|
|
|
namespace MailPoet\Segments;
|
|
|
|
use MailPoet\Entities\DynamicSegmentFilterData;
|
|
use MailPoet\Entities\DynamicSegmentFilterEntity;
|
|
use MailPoet\Entities\SegmentEntity;
|
|
use MailPoet\Entities\SubscriberEntity;
|
|
use MailPoet\Entities\SubscriberSegmentEntity;
|
|
use MailPoet\Subscribers\Source;
|
|
|
|
class SegmentSubscribersRepositoryTest extends \MailPoetTest {
|
|
|
|
/** @var SegmentSubscribersRepository */
|
|
private $repository;
|
|
|
|
/** @var SegmentsRepository */
|
|
private $segmentRepository;
|
|
|
|
public function _before() {
|
|
parent::_before();
|
|
$this->segmentRepository = $this->diContainer->get(SegmentsRepository::class);
|
|
$this->repository = $this->diContainer->get(SegmentSubscribersRepository::class);
|
|
$this->cleanup();
|
|
}
|
|
|
|
public function testItReturnsOnlySubscribedSubscribersForStaticSegment() {
|
|
$segment = $this->segmentRepository->createOrUpdate('Segment' . rand(0, 10000));
|
|
|
|
$this->createSubscriberEntity(); // Subscriber without segment
|
|
|
|
$subscriberInSegment = $this->createSubscriberEntity();
|
|
$subscriberInSegment->getSegments()->add(
|
|
$this->createSubscriberSegmentEntity($segment, $subscriberInSegment)
|
|
);
|
|
|
|
$globallyUnsubscribedSubscriberInSegment = $this->createSubscriberEntity();
|
|
$subscriberInSegment->getSegments()->add(
|
|
$this->createSubscriberSegmentEntity($segment, $globallyUnsubscribedSubscriberInSegment)
|
|
);
|
|
$globallyUnsubscribedSubscriberInSegment->setStatus(SubscriberEntity::STATUS_UNSUBSCRIBED);
|
|
|
|
$unsubscribedSubscriberInSegment = $this->createSubscriberEntity();
|
|
$subscriberSegment = $this->createSubscriberSegmentEntity($segment, $unsubscribedSubscriberInSegment);
|
|
$subscriberSegment->setStatus(SubscriberEntity::STATUS_UNSUBSCRIBED);
|
|
$subscriberInSegment->getSegments()->add(
|
|
$subscriberSegment
|
|
);
|
|
$this->entityManager->flush();
|
|
|
|
$count = $this->repository->getSubscribersCount((int)$segment->getId());
|
|
expect($count)->equals(3);
|
|
$count = $this->repository->getSubscribersCount((int)$segment->getId(), SubscriberEntity::STATUS_SUBSCRIBED);
|
|
expect($count)->equals(1);
|
|
$ids = $this->repository->getSubscriberIdsInSegment((int)$segment->getId());
|
|
expect($ids)->equals([$subscriberInSegment->getId()]);
|
|
$filteredIds = $this->repository->findSubscribersIdsInSegment((int)$segment->getId(), [$subscriberInSegment->getId(), 20, 30]);
|
|
expect($filteredIds)->equals([$subscriberInSegment->getId()]);
|
|
$filteredIds = $this->repository->findSubscribersIdsInSegment((int)$segment->getId(), [$globallyUnsubscribedSubscriberInSegment->getId()]);
|
|
expect($filteredIds)->equals([]);
|
|
}
|
|
|
|
public function testItReturnsSubscribersInDynamicSegments() {
|
|
$segment = $this->createDynamicSegmentEntity();
|
|
|
|
$wpUserEmail = 'user-role-test1@example.com';
|
|
$this->tester->deleteWordPressUser($wpUserEmail);
|
|
$this->tester->createWordPressUser($wpUserEmail, 'editor');
|
|
$wpUserSubscriber = $this->entityManager
|
|
->getRepository(SubscriberEntity::class)
|
|
->findOneBy(['email' => $wpUserEmail]);
|
|
assert($wpUserSubscriber instanceof SubscriberEntity);
|
|
$wpUserSubscriber->setStatus(SubscriberEntity::STATUS_SUBSCRIBED);
|
|
assert($wpUserSubscriber instanceof SubscriberEntity);
|
|
$subscriberNoList = $this->createSubscriberEntity(); // Subscriber without segment
|
|
$this->entityManager->flush();
|
|
|
|
$count = $this->repository->getSubscribersCount((int)$segment->getId());
|
|
expect($count)->equals(1);
|
|
$count = $this->repository->getSubscribersCount((int)$segment->getId(), SubscriberEntity::STATUS_UNSUBSCRIBED);
|
|
expect($count)->equals(0);
|
|
$count = $this->repository->getSubscribersCount((int)$segment->getId(), SubscriberEntity::STATUS_SUBSCRIBED);
|
|
expect($count)->equals(1);
|
|
$ids = $this->repository->getSubscriberIdsInSegment((int)$segment->getId());
|
|
expect($ids)->equals([$wpUserSubscriber->getId()]);
|
|
$filteredIds = $this->repository->findSubscribersIdsInSegment((int)$segment->getId(), [$wpUserSubscriber->getId(), 20, 30]);
|
|
expect($filteredIds)->equals([$wpUserSubscriber->getId()]);
|
|
$filteredIds = $this->repository->findSubscribersIdsInSegment((int)$segment->getId(), [$subscriberNoList->getId()]);
|
|
expect($filteredIds)->equals([]);
|
|
|
|
$this->tester->deleteWordPressUser($wpUserEmail);
|
|
}
|
|
|
|
public function testGetSubscribersStatisticsCount() {
|
|
$segment = $this->segmentRepository->createOrUpdate('Segment' . rand(0, 10000));
|
|
$this->entityManager->persist($segment);
|
|
$subscribersData = [
|
|
['status' => SubscriberEntity::STATUS_UNSUBSCRIBED],
|
|
['status' => SubscriberEntity::STATUS_SUBSCRIBED],
|
|
['status' => SubscriberEntity::STATUS_UNCONFIRMED],
|
|
['status' => SubscriberEntity::STATUS_BOUNCED],
|
|
];
|
|
$subscriberSegments = [];
|
|
$subscribers = [];
|
|
|
|
// normal subscribers
|
|
foreach ($subscribersData as $subscriberData) {
|
|
$subscriber = $this->createSubscriberEntity();
|
|
$subscriber->setStatus($subscriberData['status']);
|
|
$subscriberSegments[] = $this->createSubscriberSegmentEntity($segment, $subscriber);
|
|
$subscribers[] = $subscriber;
|
|
}
|
|
$this->entityManager->flush();
|
|
|
|
$subscribersCount = $this->repository->getSubscribersStatisticsCount($segment);
|
|
expect($subscribersCount[SubscriberEntity::STATUS_SUBSCRIBED])->equals(1);
|
|
expect($subscribersCount[SubscriberEntity::STATUS_UNSUBSCRIBED])->equals(1);
|
|
expect($subscribersCount[SubscriberEntity::STATUS_UNCONFIRMED])->equals(1);
|
|
expect($subscribersCount[SubscriberEntity::STATUS_BOUNCED])->equals(1);
|
|
|
|
// unsubscribed from this particular segment
|
|
foreach ($subscriberSegments as $subscriberSegment) {
|
|
$subscriberSegment->setStatus(SubscriberEntity::STATUS_UNSUBSCRIBED);
|
|
}
|
|
$this->entityManager->flush();
|
|
|
|
$subscribersCount = $this->repository->getSubscribersStatisticsCount($segment);
|
|
expect($subscribersCount[SubscriberEntity::STATUS_SUBSCRIBED])->equals(0);
|
|
expect($subscribersCount[SubscriberEntity::STATUS_UNSUBSCRIBED])->equals(4);
|
|
expect($subscribersCount[SubscriberEntity::STATUS_UNCONFIRMED])->equals(0);
|
|
expect($subscribersCount[SubscriberEntity::STATUS_BOUNCED])->equals(0);
|
|
|
|
// trashed subscribers
|
|
foreach ($subscriberSegments as $subscriberSegment) {
|
|
$subscriberSegment->setStatus(SubscriberEntity::STATUS_SUBSCRIBED);
|
|
}
|
|
foreach ($subscribers as $subscriber) {
|
|
$subscriber->setDeletedAt(new \DateTimeImmutable());
|
|
}
|
|
$this->entityManager->flush();
|
|
|
|
$subscribersCount = $this->repository->getSubscribersStatisticsCount($segment);
|
|
expect($subscribersCount[SubscriberEntity::STATUS_SUBSCRIBED])->equals(0);
|
|
expect($subscribersCount[SubscriberEntity::STATUS_UNSUBSCRIBED])->equals(0);
|
|
expect($subscribersCount[SubscriberEntity::STATUS_UNCONFIRMED])->equals(0);
|
|
expect($subscribersCount[SubscriberEntity::STATUS_BOUNCED])->equals(0);
|
|
}
|
|
|
|
public function testGetSubscribersCountBySegmentIds(): void {
|
|
$segmentOne = $this->segmentRepository->createOrUpdate('Segment' . rand(0, 10000));
|
|
$segmentTwo = $this->segmentRepository->createOrUpdate('Segment' . rand(0, 10000));
|
|
|
|
$subscriberOne = $this->createSubscriberEntity();
|
|
$subscriberTwo = $this->createSubscriberEntity();
|
|
$subscriberThree = $this->createSubscriberEntity();
|
|
|
|
$this->createSubscriberSegmentEntity($segmentOne, $subscriberOne);
|
|
$this->createSubscriberSegmentEntity($segmentOne, $subscriberTwo);
|
|
$this->createSubscriberSegmentEntity($segmentTwo, $subscriberThree);
|
|
$this->entityManager->flush();
|
|
|
|
// two static segments
|
|
$count = $this->repository->getSubscribersCountBySegmentIds([$segmentOne->getId(), $segmentTwo->getId()]);
|
|
expect($count)->equals(3);
|
|
|
|
$dynamicSegmentOne = $this->createDynamicSegmentEntity();
|
|
$this->entityManager->flush();
|
|
|
|
$wpAuthorOne = 'user-role-editor1@example.com';
|
|
$this->tester->deleteWordPressUser($wpAuthorOne);
|
|
$this->tester->createWordPressUser($wpAuthorOne, 'editor');
|
|
|
|
$wpEditorTwo = 'user-role-editor2@example.com';
|
|
$this->tester->deleteWordPressUser($wpEditorTwo);
|
|
$this->tester->createWordPressUser($wpEditorTwo, 'editor');
|
|
|
|
// two static segments and one dynamic segment
|
|
$count = $this->repository->getSubscribersCountBySegmentIds([
|
|
$segmentOne->getId(),
|
|
$segmentTwo->getId(),
|
|
$dynamicSegmentOne->getId(),
|
|
]);
|
|
expect($count)->equals(5);
|
|
|
|
$dynamicSegmentTwo = $this->createDynamicSegmentEntity('author');
|
|
$this->entityManager->flush();
|
|
|
|
$wpAuthorOne = 'user-role-author1@example.com';
|
|
$this->tester->deleteWordPressUser($wpAuthorOne);
|
|
$this->tester->createWordPressUser($wpAuthorOne, 'author');
|
|
|
|
// two dynamic segments
|
|
$count = $this->repository->getSubscribersCountBySegmentIds([
|
|
$dynamicSegmentOne->getId(),
|
|
$dynamicSegmentTwo->getId(),
|
|
]);
|
|
expect($count)->equals(3);
|
|
|
|
// all four segments
|
|
$count = $this->repository->getSubscribersCountBySegmentIds([
|
|
$segmentOne->getId(),
|
|
$segmentTwo->getId(),
|
|
$dynamicSegmentOne->getId(),
|
|
$dynamicSegmentTwo->getId(),
|
|
]);
|
|
expect($count)->equals(6);
|
|
}
|
|
|
|
public function _after() {
|
|
parent::_after();
|
|
$this->cleanup();
|
|
}
|
|
|
|
private function createSubscriberEntity(): SubscriberEntity {
|
|
$subscriber = new SubscriberEntity();
|
|
$rand = rand(0, 100000);
|
|
$subscriber->setEmail("john{$rand}@mailpoet.com");
|
|
$subscriber->setFirstName('John');
|
|
$subscriber->setLastName('Doe');
|
|
$subscriber->setStatus(SubscriberEntity::STATUS_SUBSCRIBED);
|
|
$subscriber->setSource(Source::API);
|
|
$this->entityManager->persist($subscriber);
|
|
return $subscriber;
|
|
}
|
|
|
|
private function createSubscriberSegmentEntity(SegmentEntity $segment, SubscriberEntity $subscriber): SubscriberSegmentEntity {
|
|
$subscriberSegment = new SubscriberSegmentEntity($segment, $subscriber, SubscriberEntity::STATUS_SUBSCRIBED);
|
|
$this->entityManager->persist($subscriberSegment);
|
|
return $subscriberSegment;
|
|
}
|
|
|
|
private function createDynamicSegmentEntity(string $role = 'editor'): SegmentEntity {
|
|
$segment = new SegmentEntity('Segment' . rand(0, 10000), SegmentEntity::TYPE_DYNAMIC, 'Segment description');
|
|
$dynamicFilter = new DynamicSegmentFilterEntity($segment, new DynamicSegmentFilterData([
|
|
'wordpressRole' => $role,
|
|
'segmentType' => DynamicSegmentFilterData::TYPE_USER_ROLE,
|
|
]));
|
|
$segment->getDynamicFilters()->add($dynamicFilter);
|
|
$this->entityManager->persist($segment);
|
|
$this->entityManager->persist($dynamicFilter);
|
|
return $segment;
|
|
}
|
|
|
|
private function cleanup() {
|
|
$this->truncateEntity(SegmentEntity::class);
|
|
$this->truncateEntity(SubscriberEntity::class);
|
|
$this->truncateEntity(SubscriberSegmentEntity::class);
|
|
$this->truncateEntity(DynamicSegmentFilterEntity::class);
|
|
}
|
|
}
|