118 lines
4.0 KiB
PHP
118 lines
4.0 KiB
PHP
<?php
|
|
namespace MailPoet\Test\Models;
|
|
|
|
use Carbon\Carbon;
|
|
use MailPoet\Models\Newsletter;
|
|
use MailPoet\Models\ScheduledTask;
|
|
use MailPoet\Models\SendingQueue;
|
|
use MailPoet\Models\ScheduledTaskSubscriber;
|
|
|
|
class ScheduledTaskTest extends \MailPoetTest {
|
|
function _before() {
|
|
parent::_before();
|
|
$this->task = ScheduledTask::create();
|
|
$this->task->hydrate(array(
|
|
'status' => ScheduledTask::STATUS_SCHEDULED
|
|
));
|
|
$this->task->save();
|
|
}
|
|
|
|
function testItCanBeCompleted() {
|
|
$this->task->complete();
|
|
expect($this->task->status)->equals(ScheduledTask::STATUS_COMPLETED);
|
|
}
|
|
|
|
function testItSetsDefaultPriority() {
|
|
expect($this->task->priority)->equals(ScheduledTask::PRIORITY_MEDIUM);
|
|
}
|
|
|
|
function testItUnPauseAllByNewsletters() {
|
|
$newsletter = Newsletter::createOrUpdate(array(
|
|
'type' => Newsletter::TYPE_NOTIFICATION
|
|
));
|
|
$task1 = ScheduledTask::createOrUpdate(array(
|
|
'status' => ScheduledTask::STATUS_PAUSED,
|
|
'scheduled_at' => Carbon::createFromTimestamp(current_time('timestamp'))->addDays(10)->format('Y-m-d H:i:s'),
|
|
));
|
|
$task2 = ScheduledTask::createOrUpdate(array(
|
|
'status' => ScheduledTask::STATUS_COMPLETED,
|
|
'scheduled_at' => Carbon::createFromTimestamp(current_time('timestamp'))->addDays(10)->format('Y-m-d H:i:s'),
|
|
));
|
|
$task3 = ScheduledTask::createOrUpdate(array(
|
|
'status' => ScheduledTask::STATUS_PAUSED,
|
|
'scheduled_at' => Carbon::createFromTimestamp(current_time('timestamp'))->subDays(10)->format('Y-m-d H:i:s'),
|
|
));
|
|
SendingQueue::createOrUpdate(array(
|
|
'newsletter_id' => $newsletter->id(),
|
|
'task_id' => $task1->id(),
|
|
));
|
|
SendingQueue::createOrUpdate(array(
|
|
'newsletter_id' => $newsletter->id(),
|
|
'task_id' => $task2->id(),
|
|
));
|
|
SendingQueue::createOrUpdate(array(
|
|
'newsletter_id' => $newsletter->id(),
|
|
'task_id' => $task3->id(),
|
|
));
|
|
ScheduledTask::setScheduledAllByNewsletter($newsletter);
|
|
$task1_found = ScheduledTask::findOne($task1->id());
|
|
expect($task1_found->status)->equals(ScheduledTask::STATUS_SCHEDULED);
|
|
$task2_found = ScheduledTask::findOne($task2->id());
|
|
expect($task2_found->status)->equals(ScheduledTask::STATUS_COMPLETED);
|
|
$task3_found = ScheduledTask::findOne($task3->id());
|
|
expect($task3_found->status)->equals(ScheduledTask::STATUS_PAUSED);
|
|
}
|
|
|
|
function testItPauseAllByNewsletters() {
|
|
$newsletter = Newsletter::createOrUpdate(array(
|
|
'type' => Newsletter::TYPE_NOTIFICATION
|
|
));
|
|
$task1 = ScheduledTask::createOrUpdate(array(
|
|
'status' => ScheduledTask::STATUS_COMPLETED,
|
|
));
|
|
$task2 = ScheduledTask::createOrUpdate(array(
|
|
'status' => ScheduledTask::STATUS_SCHEDULED,
|
|
));
|
|
SendingQueue::createOrUpdate(array(
|
|
'newsletter_id' => $newsletter->id(),
|
|
'task_id' => $task1->id(),
|
|
));
|
|
SendingQueue::createOrUpdate(array(
|
|
'newsletter_id' => $newsletter->id(),
|
|
'task_id' => $task2->id(),
|
|
));
|
|
ScheduledTask::pauseAllByNewsletter($newsletter);
|
|
$task1_found = ScheduledTask::findOne($task1->id());
|
|
expect($task1_found->status)->equals(ScheduledTask::STATUS_COMPLETED);
|
|
$task2_found = ScheduledTask::findOne($task2->id());
|
|
expect($task2_found->status)->equals(ScheduledTask::STATUS_PAUSED);
|
|
}
|
|
|
|
function testItDeletesRelatedScheduledTaskSubscriber() {
|
|
$task_id = $this->task->id;
|
|
ScheduledTaskSubscriber::createOrUpdate([
|
|
'task_id' => $task_id,
|
|
'subscriber_id' => 1
|
|
]);
|
|
ScheduledTaskSubscriber::createOrUpdate([
|
|
'task_id' => $task_id,
|
|
'subscriber_id' => 2
|
|
]);
|
|
ScheduledTaskSubscriber::createOrUpdate([
|
|
'task_id' => $task_id,
|
|
'subscriber_id' => 3
|
|
]);
|
|
$count = ScheduledTaskSubscriber::where('task_id', $task_id)->count();
|
|
expect($count)->equals(3);
|
|
|
|
$this->task->delete();
|
|
$count = ScheduledTaskSubscriber::where('task_id', $task_id)->count();
|
|
expect($count)->equals(0);
|
|
}
|
|
|
|
function _after() {
|
|
\ORM::raw_execute('TRUNCATE ' . ScheduledTask::$_table);
|
|
\ORM::raw_execute('TRUNCATE ' . ScheduledTaskSubscriber::$_table);
|
|
}
|
|
}
|