Cavemanon/piratepoet
8
Fork 0
Code Pull Requests Actions Releases 16 Activity

Add test for switching active automation to draft

Browse Source 
[MAILPOET-6131]
This commit is contained in:
Jan Jakes
2024-07-04 10:05:09 +02:00
committed by Aschepikov
parent 9be6895cf1
commit c84910c075
1 changed files with 118 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

118
mailpoet/tests/integration/Automation/Engine/Builder/UpdateAutomationControllerTest.php Normal file
View File

@@ -0,0 +1,118 @@
<?php declare(strict_types = 1);
namespace MailPoet\Test\Automation\Engine\Control;
use ActionScheduler_Store;
use MailPoet\Automation\Engine\Builder\UpdateAutomationController;
use MailPoet\Automation\Engine\Data\Automation;
use MailPoet\Automation\Engine\Data\AutomationRun;
use MailPoet\Automation\Engine\Data\NextStep;
use MailPoet\Automation\Engine\Data\Step;
use MailPoet\Automation\Engine\Hooks;
use MailPoet\Automation\Engine\Storage\AutomationRunStorage;
use MailPoet\Automation\Engine\Storage\AutomationStorage;
use MailPoetTest;
class UpdateAutomationControllerTest extends MailPoetTest {
public function testItUnschedulesTasksWhenSwitchedToDraft(): void {
$automation = $this->tester->createAutomation(
'Test automation',
new Step('t', Step::TYPE_TRIGGER, 'test:trigger', [], [new NextStep('a1')]),
new Step('a1', Step::TYPE_ACTION, 'test:action', [], [new NextStep('a2')]),
new Step('a2', Step::TYPE_ACTION, 'test:action', [], [])
);
$run1 = $this->tester->createAutomationRun($automation);
$run2 = $this->tester->createAutomationRun($automation);
$run3 = $this->tester->createAutomationRun($automation);
$run4 = $this->tester->createAutomationRun($automation);
$runStorage = $this->diContainer->get(AutomationRunStorage::class);
$runStorage->updateStatus($run1->getId(), AutomationRun::STATUS_RUNNING);
$runStorage->updateStatus($run2->getId(), AutomationRun::STATUS_COMPLETE);
$runStorage->updateStatus($run3->getId(), AutomationRun::STATUS_FAILED);
$runStorage->updateStatus($run4->getId(), AutomationRun::STATUS_CANCELLED);
$this->scheduleAction(time() + 100, ['automation_run_id' => $run1->getId(), 'step_id' => 'aaa', 'run_number' => 1]);
$this->scheduleAction(time() + 200, ['automation_run_id' => $run1->getId(), 'step_id' => 'bbb', 'run_number' => 1]);
$this->scheduleAction(time() + 300, ['automation_run_id' => $run1->getId(), 'step_id' => 'bbb', 'run_number' => 2]);
$this->assertCount(3, $this->getActions());
$this->assertCount(3, $this->getActions(['status' => ActionScheduler_Store::STATUS_PENDING]));
// switch to draft
$controller = $this->diContainer->get(UpdateAutomationController::class);
$controller->updateAutomation($automation->getId(), [
'status' => Automation::STATUS_DRAFT,
]);
$automationStorage = $this->diContainer->get(AutomationStorage::class);
$updatedAutomation = $automationStorage->getAutomation($automation->getId());
$this->assertInstanceOf(Automation::class, $updatedAutomation);
$this->assertSame(Automation::STATUS_DRAFT, $updatedAutomation->getStatus());
$this->assertSame(AutomationRun::STATUS_CANCELLED, $this->getAutomationRun($run1->getId())->getStatus());
$this->assertSame(AutomationRun::STATUS_COMPLETE, $this->getAutomationRun($run2->getId())->getStatus());
$this->assertSame(AutomationRun::STATUS_FAILED, $this->getAutomationRun($run3->getId())->getStatus());
$this->assertSame(AutomationRun::STATUS_CANCELLED, $this->getAutomationRun($run4->getId())->getStatus());
$this->assertCount(3, $this->getActions());
$this->assertCount(3, $this->getActions(['status' => ActionScheduler_Store::STATUS_CANCELED]));
$this->assertCount(0, $this->getActions(['status' => ActionScheduler_Store::STATUS_PENDING]));
// resume automation and check that no run status was changed
$controller->updateAutomation($automation->getId(), [
'status' => Automation::STATUS_ACTIVE,
]);
$automationStorage = $this->diContainer->get(AutomationStorage::class);
$updatedAutomation = $automationStorage->getAutomation($automation->getId());
$this->assertInstanceOf(Automation::class, $updatedAutomation);
$this->assertSame(Automation::STATUS_ACTIVE, $updatedAutomation->getStatus());
$this->assertSame(AutomationRun::STATUS_CANCELLED, $this->getAutomationRun($run1->getId())->getStatus());
$this->assertSame(AutomationRun::STATUS_COMPLETE, $this->getAutomationRun($run2->getId())->getStatus());
$this->assertSame(AutomationRun::STATUS_FAILED, $this->getAutomationRun($run3->getId())->getStatus());
$this->assertSame(AutomationRun::STATUS_CANCELLED, $this->getAutomationRun($run4->getId())->getStatus());
$this->assertCount(3, $this->getActions());
$this->assertCount(3, $this->getActions(['status' => ActionScheduler_Store::STATUS_CANCELED]));
$this->assertCount(0, $this->getActions(['status' => ActionScheduler_Store::STATUS_PENDING]));
}
private function getAutomationRun(int $id): AutomationRun {
$runStorage = $this->diContainer->get(AutomationRunStorage::class);
$run = $runStorage->getAutomationRun($id);
$this->assertInstanceOf(AutomationRun::class, $run);
return $run;
}
private function scheduleAction(int $timestamp, array $args): int {
return as_schedule_single_action($timestamp, Hooks::AUTOMATION_STEP, [$args], 'mailpoet-automation');
}
private function getActions(array $args = []): array {
return array_values(
as_get_scheduled_actions(
array_merge(['hook' => Hooks::AUTOMATION_STEP, 'group' => 'mailpoet-automation'], $args)
)
);
}
public function _before(): void {
$this->cleanup();
}
public function _after(): void {
$this->cleanup();
}
private function cleanup() {
$this->diContainer->get(AutomationStorage::class)->truncate();
$this->diContainer->get(AutomationRunStorage::class)->truncate();
global $wpdb;
$actionsTable = $wpdb->prefix . 'actionscheduler_actions';
$wpdb->query('TRUNCATE ' . $actionsTable);
}
}