MessageStatusProcessorTest.php

Go to the documentation of this file.

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

use Magento\Framework\TestFramework\Unit\Helper\ObjectManager as ObjectManagerHelper;

class MessageStatusProcessorTest extends \PHPUnit\Framework\TestCase
{
    private $messageStatusProcessor;

    protected function setUp()
    {
        $objectManagerHelper = new ObjectManagerHelper($this);
        $this->messageStatusProcessor = $objectManagerHelper->getObject(
            \Magento\Framework\MessageQueue\MessageStatusProcessor::class
        );
    }

    public function testAcknowledgeMessages()
    {
        $queue = $this->getMockBuilder(\Magento\Framework\MessageQueue\QueueInterface::class)
            ->disableOriginalConstructor()
            ->getMockForAbstractClass();
        $queue->expects($this->atLeastOnce())->method('acknowledge');
        $message = $this->getMockBuilder(\Magento\Framework\MessageQueue\EnvelopeInterface::class)
            ->disableOriginalConstructor()
            ->getMockForAbstractClass();

        $this->messageStatusProcessor->acknowledgeMessages($queue, [$message]);
    }

    public function testRejectMessages()
    {
        $queue = $this->getMockBuilder(\Magento\Framework\MessageQueue\QueueInterface::class)
            ->disableOriginalConstructor()
            ->getMockForAbstractClass();
        $queue->expects($this->atLeastOnce())->method('reject');
        $message = $this->getMockBuilder(\Magento\Framework\MessageQueue\EnvelopeInterface::class)
            ->disableOriginalConstructor()
            ->getMockForAbstractClass();

        $this->messageStatusProcessor->rejectMessages($queue, [$message]);
    }
}