fjbender / magento-2

use Payone\Core\Test\Unit\BaseTestCase;

use Payone\Core\Model\Test\PayoneObjectManager;

class KlarnaCountryTest extends BaseTestCase

{

    /**

     * @var ClassToTest

     */

    private $classToTest;

    protected function setUp()

    {

        $objectManager = $this->getObjectManager();

        $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 Payone\Core\Test\Unit\BaseTestCase;

use Payone\Core\Model\Test\PayoneObjectManager;

class SepaCountryTest extends BaseTestCase

{

    /**

     * @var ClassToTest

     */

    private $classToTest;

    protected function setUp()

    {

        $objectManager = $this->getObjectManager();

        $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);

    }

}