73 lines
2.6 KiB
PHP
73 lines
2.6 KiB
PHP
<?php
|
|
namespace MailPoet\Test\Cron\Workers;
|
|
|
|
use Carbon\Carbon;
|
|
use MailPoet\Models\ScheduledTask;
|
|
|
|
require_once('SingleInstanceSimpleWorkerMockImplementation.php');
|
|
use MailPoet\Cron\Workers\SingleInstanceSimpleWorkerMockImplementation as MockSimpleWorker;
|
|
|
|
class SingleInstanceSimpleWorkerTest extends \MailPoetTest {
|
|
function _before() {
|
|
$this->worker = $this->getMockBuilder(MockSimpleWorker::class)
|
|
->setMethods(['processTaskStrategy'])
|
|
->getMock();
|
|
}
|
|
|
|
function testItWillNotRunInMultipleInstances() {
|
|
$this->worker->expects($this->once())
|
|
->method('processTaskStrategy')
|
|
->willReturn(true);
|
|
$task = $this->createScheduledTask();
|
|
expect(empty($task->getMeta()['in_progress']))->equals(true);
|
|
expect($this->worker->processTask($task))->equals(true);
|
|
$this->worker->startProgress($task);
|
|
expect(empty($task->getMeta()['in_progress']))->equals(false);
|
|
expect($this->worker->processTask($task))->equals(false);
|
|
}
|
|
|
|
function testItWillResetTheInProgressFlagOnFail() {
|
|
$task = $this->createScheduledTask();
|
|
$this->worker->expects($this->once())
|
|
->method('processTaskStrategy')
|
|
->willThrowException(new \Exception('test error'));
|
|
try {
|
|
$this->worker->processTask($task);
|
|
$this->fail('An exception should be thrown');
|
|
} catch (\Exception $e) {
|
|
expect($e->getMessage())->equals('test error');
|
|
expect(empty($task->getMeta()['in_progress']))->equals(true);
|
|
}
|
|
}
|
|
|
|
function testItWillRescheduleTaskIfItIsRunningForTooLong() {
|
|
$this->worker->expects($this->once())
|
|
->method('processTaskStrategy')
|
|
->willReturn(true);
|
|
$task = $this->createScheduledTask();
|
|
$task = ScheduledTask::findOne($task->id); // make sure `updated_at` is set by the DB
|
|
expect($this->worker->processTask($task))->equals(true);
|
|
$scheduled_at = $task->scheduled_at;
|
|
$task->updated_at = Carbon::createFromTimestamp(strtotime($task->updated_at))
|
|
->subMinutes(MockSimpleWorker::TASK_RUN_TIMEOUT + 1);
|
|
expect($this->worker->processTask($task))->equals(false);
|
|
$task = ScheduledTask::findOne($task->id);
|
|
expect($scheduled_at < $task->scheduled_at)->true();
|
|
expect($task->status)->equals(ScheduledTask::STATUS_SCHEDULED);
|
|
expect(empty($task->getMeta()['in_progress']))->equals(true);
|
|
}
|
|
|
|
private function createScheduledTask() {
|
|
$task = ScheduledTask::create();
|
|
$task->type = MockSimpleWorker::TASK_TYPE;
|
|
$task->status = null;
|
|
$task->scheduled_at = Carbon::createFromTimestamp(current_time('timestamp'));
|
|
$task->save();
|
|
return $task;
|
|
}
|
|
|
|
function _after() {
|
|
\ORM::raw_execute('TRUNCATE ' . ScheduledTask::$_table);
|
|
}
|
|
}
|