91 lines
3.8 KiB
PHP
91 lines
3.8 KiB
PHP
<?php
|
|
|
|
namespace MailPoet\Test\API\JSON\v1;
|
|
|
|
use MailPoet\API\JSON\v1\ImportExport;
|
|
use MailPoet\Cron\Workers\WooCommerceSync;
|
|
use MailPoet\DI\ContainerWrapper;
|
|
use MailPoet\Entities\ScheduledTaskEntity;
|
|
use MailPoet\Newsletter\Sending\ScheduledTasksRepository;
|
|
use MailPoet\WP\Functions as WPFunctions;
|
|
use MailPoetVendor\Carbon\Carbon;
|
|
|
|
class ImportExportTest extends \MailPoetTest {
|
|
|
|
/** @var ImportExport */
|
|
private $endpoint;
|
|
|
|
/** @var ScheduledTasksRepository */
|
|
private $scheduledTasksRepository;
|
|
|
|
public function _before() {
|
|
parent::_before();
|
|
$this->endpoint = ContainerWrapper::getInstance()->get(ImportExport::class);
|
|
$this->scheduledTasksRepository = ContainerWrapper::getInstance()->get(ScheduledTasksRepository::class);
|
|
$this->entityManager->createQueryBuilder()
|
|
->delete(ScheduledTaskEntity::class, 's')
|
|
->where('s.type = :type')
|
|
->setParameter(':type', WooCommerceSync::TASK_TYPE )
|
|
->getQuery()
|
|
->execute();
|
|
}
|
|
|
|
public function testItSchedulesTaskWhenNoneExists() {
|
|
$response = $this->endpoint->setupWooCommerceInitialImport();
|
|
expect($response->status)->equals(200);
|
|
$task = $this->scheduledTasksRepository->findOneBy(['type' => WooCommerceSync::TASK_TYPE]);
|
|
assert($task instanceof ScheduledTaskEntity);
|
|
expect($task->getStatus())->equals(ScheduledTaskEntity::STATUS_SCHEDULED);
|
|
$now = time();
|
|
$scheduledAt = new Carbon($task->getScheduledAt());
|
|
expect($scheduledAt->timestamp)->greaterOrEquals($now - 1);
|
|
expect($scheduledAt->timestamp)->lessOrEquals($now + 1);
|
|
}
|
|
|
|
public function testItReschedulesScheduledTaskToNow() {
|
|
$originalSchedule = Carbon::createFromTimestamp((int)current_time('timestamp') + 3000);
|
|
$this->createTask(WooCommerceSync::TASK_TYPE, ScheduledTaskEntity::STATUS_SCHEDULED, $originalSchedule);
|
|
$this->entityManager->clear();
|
|
$this->endpoint->setupWooCommerceInitialImport();
|
|
$task = $this->scheduledTasksRepository->findOneBy(['type' => WooCommerceSync::TASK_TYPE]);
|
|
assert($task instanceof ScheduledTaskEntity);
|
|
expect($task->getStatus())->equals(ScheduledTaskEntity::STATUS_SCHEDULED);
|
|
$now = time();
|
|
$scheduledAt = new Carbon($task->getScheduledAt());
|
|
expect($scheduledAt->timestamp)->greaterOrEquals($now - 1);
|
|
expect($scheduledAt->timestamp)->lessOrEquals($now + 1);
|
|
$taskCount = $this->scheduledTasksRepository->countBy(['type' => WooCommerceSync::TASK_TYPE]);
|
|
expect($taskCount)->equals(1);
|
|
}
|
|
|
|
public function testItDoesNothingForRunningTask() {
|
|
$this->createTask(WooCommerceSync::TASK_TYPE, null);
|
|
$this->endpoint->setupWooCommerceInitialImport();
|
|
$task = $this->scheduledTasksRepository->findOneBy(['type' => WooCommerceSync::TASK_TYPE]);
|
|
assert($task instanceof ScheduledTaskEntity);
|
|
expect($task->getStatus())->equals(null);
|
|
$taskCount = $this->scheduledTasksRepository->countBy(['type' => WooCommerceSync::TASK_TYPE]);
|
|
expect($taskCount)->equals(1);
|
|
}
|
|
|
|
public function testItIgnoresCompletedAndPausedTasks() {
|
|
$this->createTask(WooCommerceSync::TASK_TYPE, ScheduledTaskEntity::STATUS_PAUSED);
|
|
$this->createTask(WooCommerceSync::TASK_TYPE, ScheduledTaskEntity::STATUS_COMPLETED);
|
|
$this->endpoint->setupWooCommerceInitialImport();
|
|
$taskCount = $this->scheduledTasksRepository->countBy(['type' => WooCommerceSync::TASK_TYPE]);
|
|
expect($taskCount)->equals(3);
|
|
}
|
|
|
|
private function createTask($type, $status = null, $scheduledAt = null) {
|
|
if (!$scheduledAt) {
|
|
Carbon::createFromTimestamp(WPFunctions::get()->currentTime('timestamp'));
|
|
}
|
|
$task = new ScheduledTaskEntity();
|
|
$task->setType($type);
|
|
$task->setStatus($status);
|
|
$task->setScheduledAt($scheduledAt);
|
|
$this->scheduledTasksRepository->persist($task);
|
|
$this->scheduledTasksRepository->flush();
|
|
}
|
|
}
|