CustomerTokenManagementTest.php 2.81 KB
Newer Older
Ketan's avatar
Ketan committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96
<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
namespace Magento\Vault\Test\Unit\Model;

use Magento\Customer\Model\Session;
use Magento\Vault\Api\Data\PaymentTokenInterface;
use Magento\Vault\Model\CustomerTokenManagement;
use Magento\Vault\Model\PaymentTokenManagement;
use PHPUnit_Framework_MockObject_MockObject as MockObject;

class CustomerTokenManagementTest extends \PHPUnit\Framework\TestCase
{
    /**
     * @var PaymentTokenManagement|MockObject
     */
    private $paymentTokenManagement;

    /**
     * @var Session|MockObject
     */
    private $customerSession;

    /**
     * @var CustomerTokenManagement
     */
    private $tokenManagement;

    protected function setUp()
    {
        $this->paymentTokenManagement = $this->getMockBuilder(PaymentTokenManagement::class)
            ->disableOriginalConstructor()
            ->getMock();
        $this->customerSession = $this->getMockBuilder(Session::class)
            ->disableOriginalConstructor()
            ->getMock();

        $this->tokenManagement = new CustomerTokenManagement(
            $this->paymentTokenManagement,
            $this->customerSession
        );
    }

    /**
     * @param int|null $customerId
     * @param bool $isLoggedCustomer
     * @return void
     * @dataProvider getCustomerSessionTokensNegativeDataProvider
     */
    public function testGetCustomerSessionTokensNegative($customerId, bool $isLoggedCustomer)
    {
        $this->customerSession->method('getCustomerId')->willReturn($customerId);
        $this->customerSession->method('isLoggedIn')->willReturn($isLoggedCustomer);
        $this->paymentTokenManagement->expects(static::never())->method('getVisibleAvailableTokens');

        static::assertEquals([], $this->tokenManagement->getCustomerSessionTokens());
    }

    /**
     * @return array
     */
    public function getCustomerSessionTokensNegativeDataProvider()
    {
        return [
            'not registered customer' => [null, false],
            'not logged in customer' => [1, false],
        ];
    }

    public function testGetCustomerSessionTokens()
    {
        $customerId = 1;
        $token = $this->createMock(PaymentTokenInterface::class);
        $expectation = [$token];

        $this->customerSession->expects(static::once())
            ->method('getCustomerId')
            ->willReturn($customerId);

        $this->customerSession->expects(static::once())
            ->method('isLoggedIn')
            ->willReturn(true);

        $this->paymentTokenManagement->expects(static::once())
            ->method('getVisibleAvailableTokens')
            ->with($customerId)
            ->willReturn($expectation);

        static::assertEquals(
            $expectation,
            $this->tokenManagement->getCustomerSessionTokens()
        );
    }
}