newsupernova/vendor/magento/module-mysql-mq/Test/Unit/Model/QueueManagementTest.php

<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */

namespace Magento\MysqlMq\Test\Unit\Model;

/**
 * Unit test for QueueManagement model.
 */
class QueueManagementTest extends \PHPUnit\Framework\TestCase
{
    /**
     * @var \Magento\MysqlMq\Model\ResourceModel\Queue|\PHPUnit_Framework_MockObject_MockObject
     */
    private $messageResource;

    /**
     * @var \Magento\Framework\App\Config\ScopeConfigInterface|\PHPUnit_Framework_MockObject_MockObject
     */
    private $scopeConfig;

    /**
     * @var \Magento\Framework\Stdlib\DateTime\DateTime|\PHPUnit_Framework_MockObject_MockObject
     */
    private $dateTime;

    /**
     * @var \Magento\MysqlMq\Model\ResourceModel\MessageStatusCollectionFactory
     *      |\PHPUnit_Framework_MockObject_MockObject
     */
    private $messageStatusCollectionFactory;

    /**
     * @var \Magento\MysqlMq\Model\QueueManagement
     */
    private $queueManagement;

    /**
     * Set up.
     *
     * @return void
     */
    protected function setUp()
    {
        $this->messageResource = $this->getMockBuilder(\Magento\MysqlMq\Model\ResourceModel\Queue::class)
            ->disableOriginalConstructor()->getMock();
        $this->scopeConfig = $this->getMockBuilder(\Magento\Framework\App\Config\ScopeConfigInterface::class)
            ->disableOriginalConstructor()->getMock();
        $this->dateTime = $this->getMockBuilder(\Magento\Framework\Stdlib\DateTime\DateTime::class)
            ->disableOriginalConstructor()->getMock();
        $this->messageStatusCollectionFactory = $this
            ->getMockBuilder(\Magento\MysqlMq\Model\ResourceModel\MessageStatusCollectionFactory::class)
            ->setMethods(['create'])
            ->disableOriginalConstructor()
            ->getMock();

        $objectManager = new \Magento\Framework\TestFramework\Unit\Helper\ObjectManager($this);
        $this->queueManagement = $objectManager->getObject(
            \Magento\MysqlMq\Model\QueueManagement::class,
            [
                'messageResource' => $this->messageResource,
                'scopeConfig' => $this->scopeConfig,
                'dateTime' => $this->dateTime,
                'messageStatusCollectionFactory' => $this->messageStatusCollectionFactory,
            ]
        );
    }

    /**
     * Test for addMessageToQueues method.
     *
     * @return void
     */
    public function testAddMessageToQueues()
    {
        $topicName = 'topic.name';
        $queueNames = ['queue0', 'queue1'];
        $message = 'test_message';
        $messageId = 1;
        $this->messageResource->expects($this->once())
            ->method('saveMessage')->with($topicName, $message)->willReturn($messageId);
        $this->messageResource->expects($this->once())
            ->method('linkQueues')->with($messageId, $queueNames)->willReturnSelf();
        $this->assertEquals(
            $this->queueManagement,
            $this->queueManagement->addMessageToQueues($topicName, $message, $queueNames)
        );
    }

    /**
     * Test for addMessagesToQueues method.
     *
     * @return void
     */
    public function testAddMessagesToQueues()
    {
        $topicName = 'topic.name';
        $queueNames = ['queue0', 'queue1'];
        $messages = ['test_message0', 'test_message1'];
        $messageIds = [1, 2];
        $this->messageResource->expects($this->once())
            ->method('saveMessages')->with($topicName, $messages)->willReturn($messageIds);
        $this->messageResource->expects($this->once())
            ->method('linkMessagesWithQueues')->with($messageIds, $queueNames)->willReturnSelf();
        $this->assertEquals(
            $this->queueManagement,
            $this->queueManagement->addMessagesToQueues($topicName, $messages, $queueNames)
        );
    }

