paper-orm/tests/MigrationTest.php

<?php

namespace Test\PhpDevCommunity\PaperORM;

use PhpDevCommunity\PaperORM\Driver\MariaDBDriver;
use PhpDevCommunity\PaperORM\Driver\SqliteDriver;
use PhpDevCommunity\PaperORM\EntityManager;
use PhpDevCommunity\PaperORM\Mapper\ColumnMapper;
use PhpDevCommunity\PaperORM\Mapping\Column\IntColumn;
use PhpDevCommunity\PaperORM\Mapping\Column\StringColumn;
use PhpDevCommunity\PaperORM\Migration\PaperMigration;
use PhpDevCommunity\UniTester\TestCase;
use RuntimeException;
use Test\PhpDevCommunity\PaperORM\Entity\PostTest;
use Test\PhpDevCommunity\PaperORM\Entity\UserTest;
use Test\PhpDevCommunity\PaperORM\Helper\DataBaseHelperTest;

class MigrationTest extends TestCase
{
    private string $migrationDir;

    protected function setUp(): void
    {
        $this->migrationDir = __DIR__ . '/migrations';
        $this->tearDown();
    }

    protected function tearDown(): void
    {
        $folder = $this->migrationDir;
        array_map('unlink', glob("$folder/*.*"));
    }

    protected function execute(): void
    {
        foreach (DataBaseHelperTest::drivers() as $params) {
            $em = new EntityManager($params);
            $paperMigration = PaperMigration::create($em, 'mig_versions', $this->migrationDir);
            $platform = $em->getPlatform();
            $platform->createDatabaseIfNotExists();
            $platform->dropDatabase();
            $platform->createDatabaseIfNotExists();
            $this->testDiff($paperMigration);
            $this->testExecute($paperMigration);
            $this->testColumnModification($paperMigration);
            $this->testFailedMigration($paperMigration);
            $em->getConnection()->close();
            $this->tearDown();
        }
    }


    private function testDiff(PaperMigration $paperMigration): void
    {
        $em = $paperMigration->getEntityManager();
        $driver = $em->getConnection()->getDriver();
        $em->getConnection()->close();
        $migrationFile = $paperMigration->generateMigrationFromEntities([
            UserTest::class,
            PostTest::class
        ]);

        switch (get_class($driver)) {
            case SqliteDriver::class:
                $lines = file($migrationFile, FILE_IGNORE_NEW_LINES | FILE_SKIP_EMPTY_LINES);
                $this->assertEquals($lines, array(
                    0 => '-- UP MIGRATION --',
                    1 => 'CREATE TABLE `user` (`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,`firstname` VARCHAR(255) NOT NULL,`lastname` VARCHAR(255) NOT NULL,`email` VARCHAR(255) NOT NULL,`password` VARCHAR(255) NOT NULL,`is_active` BOOLEAN NOT NULL,`created_at` DATETIME,`last_post_id` INTEGER,FOREIGN KEY (`last_post_id`) REFERENCES `post` (id) ON DELETE SET NULL ON UPDATE NO ACTION);',
                    2 => 'CREATE UNIQUE INDEX IX_8D93D6492D053F64 ON `user` (`last_post_id`);',
                    3 => 'CREATE TABLE `post` (`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,`title` VARCHAR(255) NOT NULL,`content` VARCHAR(255) NOT NULL,`created_at` DATETIME NOT NULL,`user_id` INTEGER,FOREIGN KEY (`user_id`) REFERENCES `user` (id) ON DELETE SET NULL ON UPDATE NO ACTION);',
                    4 => 'CREATE INDEX IX_5A8A6C8DA76ED395 ON `post` (`user_id`);',
                    5 => '-- DOWN MIGRATION --',
                    6 => 'DROP INDEX IX_8D93D6492D053F64;',
                    7 => 'DROP TABLE `user`;',
                    8 => 'DROP INDEX IX_5A8A6C8DA76ED395;',
                    9 => 'DROP TABLE `post`;',
                ));
                break;
            case MariaDBDriver::class:
                $lines = file($migrationFile, FILE_IGNORE_NEW_LINES | FILE_SKIP_EMPTY_LINES);
                $this->assertEquals($lines, array(
                    0 => '-- UP MIGRATION --',
                    1 => 'CREATE TABLE `user` (`id` INT(11) AUTO_INCREMENT PRIMARY KEY NOT NULL,`firstname` VARCHAR(255) NOT NULL,`lastname` VARCHAR(255) NOT NULL,`email` VARCHAR(255) NOT NULL,`password` VARCHAR(255) NOT NULL,`is_active` TINYINT(1) NOT NULL,`created_at` DATETIME DEFAULT NULL,`last_post_id` INT(11) DEFAULT NULL);',
                    2 => 'CREATE UNIQUE INDEX IX_8D93D6492D053F64 ON `user` (`last_post_id`);',
                    3 => 'CREATE TABLE `post` (`id` INT(11) AUTO_INCREMENT PRIMARY KEY NOT NULL,`title` VARCHAR(255) NOT NULL,`content` VARCHAR(255) NOT NULL,`created_at` DATETIME NOT NULL,`user_id` INT(11) DEFAULT NULL);',
                    4 => 'CREATE INDEX IX_5A8A6C8DA76ED395 ON `post` (`user_id`);',
                    5 => 'ALTER TABLE `user` ADD CONSTRAINT FK_8D93D6492D053F64 FOREIGN KEY (`last_post_id`) REFERENCES `post`(`id`) ON DELETE SET NULL ON UPDATE NO ACTION;',
                    6 => 'ALTER TABLE `post` ADD CONSTRAINT FK_5A8A6C8DA76ED395 FOREIGN KEY (`user_id`) REFERENCES `user`(`id`) ON DELETE SET NULL ON UPDATE NO ACTION;',
                    7 => '-- DOWN MIGRATION --',
                    8 => 'ALTER TABLE `user` DROP FOREIGN KEY FK_8D93D6492D053F64;',
                    9 => 'ALTER TABLE `post` DROP FOREIGN KEY FK_5A8A6C8DA76ED395;',
                    10 => 'DROP INDEX IX_8D93D6492D053F64 ON `user`;',
                    11 => 'DROP TABLE `user`;',
                    12 => 'DROP INDEX IX_5A8A6C8DA76ED395 ON `post`;',
                    13 => 'DROP TABLE `post`;',
                ));
                break;
            default:
                throw new RuntimeException(sprintf('Driver %s not supported', get_class($driver)));
        }

    }

