72 lines
2.4 KiB
PHP
72 lines
2.4 KiB
PHP
<?php
|
|
|
|
namespace MailPoet\DynamicSegments\Persistence\Loading;
|
|
|
|
use MailPoet\DynamicSegments\Filters\UserRole;
|
|
use MailPoet\DynamicSegments\RequirementsChecker;
|
|
use MailPoet\Models\DynamicSegment;
|
|
use MailPoet\Models\Subscriber;
|
|
use PHPUnit\Framework\MockObject\MockObject;
|
|
|
|
class SubscribersCountTest extends \MailPoetTest {
|
|
|
|
/** @var RequirementsChecker|MockObject */
|
|
private $requirementChecker;
|
|
|
|
public function _before() {
|
|
$this->cleanData();
|
|
$this->tester->createWordPressUser('user-role-test1@example.com', 'editor');
|
|
$this->tester->createWordPressUser('user-role-test2@example.com', 'administrator');
|
|
$this->tester->createWordPressUser('user-role-test3@example.com', 'editor');
|
|
$this->requirementChecker = $this
|
|
->getMockBuilder(RequirementsChecker::class)
|
|
->setMethods(['shouldSkipSegment'])
|
|
->getMock();
|
|
}
|
|
|
|
public function testItConstructsQuery() {
|
|
$this->requirementChecker->method('shouldSkipSegment')->willReturn(false);
|
|
$userRole = DynamicSegment::create();
|
|
$subscriber1 = Subscriber::findOne('user-role-test1@example.com');
|
|
$subscriber1->status = Subscriber::STATUS_SUBSCRIBED;
|
|
$subscriber1->save();
|
|
$subscriber3 = Subscriber::findOne('user-role-test3@example.com');
|
|
$subscriber3->status = Subscriber::STATUS_SUBSCRIBED;
|
|
$subscriber3->save();
|
|
$userRole->hydrate([
|
|
'name' => 'segment',
|
|
'description' => 'description',
|
|
]);
|
|
$userRole->setFilters([new UserRole('editor', 'and')]);
|
|
|
|
$loader = new SubscribersCount($this->requirementChecker);
|
|
$count = $loader->getSubscribersCount($userRole);
|
|
expect($count)->equals(2);
|
|
}
|
|
|
|
public function testItSkipsIfRequirementNotMet() {
|
|
$this->requirementChecker->method('shouldSkipSegment')->willReturn(true);
|
|
$userRole = DynamicSegment::create();
|
|
$userRole->hydrate([
|
|
'name' => 'segment',
|
|
'description' => 'description',
|
|
]);
|
|
$userRole->setFilters([new UserRole('editor', 'and')]);
|
|
|
|
$loader = new SubscribersCount($this->requirementChecker);
|
|
$count = $loader->getSubscribersCount($userRole);
|
|
expect($count)->equals(0);
|
|
}
|
|
|
|
public function _after() {
|
|
$this->cleanData();
|
|
}
|
|
|
|
private function cleanData() {
|
|
$emails = ['user-role-test1@example.com', 'user-role-test2@example.com', 'user-role-test3@example.com'];
|
|
foreach ($emails as $email) {
|
|
$this->tester->deleteWordPressUser($email);
|
|
}
|
|
}
|
|
}
|