fjbender / magento-2

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

use Payone\Core\Helper\Country;

class KlarnaCountryTest extends \PHPUnit_Framework_TestCase

{

    /**

     * @var ClassToTest

     */

    private $classToTest;

    protected function setUp()

    {

        $objectManager = new ObjectManager($this);

        $countryHelper = $this->getMockBuilder(Country::class)->disableOriginalConstructor()->getMock();

        $countryHelper->method('getCountryNameByIso2')->willReturn('Country Name');

        $this->classToTest = $objectManager->getObject(ClassToTest::class, [

            'countryHelper' => $countryHelper

        ]);

    }

    public function testToOptionArray()

    {

        $result = $this->classToTest->toOptionArray();

        $this->assertNotEmpty($result);

    }

}

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

use Payone\Core\Helper\Country;

class SepaCountryTest extends \PHPUnit_Framework_TestCase

{

    /**

     * @var ClassToTest

     */

    private $classToTest;

    protected function setUp()

    {

        $objectManager = new ObjectManager($this);

        $countryHelper = $this->getMockBuilder(Country::class)->disableOriginalConstructor()->getMock();

        $countryHelper->method('getCountryNameByIso2')->willReturn('Country Name');

        $this->classToTest = $objectManager->getObject(ClassToTest::class, [

            'countryHelper' => $countryHelper

        ]);

    }

    public function testToOptionArray()

    {

        $result = $this->classToTest->toOptionArray();

        $this->assertNotEmpty($result);

    }

}