64 lines
2.9 KiB
PHP
64 lines
2.9 KiB
PHP
<?php declare(strict_types = 1);
|
|
|
|
namespace MailPoet\Util\DataInconsistency;
|
|
|
|
use MailPoet\Cron\Workers\SendingQueue\SendingQueue as SendingQueueWorker;
|
|
use MailPoet\Entities\ScheduledTaskEntity;
|
|
use MailPoet\Entities\ScheduledTaskSubscriberEntity;
|
|
use MailPoet\Entities\SubscriberEntity;
|
|
use MailPoet\Test\DataFactories\ScheduledTask;
|
|
use MailPoet\Test\DataFactories\ScheduledTaskSubscriber;
|
|
use MailPoet\Test\DataFactories\SendingQueue;
|
|
use MailPoet\Test\DataFactories\Subscriber;
|
|
|
|
class DataInconsistencyRepositoryTest extends \MailPoetTest {
|
|
private DataInconsistencyRepository $repository;
|
|
|
|
public function _before(): void {
|
|
$this->repository = $this->diContainer->get(DataInconsistencyRepository::class);
|
|
}
|
|
|
|
public function testItFetchesOrphanedSendingTasksCount(): void {
|
|
$orphanedSendingTasksCount = $this->repository->getOrphanedSendingTasksCount();
|
|
verify($orphanedSendingTasksCount)->equals(0);
|
|
|
|
// Add non orphaned sending task
|
|
$okTask = (new ScheduledTask())->create(SendingQueueWorker::TASK_TYPE, ScheduledTaskEntity::STATUS_SCHEDULED);
|
|
(new SendingQueue())->create($okTask);
|
|
$orphanedSendingTasksCount = $this->repository->getOrphanedSendingTasksCount();
|
|
verify($orphanedSendingTasksCount)->equals(0);
|
|
|
|
// Add orphaned sending tasks
|
|
(new ScheduledTask())->create(SendingQueueWorker::TASK_TYPE, ScheduledTaskEntity::STATUS_SCHEDULED);
|
|
(new ScheduledTask())->create(SendingQueueWorker::TASK_TYPE, null);
|
|
$orphanedSendingTasksCount = $this->repository->getOrphanedSendingTasksCount();
|
|
verify($orphanedSendingTasksCount)->equals(2);
|
|
}
|
|
|
|
public function testItCleansUpOrphanedSendingTasks(): void {
|
|
$taskWithSubscriber = (new ScheduledTask())->create(SendingQueueWorker::TASK_TYPE, ScheduledTaskEntity::STATUS_SCHEDULED);
|
|
(new ScheduledTask())->create(SendingQueueWorker::TASK_TYPE, null);
|
|
|
|
$subscriber = (new Subscriber())->create();
|
|
(new ScheduledTaskSubscriber())->createProcessed($taskWithSubscriber, $subscriber);
|
|
|
|
$orphanedSendingTasksCount = $this->repository->getOrphanedSendingTasksCount();
|
|
verify($orphanedSendingTasksCount)->equals(2);
|
|
$taskSubscriberCount = $this->entityManager->getRepository(ScheduledTaskSubscriberEntity::class)->count([]);
|
|
verify($taskSubscriberCount)->equals(1);
|
|
|
|
$this->repository->cleanupOrphanedSendingTasks();
|
|
$orphanedSendingTasksCount = $this->repository->getOrphanedSendingTasksCount();
|
|
verify($orphanedSendingTasksCount)->equals(0);
|
|
|
|
// Check subscriber is not deleted
|
|
$this->entityManager->detach($subscriber);
|
|
$subscriber = $this->entityManager->find(SubscriberEntity::class, $subscriber->getId());
|
|
$this->assertInstanceOf(SubscriberEntity::class, $subscriber);
|
|
|
|
// Check task subscriber is deleted
|
|
$taskSubscriberCount = $this->entityManager->getRepository(ScheduledTaskSubscriberEntity::class)->count([]);
|
|
verify($taskSubscriberCount)->equals(0);
|
|
}
|
|
}
|