QueueTest.php

<?php

declare(strict_types=1);

use Carbon\Carbon;
use Illuminate\Support\Str;
use Jenssegers\Mongodb\Queue\Failed\MongoFailedJobProvider;
use Jenssegers\Mongodb\Queue\MongoQueue;

class QueueTest extends TestCase
{
    public function setUp(): void
    {
        parent::setUp();

        // Always start with a clean slate
        Queue::getDatabase()->table(Config::get('queue.connections.database.table'))->truncate();
        Queue::getDatabase()->table(Config::get('queue.failed.table'))->truncate();
    }

    public function testQueueJobLifeCycle(): void
    {
        $uuid = Str::uuid();

        Str::createUuidsUsing(function () use ($uuid) {
            return $uuid;
        });

        $id = Queue::push('test', ['action' => 'QueueJobLifeCycle'], 'test');
        $this->assertNotNull($id);

        // Get and reserve the test job (next available)
        $job = Queue::pop('test');
        $this->assertInstanceOf(Jenssegers\Mongodb\Queue\MongoJob::class, $job);
        $this->assertEquals(1, $job->isReserved());
        $this->assertEquals(json_encode([
            'uuid' => $uuid,
            'displayName' => 'test',
            'job' => 'test',
            'maxTries' => null,
            'maxExceptions' => null,
            'failOnTimeout' => false,
            'backoff' => null,
            'timeout' => null,
            'data' => ['action' => 'QueueJobLifeCycle'],
        ]), $job->getRawBody());

        // Remove reserved job
        $job->delete();
        $this->assertEquals(0, Queue::getDatabase()->table(Config::get('queue.connections.database.table'))->count());

        Str::createUuidsNormally();
    }

    public function testQueueJobExpired(): void
    {
        $id = Queue::push('test', ['action' => 'QueueJobExpired'], 'test');
        $this->assertNotNull($id);

        // Expire the test job
        $expiry = Carbon::now()->subSeconds(Config::get('queue.connections.database.expire'))->getTimestamp();
        Queue::getDatabase()
            ->table(Config::get('queue.connections.database.table'))
            ->where('_id', $id)
            ->update(['reserved' => 1, 'reserved_at' => $expiry]);

        // Expect an attempted older job in the queue
        $job = Queue::pop('test');
        $this->assertEquals(1, $job->attempts());
        $this->assertGreaterThan($expiry, $job->reservedAt());

        $job->delete();
        $this->assertEquals(0, Queue::getDatabase()->table(Config::get('queue.connections.database.table'))->count());
    }

    public function testFailQueueJob(): void
    {
        $provider = app('queue.failer');

        $this->assertInstanceOf(MongoFailedJobProvider::class, $provider);
    }

    public function testFindFailJobNull(): void
    {
        Config::set('queue.failed.database', 'mongodb');
        $provider = app('queue.failer');

        $this->assertNull($provider->find(1));
    }

    public function testIncrementAttempts(): void
    {
        $job_id = Queue::push('test1', ['action' => 'QueueJobExpired'], 'test');
        $this->assertNotNull($job_id);
        $job_id = Queue::push('test2', ['action' => 'QueueJobExpired'], 'test');
        $this->assertNotNull($job_id);

        $job = Queue::pop('test');
        $this->assertEquals(1, $job->attempts());
        $job->delete();

        $others_jobs = Queue::getDatabase()
            ->table(Config::get('queue.connections.database.table'))
            ->get();

        $this->assertCount(1, $others_jobs);
        $this->assertEquals(0, $others_jobs[0]['attempts']);
    }

    public function testJobRelease(): void
    {
        $queue = 'test';
        $job_id = Queue::push($queue, ['action' => 'QueueJobRelease'], 'test');
        $this->assertNotNull($job_id);

        $job = Queue::pop($queue);
        $job->release();

        $jobs = Queue::getDatabase()
            ->table(Config::get('queue.connections.database.table'))
            ->get();

        $this->assertCount(1, $jobs);
        $this->assertEquals(1, $jobs[0]['attempts']);
    }

    public function testQueueDeleteReserved(): void
    {
        $queue = 'test';
        $job_id = Queue::push($queue, ['action' => 'QueueDeleteReserved'], 'test');

        Queue::deleteReserved($queue, $job_id, 0);
        $jobs = Queue::getDatabase()
            ->table(Config::get('queue.connections.database.table'))
            ->get();

        $this->assertCount(0, $jobs);
    }

    public function testQueueRelease(): void
    {
        Carbon::setTestNow();
        $queue = 'test';
        $delay = 123;
        Queue::push($queue, ['action' => 'QueueRelease'], 'test');

        $job = Queue::pop($queue);
        $released_job_id = Queue::release($queue, $job->getJobRecord(), $delay);

        $released_job = Queue::getDatabase()
            ->table(Config::get('queue.connections.database.table'))
            ->where('_id', $released_job_id)
            ->first();

        $this->assertEquals($queue, $released_job['queue']);
        $this->assertEquals(1, $released_job['attempts']);
        $this->assertNull($released_job['reserved_at']);
        $this->assertEquals(
            Carbon::now()->addRealSeconds($delay)->getTimestamp(),
            $released_job['available_at']
        );
        $this->assertEquals(Carbon::now()->getTimestamp(), $released_job['created_at']);
        $this->assertEquals($job->getRawBody(), $released_job['payload']);
    }

    public function testQueueDeleteAndRelease(): void
    {
        $queue = 'test';
        $delay = 123;
        Queue::push($queue, ['action' => 'QueueDeleteAndRelease'], 'test');
        $job = Queue::pop($queue);

        $mock = Mockery::mock(MongoQueue::class)->makePartial();
        $mock->expects('deleteReserved')->once()->with($queue, $job->getJobId());
        $mock->expects('release')->once()->with($queue, $job->getJobRecord(), $delay);

        $mock->deleteAndRelease($queue, $job, $delay);
    }
}