ConsumerRunnerTest.php

Go to the documentation of this file.

<?php
namespace Magento\MessageQueue\Test\Unit\Model;

use Magento\Framework\MessageQueue\ConsumerInterface;
use Magento\MessageQueue\Model\ConsumerRunner;
use Magento\Framework\Exception\LocalizedException;
use Magento\Framework\Phrase;

class ConsumerRunnerTest extends \PHPUnit\Framework\TestCase
{
    private $objectManager;

    private $consumerRunner;

    private $consumerFactoryMock;

    private $maintenanceModeMock;

    protected function setUp()
    {
        $this->objectManager = new \Magento\Framework\TestFramework\Unit\Helper\ObjectManager($this);
        $this->consumerFactoryMock = $this->getMockBuilder(\Magento\Framework\MessageQueue\ConsumerFactory::class)
            ->disableOriginalConstructor()
            ->getMock();
        $this->maintenanceModeMock = $this->getMockBuilder(\Magento\Framework\App\MaintenanceMode::class)
            ->disableOriginalConstructor()
            ->getMock();
        $this->consumerRunner = $this->objectManager->getObject(
            \Magento\MessageQueue\Model\ConsumerRunner::class,
            [
                'consumerFactory' => $this->consumerFactoryMock,
                'maintenanceMode' => $this->maintenanceModeMock
            ]
        );
        parent::setUp();
    }

    public function testMagicMethod()
    {
        $isMaintenanceModeOn = false;
        $consumerMock = $this->getMockBuilder(\Magento\Framework\MessageQueue\ConsumerInterface::class)->getMock();
        $consumerMock->expects($this->once())->method('process');
        $consumerName = 'someConsumerName';
        $this->consumerFactoryMock
            ->expects($this->once())
            ->method('get')
            ->with($consumerName)
            ->willReturn($consumerMock);
        $this->maintenanceModeMock->expects($this->once())->method('isOn')->willReturn($isMaintenanceModeOn);

        $this->consumerRunner->$consumerName();
    }

    public function testMagicMethodNoRelatedConsumer()
    {
        $consumerName = 'nonDeclaredConsumer';
        $this->consumerFactoryMock
            ->expects($this->once())
            ->method('get')
            ->with($consumerName)
            ->willThrowException(new LocalizedException(new Phrase("Some exception")));

        $this->consumerRunner->$consumerName();
    }

    public function testMagicMethodMaintenanceModeIsOn()
    {
        $isMaintenanceModeOn = true;
        $consumerMock = $this->getMockBuilder(\Magento\Framework\MessageQueue\ConsumerInterface::class)->getMock();
        $consumerMock->expects($this->never())->method('process');
        $consumerName = 'someConsumerName';
        $this->consumerFactoryMock
            ->expects($this->once())
            ->method('get')
            ->with($consumerName)
            ->willReturn($consumerMock);
        $this->maintenanceModeMock->expects($this->once())->method('isOn')->willReturn($isMaintenanceModeOn);

        $this->consumerRunner->$consumerName();
    }
}