    /**
     * Test for markMessagesForDelete method.
     *
     * @return void
     */
    public function testMarkMessagesForDelete()
    {
        $messageId = 99;
        $collection = $this->getMockBuilder(\Magento\MysqlMq\Model\ResourceModel\MessageStatusCollection::class)
            ->disableOriginalConstructor()->getMock();
        $this->messageStatusCollectionFactory->expects($this->once())->method('create')->willReturn($collection);
        $this->scopeConfig->expects($this->exactly(8))->method('getValue')
            ->withConsecutive(
                [
                    \Magento\MysqlMq\Model\QueueManagement::XML_PATH_SUCCESSFUL_MESSAGES_LIFETIME,
                    \Magento\Store\Model\ScopeInterface::SCOPE_STORE,
                ],
                [
                    \Magento\MysqlMq\Model\QueueManagement::XML_PATH_FAILED_MESSAGES_LIFETIME,
                    \Magento\Store\Model\ScopeInterface::SCOPE_STORE,
                ],
                [
                    \Magento\MysqlMq\Model\QueueManagement::XML_PATH_NEW_MESSAGES_LIFETIME,
                    \Magento\Store\Model\ScopeInterface::SCOPE_STORE,
                ],
                [
                    \Magento\MysqlMq\Model\QueueManagement::XML_PATH_RETRY_IN_PROGRESS_AFTER,
                    \Magento\Store\Model\ScopeInterface::SCOPE_STORE,
                ],
                [
                    \Magento\MysqlMq\Model\QueueManagement::XML_PATH_SUCCESSFUL_MESSAGES_LIFETIME,
                    \Magento\Store\Model\ScopeInterface::SCOPE_STORE,
                ],
                [
                    \Magento\MysqlMq\Model\QueueManagement::XML_PATH_FAILED_MESSAGES_LIFETIME,
                    \Magento\Store\Model\ScopeInterface::SCOPE_STORE,
                ],
                [
                    \Magento\MysqlMq\Model\QueueManagement::XML_PATH_NEW_MESSAGES_LIFETIME,
                    \Magento\Store\Model\ScopeInterface::SCOPE_STORE,
                ],
                [
                    \Magento\MysqlMq\Model\QueueManagement::XML_PATH_RETRY_IN_PROGRESS_AFTER,
                    \Magento\Store\Model\ScopeInterface::SCOPE_STORE,
                ]
            )->willReturn(1);
        $collection->expects($this->once())->method('addFieldToFilter')
            ->with(
                'status',
                [
                    'in' => [
                        \Magento\MysqlMq\Model\QueueManagement::MESSAGE_STATUS_COMPLETE,
                        \Magento\MysqlMq\Model\QueueManagement::MESSAGE_STATUS_ERROR,
                        \Magento\MysqlMq\Model\QueueManagement::MESSAGE_STATUS_NEW,
                        \Magento\MysqlMq\Model\QueueManagement::MESSAGE_STATUS_IN_PROGRESS,
                    ]
                ]
            )->willReturnSelf();
        $messageStatuses =
            [
                $this->getMessageStatusMock(),
                $this->getMessageStatusMock(),
                $this->getMessageStatusMock(),
                $this->getMessageStatusMock(),
            ];
        $this->dateTime->expects($this->exactly(4))->method('gmtTimestamp')->willReturn(1486741063);
        $messageStatuses[0]->expects($this->atLeastOnce())->method('getStatus')->willReturn(
            \Magento\MysqlMq\Model\QueueManagement::MESSAGE_STATUS_COMPLETE
        );
        $messageStatuses[1]->expects($this->atLeastOnce())->method('getStatus')->willReturn(
            \Magento\MysqlMq\Model\QueueManagement::MESSAGE_STATUS_ERROR
        );
        $messageStatuses[2]->expects($this->atLeastOnce())->method('getStatus')->willReturn(
            \Magento\MysqlMq\Model\QueueManagement::MESSAGE_STATUS_NEW
        );
        $messageStatuses[3]->expects($this->atLeastOnce())->method('getStatus')->willReturn(
            \Magento\MysqlMq\Model\QueueManagement::MESSAGE_STATUS_IN_PROGRESS
        );
        $messageStatuses[0]->expects($this->once())->method('setStatus')
            ->with(\Magento\MysqlMq\Model\QueueManagement::MESSAGE_STATUS_TO_BE_DELETED)->willReturnSelf();
        $messageStatuses[1]->expects($this->once())->method('setStatus')
            ->with(\Magento\MysqlMq\Model\QueueManagement::MESSAGE_STATUS_TO_BE_DELETED)->willReturnSelf();
        $messageStatuses[2]->expects($this->once())->method('setStatus')
            ->with(\Magento\MysqlMq\Model\QueueManagement::MESSAGE_STATUS_TO_BE_DELETED)->willReturnSelf();
        $messageStatuses[0]->expects($this->once())->method('save')->willReturnSelf();
        $messageStatuses[1]->expects($this->once())->method('save')->willReturnSelf();
        $messageStatuses[2]->expects($this->once())->method('save')->willReturnSelf();
        $messageStatuses[3]->expects($this->once())->method('getId')->willReturn($messageId);
        $collection->expects($this->once())->method('getIterator')->willReturn(new \ArrayIterator($messageStatuses));
        $this->messageResource->expects($this->once())->method('pushBackForRetry')->with($messageId);
        $this->messageResource->expects($this->once())->method('deleteMarkedMessages');
        $this->queueManagement->markMessagesForDelete();
    }

    /**
     * Create mock of MessageStatus method.
     *
     * @return \PHPUnit_Framework_MockObject_MockObject
     */
    private function getMessageStatusMock()
    {
        $messageStatus = $this->getMockBuilder(\Magento\MysqlMq\Model\MessageStatus::class)
            ->setMethods(['getStatus', 'setStatus', 'save', 'getId', 'getUpdatedAt'])
            ->disableOriginalConstructor()->getMock();
        $messageStatus->expects($this->once())->method('getUpdatedAt')->willReturn('2010-01-01 00:00:00');
        return $messageStatus;
    }

    /**
     * Test for changeStatus method.
     */
    public function testChangeStatus()
    {
        $messageIds = [1, 2];
        $status = \Magento\MysqlMq\Model\QueueManagement::MESSAGE_STATUS_TO_BE_DELETED;
        $this->messageResource->expects($this->once())->method('changeStatus')->with($messageIds, $status);
        $this->queueManagement->changeStatus($messageIds, $status);
    }
}