-
-
Notifications
You must be signed in to change notification settings - Fork 70
/
ScheduleManagementTest.php
85 lines (72 loc) · 2.78 KB
/
ScheduleManagementTest.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
<?php
declare(strict_types=1);
namespace Test\Integration\Model;
use EthanYehuda\CronjobManager\Api\ScheduleManagementInterface;
use Magento\Cron\Model\Schedule;
use Magento\TestFramework\Helper\Bootstrap;
use Magento\TestFramework\ObjectManager;
use PHPUnit\Framework\TestCase;
/**
* @magentoAppArea adminhtml
*/
class ScheduleManagementTest extends TestCase
{
/**
* @var ScheduleManagementInterface
*/
private $scheduleManagement;
/**
* @var ObjectManager
*/
private $objectManager;
const NOW = '2019-02-09 18:33:00';
protected function setUp(): void
{
$this->objectManager = Bootstrap::getObjectManager();
$this->scheduleManagement = $this->objectManager->get(ScheduleManagementInterface::class);
}
public function testKillRequestForRunningJobSucceeds()
{
$this->givenRunningSchedule($schedule);
$this->whenKillRequestedFor($schedule, strtotime(self::NOW));
$this->thenScheduleHasKillRequest($schedule, self::NOW);
}
public function testKillRequestForFinishedJobFails()
{
$this->givenFinishedSchedule($schedule);
$this->whenKillRequestedFor($schedule, strtotime(self::NOW));
$this->thenScheduleDoesNotHaveKillRequest($schedule);
}
private function givenRunningSchedule(&$schedule)
{
/** @var Schedule $schedule */
$schedule = $this->objectManager->create(Schedule::class);
$schedule->setStatus(Schedule::STATUS_RUNNING);
$schedule->save();
}
private function givenFinishedSchedule(&$schedule)
{
/** @var Schedule $schedule */
$schedule = $this->objectManager->create(Schedule::class);
$schedule->setStatus(Schedule::STATUS_SUCCESS);
$schedule->save();
}
private function whenKillRequestedFor(Schedule $schedule, $timestamp)
{
$this->scheduleManagement->kill((int)$schedule->getId(), $timestamp);
}
private function thenScheduleHasKillRequest($schedule, $now)
{
/** @var \Magento\Cron\Model\ResourceModel\Schedule $scheduleResource */
$scheduleResource = $this->objectManager->get(\Magento\Cron\Model\ResourceModel\Schedule::class);
$scheduleResource->load($schedule, $schedule->getId());
$this->assertEquals($now, $schedule->getData('kill_request'), 'Kill request should be saved in schedule');
}
private function thenScheduleDoesNotHaveKillRequest($schedule)
{
/** @var \Magento\Cron\Model\ResourceModel\Schedule $scheduleResource */
$scheduleResource = $this->objectManager->get(\Magento\Cron\Model\ResourceModel\Schedule::class);
$scheduleResource->load($schedule, $schedule->getId());
$this->assertNull($schedule->getData('kill_request'), 'Kill request should not be saved in schedule');
}
}