76 lines
2.8 KiB
PHP
76 lines
2.8 KiB
PHP
<?php
|
|
namespace MailPoet\Test\Models;
|
|
|
|
use MailPoet\Models\ScheduledTaskSubscriber;
|
|
|
|
class ScheduledTaskSubscriberTest extends \MailPoetTest {
|
|
function _before() {
|
|
$this->task_id = 123;
|
|
$this->subscriber_id = 456;
|
|
$this->task_subscriber = ScheduledTaskSubscriber::createOrUpdate(array(
|
|
'task_id' => $this->task_id,
|
|
'subscriber_id' => $this->subscriber_id
|
|
));
|
|
}
|
|
|
|
function testItCanBeCreated() {
|
|
expect($this->task_subscriber->task_id)->equals($this->task_id);
|
|
expect($this->task_subscriber->subscriber_id)->equals($this->subscriber_id);
|
|
expect($this->task_subscriber->processed)->equals(ScheduledTaskSubscriber::STATUS_UNPROCESSED);
|
|
}
|
|
|
|
function testItCanBeUpdated() {
|
|
$task_subscriber = ScheduledTaskSubscriber::createOrUpdate(array(
|
|
'task_id' => $this->task_id,
|
|
'subscriber_id' => $this->subscriber_id,
|
|
'processed' => ScheduledTaskSubscriber::STATUS_PROCESSED
|
|
));
|
|
expect($task_subscriber->processed)->equals(ScheduledTaskSubscriber::STATUS_PROCESSED);
|
|
}
|
|
|
|
function testItCanAddMultipleSubscribers() {
|
|
ScheduledTaskSubscriber::deleteMany();
|
|
$subscriber_ids = array(321, 654, 987);
|
|
ScheduledTaskSubscriber::addSubscribers($this->task_id, $subscriber_ids);
|
|
$task_subscribers = ScheduledTaskSubscriber::where('task_id', $this->task_id)
|
|
->orderByAsc('subscriber_id')
|
|
->findMany();
|
|
expect(count($task_subscribers))->equals(count($subscriber_ids));
|
|
expect($task_subscribers[0]->subscriber_id)->equals($subscriber_ids[0]);
|
|
expect($task_subscribers[1]->subscriber_id)->equals($subscriber_ids[1]);
|
|
expect($task_subscribers[2]->subscriber_id)->equals($subscriber_ids[2]);
|
|
}
|
|
|
|
function testItCangetUnprocessedCount() {
|
|
$count = ScheduledTaskSubscriber::getUnprocessedCount($this->task_id);
|
|
expect($count)->equals(1);
|
|
$this->task_subscriber->processed = ScheduledTaskSubscriber::STATUS_PROCESSED;
|
|
$this->task_subscriber->save();
|
|
$count = ScheduledTaskSubscriber::getUnprocessedCount($this->task_id);
|
|
expect($count)->equals(0);
|
|
}
|
|
|
|
function testItCanGetProcessedCount() {
|
|
$count = ScheduledTaskSubscriber::getProcessedCount($this->task_id);
|
|
expect($count)->equals(0);
|
|
$this->task_subscriber->processed = ScheduledTaskSubscriber::STATUS_PROCESSED;
|
|
$this->task_subscriber->save();
|
|
$count = ScheduledTaskSubscriber::getProcessedCount($this->task_id);
|
|
expect($count)->equals(1);
|
|
}
|
|
|
|
function testItCanGetTotalCount() {
|
|
ScheduledTaskSubscriber::createOrUpdate(array(
|
|
'task_id' => $this->task_id,
|
|
'subscriber_id' => 555,
|
|
'processed' => ScheduledTaskSubscriber::STATUS_PROCESSED
|
|
));
|
|
$count = ScheduledTaskSubscriber::getTotalCount($this->task_id);
|
|
expect($count)->equals(2);
|
|
}
|
|
|
|
function _after() {
|
|
\ORM::raw_execute('TRUNCATE ' . ScheduledTaskSubscriber::$_table);
|
|
}
|
|
}
|