fjbender / magento-2

use Payone\Core\Test\Unit\BaseTestCase;

use Payone\Core\Model\Test\PayoneObjectManager;

class InstallSchemaTest extends BaseTestCase

{

    /**

     * @var ClassToTest

     */

    private $classToTest;

    /**

     * @var ObjectManager|PayoneObjectManager

     */

    private $objectManager;

    protected function setUp()

    {

        $this->objectManager = $this->getObjectManager();

        $this->classToTest = $this->objectManager->getObject(ClassToTest::class);

    }

    public function testInstall()

    {

        $table = $this->getMockBuilder(Table::class)->disableOriginalConstructor()->getMock();

        $connection = $this->getMockBuilder(AdapterInterface::class)->disableOriginalConstructor()->getMock();

        $connection->method('isTableExists')->willReturn(false);

        $connection->method('newTable')->willReturn($table);

        $installer = $this->getMockBuilder(SchemaSetupInterface::class)->disableOriginalConstructor()->getMock();

        $installer->method('getConnection')->willReturn($connection);

        $installer->method('getTable')->willReturn('table');

        $installer->method('getIdxName')->willReturn('name');

        $context = $this->getMockBuilder(ModuleContextInterface::class)->disableOriginalConstructor()->getMock();

        $context->method('getVersion')->willReturn('1.1.0');

        $result = $this->classToTest->install($installer, $context);

        $this->assertNull($result);

    }

}

use Payone\Core\Test\Unit\BaseTestCase;

use Payone\Core\Model\Test\PayoneObjectManager;

class UpgradeSchemaTest extends BaseTestCase

{

    /**

     * @var ClassToTest

     */

    private $classToTest;

    /**

     * @var ObjectManager|PayoneObjectManager

     */

    private $objectManager;

    protected function setUp()

    {

        $this->objectManager = $this->getObjectManager();

        $this->classToTest = $this->objectManager->getObject(ClassToTest::class);

    }

    public function testUpgrade()

    {

        $table = $this->getMockBuilder(Table::class)->disableOriginalConstructor()->getMock();

        $connection = $this->getMockBuilder(AdapterInterface::class)->disableOriginalConstructor()->getMock();

        $connection->method('isTableExists')->willReturn(false);

        $connection->method('newTable')->willReturn($table);

        $setup = $this->getMockBuilder(SchemaSetupInterface::class)->disableOriginalConstructor()->getMock();

        $setup->method('getConnection')->willReturn($connection);

        $setup->method('getTable')->willReturn('table');

        $setup->method('getIdxName')->willReturn('name');

        $context = $this->getMockBuilder(ModuleContextInterface::class)->disableOriginalConstructor()->getMock();

        $result = $this->classToTest->upgrade($setup, $context);

        $this->assertNull($result);

    }

}