    private function testExecute(PaperMigration $paperMigration): void
    {
        $paperMigration->migrate();
        $successList = $paperMigration->getSuccessList();
        $this->assertTrue(count($successList) === 1);

        $migrationFile = $paperMigration->generateMigrationFromEntities([UserTest::class]);
        $this->assertNull($migrationFile);
    }

    private function testColumnModification(PaperMigration $paperMigration): void
    {
        $em = $paperMigration->getEntityManager();
        $driver = $em->getConnection()->getDriver();

        $userColumns = ColumnMapper::getColumns(UserTest::class);
        $userColumns[3] = (new StringColumn(null, 255, true, null, true))->bindProperty('email');
        $userColumns[] = (new IntColumn(null, false, 0))->bindProperty('childs');
        $migrationFile = $paperMigration->generateMigrationFromTables([
            'user' => [
                'columns' => $userColumns,
                'indexes' => []
            ]
        ]);
        $paperMigration->migrate();
        $successList = $paperMigration->getSuccessList();
        $this->assertTrue(count($successList) === 1);
        $this->assertEquals(pathinfo($migrationFile, PATHINFO_FILENAME), $successList[0]);
        switch (get_class($driver)) {
            case SqliteDriver::class:
                $schema = $driver->createDatabaseSchema();
                $lines = file($migrationFile, FILE_IGNORE_NEW_LINES | FILE_SKIP_EMPTY_LINES);
                if ($schema->supportsDropColumn()) {
                    $this->assertEquals($lines, array());
                } else {
                    $this->assertEquals($lines, array(
                        0 => '-- UP MIGRATION --',
                        1 => 'ALTER TABLE `user` ADD `childs` INTEGER NOT NULL DEFAULT 0;',
                        2 => 'CREATE UNIQUE INDEX IX_8D93D649E7927C74 ON `user` (`email`);',
                        3 => '-- Modify column email is not supported with PhpDevCommunity\\PaperORM\\Schema\\SqliteSchema. Consider creating a new column and migrating the data.;',
                        4 => '-- DOWN MIGRATION --',
                        5 => '-- Drop column childs is not supported with PhpDevCommunity\\PaperORM\\Schema\\SqliteSchema. You might need to manually drop the column.;',
                        6 => 'DROP INDEX IX_8D93D649E7927C74;',
                    ));
                }
                break;
            case MariaDBDriver::class:
                $lines = file($migrationFile, FILE_IGNORE_NEW_LINES | FILE_SKIP_EMPTY_LINES);
                $this->assertEquals($lines, array (
                    0 => '-- UP MIGRATION --',
                    1 => 'ALTER TABLE `user` ADD COLUMN `childs` INT(11) NOT NULL DEFAULT 0;',
                    2 => 'CREATE UNIQUE INDEX IX_8D93D649E7927C74 ON `user` (`email`);',
                    3 => 'ALTER TABLE `user` MODIFY COLUMN `email` VARCHAR(255) DEFAULT NULL;',
                    4 => '-- DOWN MIGRATION --',
                    5 => 'ALTER TABLE `user` DROP COLUMN `childs`;',
                    6 => 'DROP INDEX IX_8D93D649E7927C74 ON `user`;',
                    7 => 'ALTER TABLE `user` MODIFY COLUMN `email` VARCHAR(255) NOT NULL;',
                ));
                break;
            default:
                throw new RuntimeException(sprintf('Driver %s not supported', get_class($driver)));
        }

    }

    private function testFailedMigration(PaperMigration $paperMigration): void
    {
        $paperMigration->generateMigration();

        $this->expectException(RuntimeException::class, function () use ($paperMigration) {
            $paperMigration->migrate();
        });
        $successList = $paperMigration->getSuccessList();
        $this->assertTrue(count($successList) === 0);

    }
}