Test resetSubscriptions method works as expected

[MAILPOET-5555]
This commit is contained in:
David Remer
2023-08-31 08:43:07 +03:00
committed by Aschepikov
parent 33078fb905
commit 6ca5f68be0

View File

@@ -0,0 +1,82 @@
<?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 = array_map(
function(SubscriberSegmentEntity $entity): ?int {
return $entity->getSegment() ? $entity->getSegment()->getId() : null;
},
$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 = array_map(
function(SubscriberSegmentEntity $entity): ?int {
return $entity->getSegment() ? $entity->getSegment()->getId() : null;
},
$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 = array_map(
function(SubscriberSegmentEntity $entity): ?int {
return $entity->getSegment() ? $entity->getSegment()->getId() : null;
},
$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());
}
}