Remove old model from Scheduler

[MAILPOET-4141]
This commit is contained in:
Jan Lysý
2022-05-25 09:05:29 +02:00
committed by Veljko V
parent 882e096c81
commit 4ba5c6ce0c
9 changed files with 88 additions and 49 deletions

View File

@ -165,11 +165,11 @@ class AbandonedCart {
} }
$meta = [self::TASK_META_NAME => $cartProductIds]; $meta = [self::TASK_META_NAME => $cartProductIds];
$this->scheduler->scheduleOrRescheduleAutomaticEmail(WooCommerceEmail::SLUG, self::SLUG, $subscriber->getId(), $meta); $this->scheduler->scheduleOrRescheduleAutomaticEmail(WooCommerceEmail::SLUG, self::SLUG, (int)$subscriber->getId(), $meta);
} }
private function rescheduleAbandonedCartEmail(SubscriberEntity $subscriberEntity) { private function rescheduleAbandonedCartEmail(SubscriberEntity $subscriberEntity) {
$this->scheduler->rescheduleAutomaticEmail(WooCommerceEmail::SLUG, self::SLUG, $subscriberEntity->getId()); $this->scheduler->rescheduleAutomaticEmail(WooCommerceEmail::SLUG, self::SLUG, (int)$subscriberEntity->getId());
} }
private function cancelAbandonedCartEmail() { private function cancelAbandonedCartEmail() {
@ -177,7 +177,7 @@ class AbandonedCart {
if (!$subscriber) { if (!$subscriber) {
return; return;
} }
$this->scheduler->cancelAutomaticEmail(WooCommerceEmail::SLUG, self::SLUG, $subscriber->getId()); $this->scheduler->cancelAutomaticEmail(WooCommerceEmail::SLUG, self::SLUG, (int)$subscriber->getId());
} }
private function getSubscriber(): ?SubscriberEntity { private function getSubscriber(): ?SubscriberEntity {

View File

@ -65,6 +65,30 @@ class NewslettersRepository extends Repository {
->getResult(); ->getResult();
} }
/**
* @return NewsletterEntity[]
*/
public function findActiveByTypeAndGroup(string $type, ?string $group): array {
$qb = $this->entityManager
->createQueryBuilder()
->select('n')
->from(NewsletterEntity::class, 'n')
->where('n.status = :status')
->setParameter(':status', NewsletterEntity::STATUS_ACTIVE)
->andWhere('n.deletedAt IS NULL')
->andWhere('n.type = :type')
->setParameter('type', $type);
if ($group) {
$qb->join('n.options', 'o', Join::WITH, 'o.value = :group')
->join('o.optionField', 'f', Join::WITH, 'f.name = :nameGroup AND f.newsletterType = :type')
->setParameter('nameGroup', NewsletterOptionFieldEntity::NAME_GROUP)
->setParameter('group', $group);
}
return $qb->getQuery()->getResult();
}
/** /**
* @param string[] $types * @param string[] $types
* @return NewsletterEntity[] * @return NewsletterEntity[]

View File

@ -2,7 +2,8 @@
namespace MailPoet\Newsletter\Scheduler; namespace MailPoet\Newsletter\Scheduler;
use MailPoet\Models\Newsletter; use MailPoet\Entities\NewsletterEntity;
use MailPoet\Newsletter\NewslettersRepository;
use MailPoet\WP\Functions as WPFunctions; use MailPoet\WP\Functions as WPFunctions;
use MailPoetVendor\Carbon\Carbon; use MailPoetVendor\Carbon\Carbon;
@ -12,10 +13,15 @@ class Scheduler {
/** @var WPFunctions */ /** @var WPFunctions */
private $wp; private $wp;
/** @var NewslettersRepository */
private $newslettersRepository;
public function __construct( public function __construct(
WPFunctions $wp WPFunctions $wp,
NewslettersRepository $newslettersRepository
) { ) {
$this->wp = $wp; $this->wp = $wp;
$this->newslettersRepository = $newslettersRepository;
} }
public function getNextRunDate($schedule, $fromTimestamp = false) { public function getNextRunDate($schedule, $fromTimestamp = false) {
@ -65,12 +71,11 @@ class Scheduler {
return $currentTime; return $currentTime;
} }
public function getNewsletters($type, $group = false) { /**
return Newsletter::getPublished() * @return NewsletterEntity[]
->filter('filterType', $type, $group) */
->filter('filterStatus', Newsletter::STATUS_ACTIVE) public function getNewsletters(string $type, ?string $group = null): array {
->filter('filterWithOptions', $type) return $this->newslettersRepository->findActiveByTypeAndGroup($type, $group);
->findMany();
} }
public function formatDatetimeString($datetimeString) { public function formatDatetimeString($datetimeString) {

View File

@ -10,6 +10,7 @@ use MailPoet\Entities\ScheduledTaskEntity;
use MailPoet\Entities\ScheduledTaskSubscriberEntity; use MailPoet\Entities\ScheduledTaskSubscriberEntity;
use MailPoet\Entities\SendingQueueEntity; use MailPoet\Entities\SendingQueueEntity;
use MailPoet\Entities\SubscriberEntity; use MailPoet\Entities\SubscriberEntity;
use MailPoet\Newsletter\NewslettersRepository;
use MailPoet\Newsletter\Options\NewsletterOptionFieldsRepository; use MailPoet\Newsletter\Options\NewsletterOptionFieldsRepository;
use MailPoet\Newsletter\Options\NewsletterOptionsRepository; use MailPoet\Newsletter\Options\NewsletterOptionsRepository;
use MailPoet\Newsletter\Scheduler\AutomaticEmailScheduler; use MailPoet\Newsletter\Scheduler\AutomaticEmailScheduler;
@ -99,7 +100,7 @@ class AbandonedCartTest extends \MailPoetTest {
$this->wp = $wp; $this->wp = $wp;
WPFunctions::set($this->wp); WPFunctions::set($this->wp);
$this->automaticEmailScheduler = new AutomaticEmailScheduler(new Scheduler($this->wp)); $this->automaticEmailScheduler = new AutomaticEmailScheduler(new Scheduler($this->wp, $this->diContainer->get(NewslettersRepository::class)));
$this->wooCommerceMock = $this->mockWooCommerceClass(WooCommerce::class, []); $this->wooCommerceMock = $this->mockWooCommerceClass(WooCommerce::class, []);
$this->wooCommerceCartMock = $this->mockWooCommerceClass(WC_Cart::class, ['is_empty', 'get_cart']); $this->wooCommerceCartMock = $this->mockWooCommerceClass(WC_Cart::class, ['is_empty', 'get_cart']);

View File

@ -8,7 +8,6 @@ use MailPoet\Entities\NewsletterEntity;
use MailPoet\Entities\NewsletterOptionEntity; use MailPoet\Entities\NewsletterOptionEntity;
use MailPoet\Entities\NewsletterOptionFieldEntity; use MailPoet\Entities\NewsletterOptionFieldEntity;
use MailPoet\Entities\NewsletterPostEntity; use MailPoet\Entities\NewsletterPostEntity;
use MailPoet\Models\Newsletter;
use MailPoet\Newsletter\NewslettersRepository; use MailPoet\Newsletter\NewslettersRepository;
use MailPoet\Newsletter\Scheduler\AutomaticEmailScheduler; use MailPoet\Newsletter\Scheduler\AutomaticEmailScheduler;
use MailPoet\WP\Functions as WPFunctions; use MailPoet\WP\Functions as WPFunctions;
@ -198,8 +197,7 @@ class AbandonedCartContentTest extends \MailPoetTest {
private function createSendingTask($newsletter, $subscriberId = null, $meta = null) { private function createSendingTask($newsletter, $subscriberId = null, $meta = null) {
$subscriberId = $subscriberId ?: 1; // dummy default value $subscriberId = $subscriberId ?: 1; // dummy default value
$meta = $meta ?: [AbandonedCart::TASK_META_NAME => array_slice($this->productIds, 0, 3)]; $meta = $meta ?: [AbandonedCart::TASK_META_NAME => array_slice($this->productIds, 0, 3)];
$sendingTask = $this->automaticEmailScheduler $sendingTask = $this->automaticEmailScheduler->createAutomaticEmailSendingTask($newsletter, $subscriberId, $meta);
->createAutomaticEmailSendingTask(Newsletter::findOne($newsletter->getId()), $subscriberId, $meta);
return $sendingTask; return $sendingTask;
} }

View File

@ -3,6 +3,8 @@
namespace MailPoet\Newsletter\Scheduler; namespace MailPoet\Newsletter\Scheduler;
use Codeception\Util\Fixtures; use Codeception\Util\Fixtures;
use MailPoet\Entities\NewsletterEntity;
use MailPoet\Entities\NewsletterOptionFieldEntity;
use MailPoet\Entities\NewsletterPostEntity; use MailPoet\Entities\NewsletterPostEntity;
use MailPoet\Models\Newsletter; use MailPoet\Models\Newsletter;
use MailPoet\Models\NewsletterOption; use MailPoet\Models\NewsletterOption;
@ -11,6 +13,7 @@ use MailPoet\Models\ScheduledTask;
use MailPoet\Models\ScheduledTaskSubscriber; use MailPoet\Models\ScheduledTaskSubscriber;
use MailPoet\Models\SendingQueue; use MailPoet\Models\SendingQueue;
use MailPoet\Models\Subscriber; use MailPoet\Models\Subscriber;
use MailPoet\Newsletter\NewslettersRepository;
use MailPoet\Tasks\Sending as SendingTask; use MailPoet\Tasks\Sending as SendingTask;
use MailPoet\WP\Functions as WPFunctions; use MailPoet\WP\Functions as WPFunctions;
use MailPoetVendor\Carbon\Carbon; use MailPoetVendor\Carbon\Carbon;
@ -21,9 +24,13 @@ class AutomaticEmailTest extends \MailPoetTest {
/** @var AutomaticEmailScheduler */ /** @var AutomaticEmailScheduler */
private $automaticEmailScheduler; private $automaticEmailScheduler;
/** @var NewslettersRepository */
private $newslettersRepository;
public function _before() { public function _before() {
parent::_before(); parent::_before();
$this->automaticEmailScheduler = $this->diContainer->get(AutomaticEmailScheduler::class); $this->automaticEmailScheduler = $this->diContainer->get(AutomaticEmailScheduler::class);
$this->newslettersRepository = $this->diContainer->get(NewslettersRepository::class);
} }
public function testItCreatesScheduledAutomaticEmailSendingTaskForUser() { public function testItCreatesScheduledAutomaticEmailSendingTaskForUser() {
@ -36,15 +43,15 @@ class AutomaticEmailTest extends \MailPoetTest {
'afterTimeNumber' => 2, 'afterTimeNumber' => 2,
] ]
); );
$newsletter = Newsletter::filter('filterWithOptions', Newsletter::TYPE_AUTOMATIC)->findOne($newsletter->id); $newsletter = $this->newslettersRepository->findOneById($newsletter->id);
assert($newsletter instanceof Newsletter); $this->assertInstanceOf(NewsletterEntity::class, $newsletter);
$subscriber = Subscriber::create(); $subscriber = Subscriber::create();
$subscriber->hydrate(Fixtures::get('subscriber_template')); $subscriber->hydrate(Fixtures::get('subscriber_template'));
$subscriber->save(); $subscriber->save();
$this->automaticEmailScheduler->createAutomaticEmailSendingTask($newsletter, $subscriber->id, $meta = null); $this->automaticEmailScheduler->createAutomaticEmailSendingTask($newsletter, $subscriber->id, $meta = null);
// new scheduled task should be created // new scheduled task should be created
$task = SendingTask::getByNewsletterId($newsletter->id); $task = SendingTask::getByNewsletterId($newsletter->getId());
$currentTime = Carbon::createFromTimestamp(WPFunctions::get()->currentTime('timestamp')); $currentTime = Carbon::createFromTimestamp(WPFunctions::get()->currentTime('timestamp'));
Carbon::setTestNow($currentTime); // mock carbon to return current time Carbon::setTestNow($currentTime); // mock carbon to return current time
expect($task->id)->greaterOrEquals(1); expect($task->id)->greaterOrEquals(1);
@ -68,8 +75,8 @@ class AutomaticEmailTest extends \MailPoetTest {
'afterTimeNumber' => 2, 'afterTimeNumber' => 2,
] ]
); );
$newsletter = Newsletter::filter('filterWithOptions', Newsletter::TYPE_AUTOMATIC)->findOne($newsletter->id); $newsletter = $this->newslettersRepository->findOneById($newsletter->id);
assert($newsletter instanceof Newsletter); $this->assertInstanceOf(NewsletterEntity::class, $newsletter);
$subscriber = Subscriber::create(); $subscriber = Subscriber::create();
$subscriber->hydrate(Fixtures::get('subscriber_template')); $subscriber->hydrate(Fixtures::get('subscriber_template'));
$subscriber->save(); $subscriber->save();
@ -77,7 +84,7 @@ class AutomaticEmailTest extends \MailPoetTest {
$this->automaticEmailScheduler->createAutomaticEmailSendingTask($newsletter, $subscriber->id, $meta); $this->automaticEmailScheduler->createAutomaticEmailSendingTask($newsletter, $subscriber->id, $meta);
// new queue record should be created with meta data // new queue record should be created with meta data
$queue = SendingQueue::where('newsletter_id', $newsletter->id)->findOne(); $queue = SendingQueue::where('newsletter_id', $newsletter->getId())->findOne();
assert($queue instanceof SendingQueue); assert($queue instanceof SendingQueue);
expect($queue->getMeta())->equals($meta); expect($queue->getMeta())->equals($meta);
} }
@ -92,12 +99,12 @@ class AutomaticEmailTest extends \MailPoetTest {
'afterTimeNumber' => 2, 'afterTimeNumber' => 2,
] ]
); );
$newsletter = Newsletter::filter('filterWithOptions', Newsletter::TYPE_AUTOMATIC)->findOne($newsletter->id); $newsletter = $this->newslettersRepository->findOneById($newsletter->id);
assert($newsletter instanceof Newsletter); $this->assertInstanceOf(NewsletterEntity::class, $newsletter);
$this->automaticEmailScheduler->createAutomaticEmailSendingTask($newsletter, $subscriber = null, $meta = null); $this->automaticEmailScheduler->createAutomaticEmailSendingTask($newsletter, $subscriber = null, $meta = null);
// new scheduled task should be created // new scheduled task should be created
$task = SendingTask::getByNewsletterId($newsletter->id); $task = SendingTask::getByNewsletterId($newsletter->getId());
$currentTime = Carbon::createFromTimestamp(WPFunctions::get()->currentTime('timestamp')); $currentTime = Carbon::createFromTimestamp(WPFunctions::get()->currentTime('timestamp'));
Carbon::setTestNow($currentTime); // mock carbon to return current time Carbon::setTestNow($currentTime); // mock carbon to return current time
expect($task->id)->greaterOrEquals(1); expect($task->id)->greaterOrEquals(1);
@ -170,15 +177,15 @@ class AutomaticEmailTest extends \MailPoetTest {
'afterTimeNumber' => 2, 'afterTimeNumber' => 2,
] ]
); );
$condition = function($email) { $condition = function(NewsletterEntity $email) {
return $email->sendTo === 'segment'; return $email->getOptionValue(NewsletterOptionFieldEntity::NAME_SEND_TO) === 'segment';
}; };
$wpMock = $this->createMock(WPFunctions::class); $wpMock = $this->createMock(WPFunctions::class);
$wpMock->expects($this->any()) $wpMock->expects($this->any())
->method('currentTime') ->method('currentTime')
->willReturn($currentTime->getTimestamp()); ->willReturn($currentTime->getTimestamp());
$automaticEmailScheduler = new AutomaticEmailScheduler(new Scheduler($wpMock)); $automaticEmailScheduler = new AutomaticEmailScheduler(new Scheduler($wpMock, $this->diContainer->get(NewslettersRepository::class)));
// email should only be scheduled if it matches condition ("send to segment") // email should only be scheduled if it matches condition ("send to segment")
$automaticEmailScheduler->scheduleAutomaticEmail('some_group', 'some_event', $condition); $automaticEmailScheduler->scheduleAutomaticEmail('some_group', 'some_event', $condition);
$result = SendingQueue::findMany(); $result = SendingQueue::findMany();

