b2pweb / bdf-queue-bundle

Tests/WithPrimeTest.php

<?php

namespace Bdf\QueueBundle\Tests;

require_once __DIR__.'/TestKernel.php';

use Bdf\Prime\Connection\Result\DoctrineResultSet;
use Bdf\Prime\ServiceLocator;
use Bdf\Queue\Connection\Prime\PrimeConnection;

The type Bdf\Queue\Connection\Prime\PrimeConnection was not found. Maybe you did not declare it correctly or list all dependencies?

use Bdf\Queue\Console\Command\Failer\AbstractFailerCommand;
use Bdf\Queue\Console\Command\Failer\InitCommand;

The type Bdf\Queue\Console\Command\Failer\InitCommand was not found. Maybe you did not declare it correctly or list all dependencies?

use Bdf\Queue\Destination\DestinationManager;
use Bdf\Queue\Failer\DbFailedJobRepository;

The type Bdf\Queue\Failer\DbFailedJobRepository was not found. Maybe you did not declare it correctly or list all dependencies?

use Bdf\Queue\Failer\DbFailedJobStorage;

The type Bdf\Queue\Failer\DbFailedJobStorage was not found. Maybe you did not declare it correctly or list all dependencies?

use Bdf\Queue\Failer\FailedJobRepositoryAdapter;
use Bdf\Queue\Failer\FailedJobStorageInterface;
use PHPUnit\Framework\TestCase;
use Symfony\Bundle\FrameworkBundle\Console\Application;
use Symfony\Component\Console\Command\LazyCommand;

class WithPrimeTest extends TestCase
{
    protected function setUp(): void
    {
        if (!class_exists(PrimeConnection::class)) {
            $this->markTestSkipped('b2pweb/bdf-queue-prime-adapter not installed');
        }
    }

    /**
     * @return void
     */
    public function testPrimeFailer()
    {
        $kernel = new \TestKernel(__DIR__.'/Fixtures/conf_with_prime_failer.yaml');
        $kernel->boot();
        $console = new Application($kernel);

        $command = $console->get('queue:failer:delete');

        if ($command instanceof LazyCommand) {
            $command = $command->getCommand();
        }

        $r = new \ReflectionProperty(AbstractFailerCommand::class, 'repository');
        $r->setAccessible(true);

        /** @var FailedJobStorageInterface $failer */
        $failer = $r->getValue($command);

        $this->assertTrue($this->isPrimeFailer($failer));

        if (class_exists(DbFailedJobRepository::class)) {
            $this->assertInstanceOf(DbFailedJobRepository::class, $failer);
            $this->assertEquals(DbFailedJobRepository::make($kernel->getContainer()->get('prime'), ['connection' => 'my_connection', 'table' => 'failed_jobs'], 15), $failer);
        } else {
            $this->assertInstanceOf(FailedJobRepositoryAdapter::class, $failer);
            $this->assertEquals(
                FailedJobRepositoryAdapter::adapt(DbFailedJobStorage::make($kernel->getContainer()->get('prime'), ['connection' => 'my_connection', 'table' => 'failed_jobs'], 15)),
                $failer
            );
        }
    }

    /**
     * @return void
     */
    public function testHostMissing()
    {
        $this->expectException(\InvalidArgumentException::class);
        $this->expectExceptionMessage('The connection name is required on prime failer DSN');

        $kernel = new \TestKernel(__DIR__.'/Fixtures/conf_with_prime_failer_missing_host.yaml');
        $kernel->boot();
    }

    /**
     * @return void
     */
    public function testTableNameMissing()
    {
        $this->expectException(\InvalidArgumentException::class);
        $this->expectExceptionMessage('The table name is required on prime failer DSN');

        $kernel = new \TestKernel(__DIR__.'/Fixtures/conf_with_prime_failer_missing_table.yaml');
        $kernel->boot();
    }

    public function testInitCommand()
    {
        $kernel = new \TestKernel(__DIR__.'/Fixtures/conf_with_prime_failer.yaml');
        $kernel->boot();
        $console = new Application($kernel);

        $command = $console->get('queue:prime:failer:init');
        if ($command instanceof LazyCommand) {
            $command = $command->getCommand();
        }

        $this->assertInstanceOf(InitCommand::class, $command);
    }

    private function isPrimeFailer($storage): bool
    {
        if ($storage instanceof DbFailedJobRepository) {
            return true;
        }

        if ($storage instanceof FailedJobRepositoryAdapter) {
            $r = new \ReflectionProperty(FailedJobRepositoryAdapter::class, 'storage');
            $r->setAccessible(true);

            return $r->getValue($storage) instanceof DbFailedJobStorage;
        }

        return false;
    }

    public function testPrimeConnection()
    {
        $kernel = new \TestKernel(__DIR__.'/Fixtures/conf_with_prime_connection.yaml');
        $kernel->boot();

        /** @var ServiceLocator $prime */
        $prime = $kernel->getContainer()->get('prime');

        /** @var DestinationManager $destinations */
        $destinations = $kernel->getContainer()->get('bdf_queue.destination_manager');
        $destination = $destinations->queue('prime', 'queue_name');
        $destination->declare();

        $count = $prime->connection('my_connection')->select('select count(*) as nb from queue');
        $count = $count instanceof DoctrineResultSet ? $count->all()[0] : $count[0];

        $this->assertEquals(0, $count->nb);

        $destination->raw('test');

        $count = $prime->connection('my_connection')->select('select count(*) as nb from queue');
        $count = $count instanceof DoctrineResultSet ? $count->all()[0] : $count[0];
        $this->assertEquals(1, $count->nb);
    }
}