82 lines
3.8 KiB
PHP
82 lines
3.8 KiB
PHP
<?php declare(strict_types = 1);
|
|
|
|
namespace MailPoet\Subscribers;
|
|
|
|
use MailPoet\Entities\SubscriberEntity;
|
|
use MailPoet\Entities\SubscriberSegmentEntity;
|
|
use MailPoet\Test\DataFactories\Segment;
|
|
use MailPoet\Test\DataFactories\Subscriber;
|
|
|
|
class SubscriberSegmentRepositoryTest extends \MailPoetTest {
|
|
|
|
/** @var SubscriberSegmentRepository */
|
|
private $testee;
|
|
|
|
public function _before() {
|
|
$this->testee = $this->diContainer->get(SubscriberSegmentRepository::class);
|
|
}
|
|
|
|
public function testResetIsWorking() {
|
|
$subscriber = (new Subscriber())->create();
|
|
$segment1 = (new Segment())->create();
|
|
$segment2 = (new Segment())->create();
|
|
$segment3 = (new Segment())->create();
|
|
$segment4 = (new Segment())->create();
|
|
|
|
$this->testee->resetSubscriptions($subscriber, [$segment1, $segment2, $segment3, $segment4]);
|
|
$subscribedSegments = $subscriber->getSubscriberSegments(SubscriberEntity::STATUS_SUBSCRIBED);
|
|
$unsubscribedSegments = $subscriber->getSubscriberSegments(SubscriberEntity::STATUS_UNSUBSCRIBED);
|
|
$subscribedSegmentIds = $this->getSubscribedSegmentIds($subscribedSegments->toArray());
|
|
|
|
$this->assertEquals(0, $unsubscribedSegments->count());
|
|
$this->assertCount(4, $subscribedSegmentIds);
|
|
$this->assertContains($segment1->getId(), $subscribedSegmentIds);
|
|
$this->assertContains($segment2->getId(), $subscribedSegmentIds);
|
|
$this->assertContains($segment3->getId(), $subscribedSegmentIds);
|
|
$this->assertContains($segment4->getId(), $subscribedSegmentIds);
|
|
|
|
$this->testee->resetSubscriptions($subscriber, [$segment2, $segment3, $segment4]);
|
|
$subscribedSegments = $subscriber->getSubscriberSegments(SubscriberEntity::STATUS_SUBSCRIBED);
|
|
$unsubscribedSegments = $subscriber->getSubscriberSegments(SubscriberEntity::STATUS_UNSUBSCRIBED);
|
|
$subscribedSegmentIds = $this->getSubscribedSegmentIds($subscribedSegments->toArray());
|
|
|
|
$this->assertEquals(1, $unsubscribedSegments->count());
|
|
$this->assertCount(3, $subscribedSegmentIds);
|
|
$this->assertNotContains($segment1->getId(), $subscribedSegmentIds);
|
|
$this->assertContains($segment2->getId(), $subscribedSegmentIds);
|
|
$this->assertContains($segment3->getId(), $subscribedSegmentIds);
|
|
$this->assertContains($segment4->getId(), $subscribedSegmentIds);
|
|
|
|
$this->testee->resetSubscriptions($subscriber, [$segment1, $segment2, $segment3, $segment4]);
|
|
$subscribedSegments = $subscriber->getSubscriberSegments(SubscriberEntity::STATUS_SUBSCRIBED);
|
|
$unsubscribedSegments = $subscriber->getSubscriberSegments(SubscriberEntity::STATUS_UNSUBSCRIBED);
|
|
$subscribedSegmentIds = $this->getSubscribedSegmentIds($subscribedSegments->toArray());
|
|
|
|
$this->assertEquals(0, $unsubscribedSegments->count());
|
|
$this->assertCount(4, $subscribedSegmentIds);
|
|
$this->assertContains($segment1->getId(), $subscribedSegmentIds);
|
|
$this->assertContains($segment2->getId(), $subscribedSegmentIds);
|
|
$this->assertContains($segment3->getId(), $subscribedSegmentIds);
|
|
$this->assertContains($segment4->getId(), $subscribedSegmentIds);
|
|
|
|
$this->testee->resetSubscriptions($subscriber, []);
|
|
$subscribedSegments = $subscriber->getSubscriberSegments(SubscriberEntity::STATUS_SUBSCRIBED);
|
|
$unsubscribedSegments = $subscriber->getSubscriberSegments(SubscriberEntity::STATUS_UNSUBSCRIBED);
|
|
$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
|
|
)));
|
|
}
|
|
}
|