CanUseForCountryTest.php

Go to the documentation of this file.

<?php
namespace Magento\Payment\Test\Unit\Model\Checks;

use Magento\Payment\Model\Checks\CanUseForCountry;

class CanUseForCountryTest extends \PHPUnit\Framework\TestCase
{
    const EXPECTED_COUNTRY_ID = 1;

    protected $countryProvider;

    protected $_model;

    protected function setUp()
    {
        $this->countryProvider = $this->createMock(
            \Magento\Payment\Model\Checks\CanUseForCountry\CountryProvider::class
        );
        $this->_model = new CanUseForCountry($this->countryProvider);
    }

    public function testIsApplicable($expectation)
    {
        $quoteMock = $this->getMockBuilder(\Magento\Quote\Model\Quote::class)->disableOriginalConstructor()->setMethods(
            []
        )->getMock();

        $paymentMethod = $this->getMockBuilder(
            \Magento\Payment\Model\MethodInterface::class
        )->disableOriginalConstructor()->setMethods([])->getMock();
        $paymentMethod->expects($this->once())->method('canUseForCountry')->with(
            self::EXPECTED_COUNTRY_ID
        )->will($this->returnValue($expectation));
        $this->countryProvider->expects($this->once())->method('getCountry')->willReturn(self::EXPECTED_COUNTRY_ID);

        $this->assertEquals($expectation, $this->_model->isApplicable($paymentMethod, $quoteMock));
    }

    public function paymentMethodDataProvider()
    {
        return [[true], [false]];
    }
}