106 lines
3.3 KiB
PHP
106 lines
3.3 KiB
PHP
<?php
|
|
|
|
namespace MailPoet\Segments;
|
|
|
|
use Codeception\Util\Stub;
|
|
use MailPoet\Models\Segment;
|
|
use MailPoet\Models\Subscriber;
|
|
use MailPoet\Models\SubscriberSegment;
|
|
use MailPoet\WP\Hooks;
|
|
|
|
require_once('SubscribersBulkActionHandlerMock.php');
|
|
|
|
class BulkActionTest extends \MailPoetTest {
|
|
|
|
function _before() {
|
|
$this->cleanData();
|
|
$this->segment_1 = Segment::createOrUpdate(array('name' => 'Segment 1', 'type' => 'default'));
|
|
$this->segment_2 = Segment::createOrUpdate(array('name' => 'Segment 3', 'type' => 'not default'));
|
|
$this->subscriber_1 = Subscriber::createOrUpdate(array(
|
|
'email' => 'john@mailpoet.com',
|
|
'first_name' => 'John',
|
|
'last_name' => 'Doe',
|
|
'status' => Subscriber::STATUS_SUBSCRIBED,
|
|
'segments' => array(
|
|
$this->segment_1->id,
|
|
),
|
|
));
|
|
$this->subscriber_2 = Subscriber::createOrUpdate(array(
|
|
'email' => 'jake@mailpoet.com',
|
|
'first_name' => 'Jake',
|
|
'last_name' => 'Doe',
|
|
'status' => Subscriber::STATUS_SUBSCRIBED,
|
|
'segments' => array(
|
|
$this->segment_2->id,
|
|
),
|
|
));
|
|
SubscriberSegment::resubscribeToAllSegments($this->subscriber_1);
|
|
SubscriberSegment::resubscribeToAllSegments($this->subscriber_2);
|
|
}
|
|
|
|
function _after() {
|
|
$this->cleanData();
|
|
}
|
|
|
|
private function cleanData() {
|
|
\ORM::raw_execute('TRUNCATE ' . Segment::$_table);
|
|
\ORM::raw_execute('TRUNCATE ' . SubscriberSegment::$_table);
|
|
\ORM::raw_execute('TRUNCATE ' . Subscriber::$_table);
|
|
}
|
|
|
|
function testBulkActionWithoutSegment() {
|
|
$handler = new BulkAction(array());
|
|
$this->setExpectedException('InvalidArgumentException');
|
|
$handler->apply();
|
|
}
|
|
|
|
function testBulkActionForDefaultSegment() {
|
|
$handler = new BulkAction(array(
|
|
'listing' => array('filter'=> array('segment' => $this->segment_1->id)),
|
|
'action' => 'trash',
|
|
));
|
|
$result = $handler->apply();
|
|
expect($result['count'])->equals(1);
|
|
}
|
|
|
|
function testBulkActionForUnknownSegment() {
|
|
$handler = new BulkAction(array(
|
|
'listing' => array('filter'=> array('segment' => 'this-segment-doesnt-exist')),
|
|
'action' => 'trash',
|
|
));
|
|
$result = $handler->apply();
|
|
expect($result)->notEmpty();
|
|
}
|
|
|
|
function testForUnknownSegmentTypeWithoutHandler() {
|
|
$handler = new BulkAction(array(
|
|
'listing' => array('filter'=> array('segment' => $this->segment_2->id)),
|
|
'action' => 'trash',
|
|
));
|
|
$this->setExpectedException('InvalidArgumentException');
|
|
remove_all_filters('mailpoet_subscribers_in_segment_apply_bulk_action_handlers');
|
|
$handler->apply();
|
|
}
|
|
|
|
function testBulkActionUsingFilter() {
|
|
$mock = Stub::makeEmpty('\MailPoet\Test\Segments\SubscribersBulkActionHandlerMock', array('apply'));
|
|
$mock
|
|
->expects($this->once())
|
|
->method('apply')
|
|
->will($this->returnValue('result'));
|
|
|
|
remove_all_filters('mailpoet_subscribers_in_segment_apply_bulk_action_handlers');
|
|
Hooks::addFilter('mailpoet_subscribers_in_segment_apply_bulk_action_handlers', function () use ($mock) {
|
|
return array($mock);
|
|
});
|
|
|
|
$handler = new BulkAction(array(
|
|
'listing' => array('filter'=> array('segment' => $this->segment_2->id)),
|
|
'action' => 'trash',
|
|
));
|
|
$result = $handler->apply();
|
|
expect($result)->equals('result');
|
|
}
|
|
|
|
}
|