123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316 |
- <?php
- namespace Test\BackgroundJob;
- use OCP\AppFramework\Utility\ITimeFactory;
- use OCP\BackgroundJob\IJob;
- use OCP\DB\QueryBuilder\IQueryBuilder;
- use OCP\IConfig;
- use Psr\Log\LoggerInterface;
- use Test\TestCase;
- class JobListTest extends TestCase {
-
- protected $instance;
-
- protected $connection;
-
- protected $config;
-
- protected $timeFactory;
- private bool $ran = false;
- protected function setUp(): void {
- parent::setUp();
- $this->connection = \OC::$server->getDatabaseConnection();
- $this->clearJobsList();
- $this->config = $this->createMock(IConfig::class);
- $this->timeFactory = $this->createMock(ITimeFactory::class);
- $this->instance = new \OC\BackgroundJob\JobList(
- $this->connection,
- $this->config,
- $this->timeFactory,
- \OC::$server->get(LoggerInterface::class),
- );
- }
- protected function clearJobsList() {
- $query = $this->connection->getQueryBuilder();
- $query->delete('jobs');
- $query->execute();
- }
- protected function getAllSorted() {
- $iterator = $this->instance->getJobsIterator(null, null, 0);
- $jobs = [];
- foreach ($iterator as $job) {
- $jobs[] = clone $job;
- }
- usort($jobs, function (IJob $job1, IJob $job2) {
- return $job1->getId() - $job2->getId();
- });
- return $jobs;
- }
- public function argumentProvider() {
- return [
- [null],
- [false],
- ['foobar'],
- [12],
- [[
- 'asd' => 5,
- 'foo' => 'bar'
- ]]
- ];
- }
-
- public function testAddRemove($argument) {
- $existingJobs = $this->getAllSorted();
- $job = new TestJob();
- $this->instance->add($job, $argument);
- $jobs = $this->getAllSorted();
- $this->assertCount(count($existingJobs) + 1, $jobs);
- $addedJob = $jobs[count($jobs) - 1];
- $this->assertInstanceOf('\Test\BackgroundJob\TestJob', $addedJob);
- $this->assertEquals($argument, $addedJob->getArgument());
- $this->instance->remove($job, $argument);
- $jobs = $this->getAllSorted();
- $this->assertEquals($existingJobs, $jobs);
- }
-
- public function testRemoveDifferentArgument($argument) {
- $existingJobs = $this->getAllSorted();
- $job = new TestJob();
- $this->instance->add($job, $argument);
- $jobs = $this->getAllSorted();
- $this->instance->remove($job, 10);
- $jobs2 = $this->getAllSorted();
- $this->assertEquals($jobs, $jobs2);
- $this->instance->remove($job, $argument);
- $jobs = $this->getAllSorted();
- $this->assertEquals($existingJobs, $jobs);
- }
-
- public function testHas($argument) {
- $job = new TestJob();
- $this->assertFalse($this->instance->has($job, $argument));
- $this->instance->add($job, $argument);
- $this->assertTrue($this->instance->has($job, $argument));
- $this->instance->remove($job, $argument);
- $this->assertFalse($this->instance->has($job, $argument));
- }
-
- public function testHasDifferentArgument($argument) {
- $job = new TestJob();
- $this->instance->add($job, $argument);
- $this->assertFalse($this->instance->has($job, 10));
- }
- protected function createTempJob($class, $argument, $reservedTime = 0, $lastChecked = 0) {
- if ($lastChecked === 0) {
- $lastChecked = time();
- }
- $query = $this->connection->getQueryBuilder();
- $query->insert('jobs')
- ->values([
- 'class' => $query->createNamedParameter($class),
- 'argument' => $query->createNamedParameter($argument),
- 'last_run' => $query->createNamedParameter(0, IQueryBuilder::PARAM_INT),
- 'last_checked' => $query->createNamedParameter($lastChecked, IQueryBuilder::PARAM_INT),
- 'reserved_at' => $query->createNamedParameter($reservedTime, IQueryBuilder::PARAM_INT),
- ]);
- $query->execute();
- }
- public function testGetNext() {
- $job = new TestJob();
- $this->createTempJob(get_class($job), 1, 0, 12345);
- $this->createTempJob(get_class($job), 2, 0, 12346);
- $jobs = $this->getAllSorted();
- $savedJob1 = $jobs[0];
- $this->timeFactory->expects($this->atLeastOnce())
- ->method('getTime')
- ->willReturn(123456789);
- $nextJob = $this->instance->getNext();
- $this->assertEquals($savedJob1, $nextJob);
- }
- public function testGetNextSkipReserved() {
- $job = new TestJob();
- $this->createTempJob(get_class($job), 1, 123456789, 12345);
- $this->createTempJob(get_class($job), 2, 0, 12346);
- $this->timeFactory->expects($this->atLeastOnce())
- ->method('getTime')
- ->willReturn(123456789);
- $nextJob = $this->instance->getNext();
- $this->assertEquals(get_class($job), get_class($nextJob));
- $this->assertEquals(2, $nextJob->getArgument());
- }
- public function testGetNextSkipNonExisting() {
- $job = new TestJob();
- $this->createTempJob('\OC\Non\Existing\Class', 1, 0, 12345);
- $this->createTempJob(get_class($job), 2, 0, 12346);
- $this->timeFactory->expects($this->atLeastOnce())
- ->method('getTime')
- ->willReturn(123456789);
- $nextJob = $this->instance->getNext();
- $this->assertEquals(get_class($job), get_class($nextJob));
- $this->assertEquals(2, $nextJob->getArgument());
- }
-
- public function testGetById($argument) {
- $job = new TestJob();
- $this->instance->add($job, $argument);
- $jobs = $this->getAllSorted();
- $addedJob = $jobs[count($jobs) - 1];
- $this->assertEquals($addedJob, $this->instance->getById($addedJob->getId()));
- }
- public function testSetLastRun() {
- $job = new TestJob();
- $this->instance->add($job);
- $jobs = $this->getAllSorted();
- $addedJob = $jobs[count($jobs) - 1];
- $timeStart = time();
- $this->instance->setLastRun($addedJob);
- $timeEnd = time();
- $addedJob = $this->instance->getById($addedJob->getId());
- $this->assertGreaterThanOrEqual($timeStart, $addedJob->getLastRun());
- $this->assertLessThanOrEqual($timeEnd, $addedJob->getLastRun());
- }
- public function testHasReservedJobs() {
- $this->clearJobsList();
- $this->timeFactory->expects($this->atLeastOnce())
- ->method('getTime')
- ->willReturn(123456789);
- $job = new TestJob($this->timeFactory, $this, function () {
- });
- $job2 = new TestJob($this->timeFactory, $this, function () {
- });
- $this->instance->add($job, 1);
- $this->instance->add($job2, 2);
- $this->assertCount(2, iterator_to_array($this->instance->getJobsIterator(null, 10, 0)));
- $this->assertFalse($this->instance->hasReservedJob());
- $this->assertFalse($this->instance->hasReservedJob(TestJob::class));
- $job = $this->instance->getNext();
- $this->assertNotNull($job);
- $this->assertTrue($this->instance->hasReservedJob());
- $this->assertTrue($this->instance->hasReservedJob(TestJob::class));
- $job = $this->instance->getNext();
- $this->assertNotNull($job);
- $this->assertTrue($this->instance->hasReservedJob());
- $this->assertTrue($this->instance->hasReservedJob(TestJob::class));
- }
- public function testHasReservedJobsAndParallelAwareJob() {
- $this->clearJobsList();
- $this->timeFactory->expects($this->atLeastOnce())
- ->method('getTime')
- ->willReturnCallback(function () use (&$time) {
- return time();
- });
- $job = new TestParallelAwareJob($this->timeFactory, $this, function () {
- });
- $job2 = new TestParallelAwareJob($this->timeFactory, $this, function () {
- });
- $this->instance->add($job, 1);
- $this->instance->add($job2, 2);
- $this->assertCount(2, iterator_to_array($this->instance->getJobsIterator(null, 10, 0)));
- $this->assertFalse($this->instance->hasReservedJob());
- $this->assertFalse($this->instance->hasReservedJob(TestParallelAwareJob::class));
- $job = $this->instance->getNext();
- $this->assertNotNull($job);
- $this->assertTrue($this->instance->hasReservedJob());
- $this->assertTrue($this->instance->hasReservedJob(TestParallelAwareJob::class));
- $job = $this->instance->getNext();
- $this->assertNull($job);
- }
- public function markRun() {
- $this->ran = true;
- }
- }
|