View File

@ -2,18 +2,20 @@
namespace MailPoet\Test\Newsletter\Scheduler; namespace MailPoet\Test\Newsletter\Scheduler;
use MailPoet\Entities\NewsletterEntity;
use MailPoet\Entities\NewsletterOptionEntity;
use MailPoet\Entities\NewsletterOptionFieldEntity;
use MailPoet\Entities\NewsletterPostEntity; use MailPoet\Entities\NewsletterPostEntity;
use MailPoet\Entities\ScheduledTaskEntity;
use MailPoet\Entities\ScheduledTaskSubscriberEntity;
use MailPoet\Entities\SendingQueueEntity;
use MailPoet\Entities\SubscriberEntity;
use MailPoet\Models\Newsletter; use MailPoet\Models\Newsletter;
use MailPoet\Models\NewsletterOption; use MailPoet\Models\NewsletterOption;
use MailPoet\Models\NewsletterOptionField; use MailPoet\Models\NewsletterOptionField;
use MailPoet\Models\ScheduledTask;
use MailPoet\Models\ScheduledTaskSubscriber;
use MailPoet\Models\SendingQueue;
use MailPoet\Models\Subscriber;
use MailPoet\Newsletter\Scheduler\Scheduler; use MailPoet\Newsletter\Scheduler\Scheduler;
use MailPoet\WP\Functions as WPFunctions; use MailPoet\WP\Functions as WPFunctions;
use MailPoetVendor\Carbon\Carbon; use MailPoetVendor\Carbon\Carbon;
use MailPoetVendor\Idiorm\ORM;
class SchedulerTest extends \MailPoetTest { class SchedulerTest extends \MailPoetTest {
@ -26,26 +28,26 @@ class SchedulerTest extends \MailPoetTest {
} }
public function testItGetsActiveNewslettersFilteredByTypeAndGroup() { public function testItGetsActiveNewslettersFilteredByTypeAndGroup() {
$this->_createNewsletter($type = Newsletter::TYPE_WELCOME); $this->_createNewsletter(NewsletterEntity::TYPE_WELCOME);
// no newsletters with type "notification" should be found // no newsletters with type "notification" should be found
expect($this->testee->getNewsletters(Newsletter::TYPE_NOTIFICATION))->isEmpty(); expect($this->testee->getNewsletters(NewsletterEntity::TYPE_NOTIFICATION))->isEmpty();
// one newsletter with type "welcome" should be found // one newsletter with type "welcome" should be found
expect($this->testee->getNewsletters(Newsletter::TYPE_WELCOME))->count(1); expect($this->testee->getNewsletters(NewsletterEntity::TYPE_WELCOME))->count(1);
// one automatic email belonging to "test" group should be found // one automatic email belonging to "test" group should be found
$newsletter = $this->_createNewsletter($type = Newsletter::TYPE_AUTOMATIC); $newsletter = $this->_createNewsletter(NewsletterEntity::TYPE_AUTOMATIC);
$this->_createNewsletterOptions( $this->_createNewsletterOptions(
$newsletter->id, $newsletter->id,
Newsletter::TYPE_AUTOMATIC, NewsletterEntity::TYPE_AUTOMATIC,
[ [
'group' => 'test', NewsletterOptionFieldEntity::NAME_GROUP => 'test',
] ]
); );
expect($this->testee->getNewsletters(Newsletter::TYPE_AUTOMATIC, 'group_does_not_exist'))->isEmpty(); expect($this->testee->getNewsletters(Newsletter::TYPE_WELCOME, 'group_does_not_exist'))->isEmpty();
expect($this->testee->getNewsletters(Newsletter::TYPE_WELCOME, 'test'))->count(1); expect($this->testee->getNewsletters(Newsletter::TYPE_AUTOMATIC, 'test'))->count(1);
} }
public function testItCanGetNextRunDate() { public function testItCanGetNextRunDate() {
@ -107,13 +109,13 @@ class SchedulerTest extends \MailPoetTest {
public function _after() { public function _after() {
Carbon::setTestNow(); Carbon::setTestNow();
ORM::raw_execute('TRUNCATE ' . Newsletter::$_table); $this->truncateEntity(NewsletterEntity::class);
ORM::raw_execute('TRUNCATE ' . NewsletterOption::$_table); $this->truncateEntity(NewsletterOptionEntity::class);
ORM::raw_execute('TRUNCATE ' . NewsletterOptionField::$_table); $this->truncateEntity(NewsletterOptionFieldEntity::class);
$this->truncateEntity(NewsletterPostEntity::class); $this->truncateEntity(NewsletterPostEntity::class);
ORM::raw_execute('TRUNCATE ' . ScheduledTask::$_table); $this->truncateEntity(ScheduledTaskEntity::class);
ORM::raw_execute('TRUNCATE ' . ScheduledTaskSubscriber::$_table); $this->truncateEntity(ScheduledTaskSubscriberEntity::class);
ORM::raw_execute('TRUNCATE ' . SendingQueue::$_table); $this->truncateEntity(SendingQueueEntity::class);
ORM::raw_execute('TRUNCATE ' . Subscriber::$_table); $this->truncateEntity(SubscriberEntity::class);
} }
} }

View File

@ -467,7 +467,7 @@ class WelcomeTest extends \MailPoetTest {
$this->segmentRepository, $this->segmentRepository,
$this->diContainer->get(NewslettersRepository::class), $this->diContainer->get(NewslettersRepository::class),
$this->diContainer->get(ScheduledTasksRepository::class), $this->diContainer->get(ScheduledTasksRepository::class),
new Scheduler($wpMock) new Scheduler($wpMock, $this->diContainer->get(NewslettersRepository::class))
); );
} }

View File

@ -2,6 +2,7 @@
namespace MailPoet\Test\Newsletter\Editor; namespace MailPoet\Test\Newsletter\Editor;
use MailPoet\Newsletter\NewslettersRepository;
use MailPoet\Newsletter\Scheduler\Scheduler; use MailPoet\Newsletter\Scheduler\Scheduler;
use MailPoet\WP\Functions as WPFunctions; use MailPoet\WP\Functions as WPFunctions;
use MailPoetVendor\Carbon\Carbon; use MailPoetVendor\Carbon\Carbon;
@ -24,7 +25,8 @@ class SchedulerTest extends \MailPoetUnitTest {
$wp = $this->makeEmpty(WPFunctions::class, [ $wp = $this->makeEmpty(WPFunctions::class, [
'currentTime' => $this->currentTime->getTimestamp(), 'currentTime' => $this->currentTime->getTimestamp(),
]); ]);
$this->testee = new Scheduler($wp); $newslettersRepository = $this->makeEmpty(NewslettersRepository::class);
$this->testee = new Scheduler($wp, $newslettersRepository);
} }
public function testItScheduleTimeWithDelayByHours(): void { public function testItScheduleTimeWithDelayByHours(): void {