fixes and tests

This commit is contained in:
Yuri Kuznetsov
2021-06-15 18:48:42 +03:00
parent 7d98a01e9a
commit faa8545fcf
3 changed files with 218 additions and 7 deletions

View File

@@ -126,10 +126,10 @@ class JobScheduler
*/
public function setTime(?DateTimeInterface $time): self
{
$this->time = null;
$this->time = $time;
if (!is_null($time)) {
$this->time = DateTimeImmutable::createFromInterface($time);
if (!is_null($time) && !$time instanceof DateTimeImmutable) {
$this->time = DateTimeImmutable::createFromMutable($time);
}
return $this;
@@ -171,7 +171,7 @@ class JobScheduler
throw new RuntimeException("Class name is not set.");
}
if ($this->group && $this->queue !== null) {
if ($this->group && $this->queue === null) {
throw new RuntimeException("A group can't be set w/o queue.");
}
@@ -181,13 +181,19 @@ class JobScheduler
$time = $time->add($this->delay);
}
$data = $this->data;
if (!$data) {
$data = JobData::create();
}
return $this->entityManager->createEntity(JobEntity::ENTITY_TYPE, [
'className' => $this->className,
'queue' => $this->queue,
'group' => $this->group,
'targetType' => $this->data->getTargetType(),
'targetId' => $this->data->getTargetId(),
'data' => $this->data->getRaw(),
'targetType' => $data->getTargetType(),
'targetId' => $data->getTargetId(),
'data' => $data->getRaw(),
'executeTime' => $time->format(DateTime::SYSTEM_DATE_TIME_FORMAT),
]);
}

View File

@@ -0,0 +1,167 @@
<?php
/************************************************************************
* This file is part of EspoCRM.
*
* EspoCRM - Open Source CRM application.
* Copyright (C) 2014-2021 Yurii Kuznietsov, Taras Machyshyn, Oleksii Avramenko
* Website: https://www.espocrm.com
*
* EspoCRM is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* EspoCRM is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with EspoCRM. If not, see http://www.gnu.org/licenses/.
*
* The interactive user interfaces in modified source and object code versions
* of this program must display Appropriate Legal Notices, as required under
* Section 5 of the GNU General Public License version 3.
*
* In accordance with Section 7(b) of the GNU General Public License version 3,
* these Appropriate Legal Notices must retain the display of the "EspoCRM" word.
************************************************************************/
namespace tests\unit\Espo\Core\Job;
use Espo\Core\Job\JobScheduler;
use Espo\Core\Job\QueueName;
use Espo\Core\Utils\DateTime;
use Espo\Core\Job\JobData;
use Espo\ORM\EntityManager;
use Espo\Entities\Job as JobEntity;
use tests\unit\testClasses\Core\Job\TestJob;
use DateTimeImmutable;
use DateInterval;
class JobSchedulerTest extends \PHPUnit\Framework\TestCase
{
private $entityManager;
protected function setUp(): void
{
$this->entityManager = $this->createMock(EntityManager::class);
}
public function testSchedule1(): void
{
$scheduler = new JobScheduler($this->entityManager);
$jobEntity = $this->createMock(JobEntity::class);
$time = new DateTimeImmutable();
$delay = DateInterval::createFromDateString('1 minute');
$this->entityManager
->expects($this->once())
->method('createEntity')
->with(
JobEntity::ENTITY_TYPE,
[
'className' => TestJob::class,
'queue' => QueueName::Q0,
'group' => 'test-group',
'data' => (object) [
'test' => '1',
],
'executeTime' => $time->modify('+1 minute')->format(DateTime::SYSTEM_DATE_TIME_FORMAT),
'targetId' => null,
'targetType' => null,
]
)
->willReturn($jobEntity);
$jobEntityReturned = $scheduler
->setClassName(TestJob::class)
->setQueue(QueueName::Q0)
->setGroup('test-group')
->setData([
'test' => '1',
])
->setTime($time)
->setDelay($delay)
->schedule();
$this->assertSame($jobEntityReturned, $jobEntity);
}
public function testSchedule2(): void
{
$scheduler = new JobScheduler($this->entityManager);
$jobEntity = $this->createMock(JobEntity::class);
$time = new DateTimeImmutable();
$this->entityManager
->expects($this->once())
->method('createEntity')
->with(
JobEntity::ENTITY_TYPE,
[
'className' => TestJob::class,
'queue' => null,
'group' => null,
'data' => (object) [
'test' => '1',
],
'executeTime' => $time->format(DateTime::SYSTEM_DATE_TIME_FORMAT),
'targetId' => 'test-id',
'targetType' => 'TestType',
]
)
->willReturn($jobEntity);
$data = JobData
::create([
'test' => '1',
])
->withTargetId('test-id')
->withTargetType('TestType');
$jobEntityReturned = $scheduler
->setClassName(TestJob::class)
->setData($data)
->setTime($time)
->schedule();
$this->assertSame($jobEntityReturned, $jobEntity);
}
public function testSchedule3(): void
{
$scheduler = new JobScheduler($this->entityManager);
$jobEntity = $this->createMock(JobEntity::class);
$this->entityManager
->expects($this->once())
->method('createEntity')
->with(
JobEntity::ENTITY_TYPE,
$this->callback(
function (array $data): bool {
return is_string($data['executeTime']);
}
)
)
->willReturn($jobEntity);
$jobEntityReturned = $scheduler
->setClassName(TestJob::class)
->schedule();
$this->assertSame($jobEntityReturned, $jobEntity);
}
}

View File

@@ -0,0 +1,38 @@
<?php
/************************************************************************
* This file is part of EspoCRM.
*
* EspoCRM - Open Source CRM application.
* Copyright (C) 2014-2021 Yurii Kuznietsov, Taras Machyshyn, Oleksii Avramenko
* Website: https://www.espocrm.com
*
* EspoCRM is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* EspoCRM is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with EspoCRM. If not, see http://www.gnu.org/licenses/.
*
* The interactive user interfaces in modified source and object code versions
* of this program must display Appropriate Legal Notices, as required under
* Section 5 of the GNU General Public License version 3.
*
* In accordance with Section 7(b) of the GNU General Public License version 3,
* these Appropriate Legal Notices must retain the display of the "EspoCRM" word.
************************************************************************/
namespace tests\unit\testClasses\Core\Job;
use Espo\Core\Job\Job;
use Espo\Core\Job\JobData;
class TestJob implements Job
{
public function run(JobData $data): void {}
}