Add method getSubscribedSegmentIds

[MAILPOET-5555]
This commit is contained in:
David Remer
2023-09-05 09:43:23 +03:00
committed by Aschepikov
parent 974559876b
commit e53cfa95e8

View File

@@ -26,12 +26,7 @@ class SubscriberSegmentRepositoryTest extends \MailPoetTest {
$this->testee->resetSubscriptions($subscriber, [$segment1, $segment2, $segment3, $segment4]);
$subscribedSegments = $subscriber->getSubscriberSegments(SubscriberEntity::STATUS_SUBSCRIBED);
$unsubscribedSegments = $subscriber->getSubscriberSegments(SubscriberEntity::STATUS_UNSUBSCRIBED);
$subscribedSegmentIds = array_map(
function(SubscriberSegmentEntity $entity): ?int {
return $entity->getSegment() ? $entity->getSegment()->getId() : null;
},
$subscribedSegments->toArray()
);
$subscribedSegmentIds = $this->getSubscribedSegmentIds($subscribedSegments->toArray());
$this->assertEquals(0, $unsubscribedSegments->count());
$this->assertCount(4, $subscribedSegmentIds);
@@ -43,12 +38,7 @@ class SubscriberSegmentRepositoryTest extends \MailPoetTest {
$this->testee->resetSubscriptions($subscriber, [$segment2, $segment3, $segment4]);
$subscribedSegments = $subscriber->getSubscriberSegments(SubscriberEntity::STATUS_SUBSCRIBED);
$unsubscribedSegments = $subscriber->getSubscriberSegments(SubscriberEntity::STATUS_UNSUBSCRIBED);
$subscribedSegmentIds = array_map(
function(SubscriberSegmentEntity $entity): ?int {
return $entity->getSegment() ? $entity->getSegment()->getId() : null;
},
$subscribedSegments->toArray()
);
$subscribedSegmentIds = $this->getSubscribedSegmentIds($subscribedSegments->toArray());
$this->assertEquals(1, $unsubscribedSegments->count());
$this->assertCount(3, $subscribedSegmentIds);
@@ -60,12 +50,8 @@ class SubscriberSegmentRepositoryTest extends \MailPoetTest {
$this->testee->resetSubscriptions($subscriber, [$segment1, $segment2, $segment3, $segment4]);
$subscribedSegments = $subscriber->getSubscriberSegments(SubscriberEntity::STATUS_SUBSCRIBED);
$unsubscribedSegments = $subscriber->getSubscriberSegments(SubscriberEntity::STATUS_UNSUBSCRIBED);
$subscribedSegmentIds = array_map(
function(SubscriberSegmentEntity $entity): ?int {
return $entity->getSegment() ? $entity->getSegment()->getId() : null;
},
$subscribedSegments->toArray()
);
$subscribedSegmentIds = $this->getSubscribedSegmentIds($subscribedSegments->toArray());
$this->assertEquals(0, $unsubscribedSegments->count());
$this->assertCount(4, $subscribedSegmentIds);
$this->assertContains($segment1->getId(), $subscribedSegmentIds);
@@ -79,4 +65,17 @@ class SubscriberSegmentRepositoryTest extends \MailPoetTest {
$this->assertEquals(4, $unsubscribedSegments->count());
$this->assertEquals(0, $subscribedSegments->count());
}
/**
* @param SubscriberSegmentEntity[] $subscribedSegments
* @return int[]
*/
private function getSubscribedSegmentIds(array $subscribedSegments): array {
return array_values(array_filter(array_map(
function(SubscriberSegmentEntity $entity): ?int {
return $entity->getSegment() ? $entity->getSegment()->getId() : null;
},
$subscribedSegments
)));
}
}