AccessChangeQuoteControlTest.php

Go to the documentation of this file.

<?php
namespace Magento\Quote\Test\Unit\Model\QuoteRepository\Plugin;

use Magento\Authorization\Model\UserContextInterface;
use Magento\Quote\Model\ChangeQuoteControl;
use Magento\Quote\Model\QuoteRepository\Plugin\AccessChangeQuoteControl;
use Magento\Quote\Model\Quote;
use Magento\Framework\TestFramework\Unit\Helper\ObjectManager;
use Magento\Quote\Model\QuoteRepository;
use \PHPUnit_Framework_MockObject_MockObject as MockObject;

class AccessChangeQuoteControlTest extends \PHPUnit\Framework\TestCase
{
    private $accessChangeQuoteControl;

    private $userContextMock;

    private $quoteMock;

    private $quoteRepositoryMock;

    private $changeQuoteControlMock;

    protected function setUp()
    {
        $this->userContextMock = $this->getMockBuilder(UserContextInterface::class)
            ->getMockForAbstractClass();
        $this->userContextMock->method('getUserId')
            ->willReturn(1);

        $this->quoteMock = $this->getMockBuilder(Quote::class)
            ->disableOriginalConstructor()
            ->setMethods(['getCustomerId'])
            ->getMock();

        $this->quoteRepositoryMock = $this->getMockBuilder(QuoteRepository::class)
            ->disableOriginalConstructor()
            ->getMock();

        $this->changeQuoteControlMock = $this->getMockBuilder(ChangeQuoteControl::class)
            ->disableOriginalConstructor()
            ->getMock();

        $objectManagerHelper = new ObjectManager($this);
        $this->accessChangeQuoteControl = $objectManagerHelper->getObject(
            AccessChangeQuoteControl::class,
            ['changeQuoteControl' => $this->changeQuoteControlMock]
        );
    }

    public function testBeforeSaveForCustomer()
    {
        $this->quoteMock->method('getCustomerId')
            ->willReturn(1);

        $this->userContextMock->method('getUserType')
            ->willReturn(UserContextInterface::USER_TYPE_CUSTOMER);

        $this->changeQuoteControlMock->method('isAllowed')
            ->willReturn(true);

        $result = $this->accessChangeQuoteControl->beforeSave($this->quoteRepositoryMock, $this->quoteMock);

        $this->assertNull($result);
    }

    public function testBeforeSaveException()
    {
        $this->quoteMock->method('getCustomerId')
            ->willReturn(2);

        $this->userContextMock->method('getUserType')
            ->willReturn(UserContextInterface::USER_TYPE_CUSTOMER);

        $this->changeQuoteControlMock->method('isAllowed')
            ->willReturn(false);

        $this->accessChangeQuoteControl->beforeSave($this->quoteRepositoryMock, $this->quoteMock);
    }

    public function testBeforeSaveForAdmin()
    {
        $this->quoteMock->method('getCustomerId')
            ->willReturn(2);

        $this->userContextMock->method('getUserType')
            ->willReturn(UserContextInterface::USER_TYPE_ADMIN);

        $this->changeQuoteControlMock->method('isAllowed')
            ->willReturn(true);

        $result = $this->accessChangeQuoteControl->beforeSave($this->quoteRepositoryMock, $this->quoteMock);

        $this->assertNull($result);
    }

    public function testBeforeSaveForGuest()
    {
        $this->quoteMock->method('getCustomerId')
            ->willReturn(null);

        $this->userContextMock->method('getUserType')
            ->willReturn(UserContextInterface::USER_TYPE_GUEST);

        $this->changeQuoteControlMock->method('isAllowed')
            ->willReturn(true);

        $result = $this->accessChangeQuoteControl->beforeSave($this->quoteRepositoryMock, $this->quoteMock);

        $this->assertNull($result);
    }

    public function testBeforeSaveForGuestException()
    {
        $this->quoteMock->method('getCustomerId')
            ->willReturn(1);

        $this->userContextMock->method('getUserType')
            ->willReturn(UserContextInterface::USER_TYPE_GUEST);

        $this->changeQuoteControlMock->method('isAllowed')
            ->willReturn(false);

        $this->accessChangeQuoteControl->beforeSave($this->quoteRepositoryMock, $this->quoteMock);
    }

    public function testBeforeSaveForUnknownUserTypeException()
    {
        $this->quoteMock->method('getCustomerId')
            ->willReturn(2);

        $this->userContextMock->method('getUserType')
            ->willReturn(10);

        $this->changeQuoteControlMock->method('isAllowed')
            ->willReturn(false);

        $this->accessChangeQuoteControl->beforeSave($this->quoteRepositoryMock, $this->quoteMock);
    }
}