Phraseanet/tests/Alchemy/Tests/Phrasea/Setup/Version/PreSchemaUpgrade/Upgrade39UsersTest.php

<?php

namespace Alchemy\Tests\Phrasea\Setup\Version;

use Alchemy\Phrasea\Model\Entities\Session;
use Alchemy\Phrasea\Setup\Version\PreSchemaUpgrade\Upgrade39Users;
use Alchemy\Phrasea\Model\Entities\User;
use Doctrine\DBAL\DBALException;
use Doctrine\DBAL\Migrations\Configuration\YamlConfiguration;
use Doctrine\ORM\EntityManager;
use Doctrine\ORM\Query\ResultSetMapping;
use Doctrine\ORM\Tools\SchemaTool;
use Doctrine\DBAL\Types\Type;

/**
 * @group functional
 * @group legacy
 */
class Upgrade39UsersTest extends \PhraseanetTestCase
{
    const DB_NAME = 'update39_test';

    /**
     * @dataProvider provideVariousFixtures
     */
    public function testApply($fixture)
    {
        $em = $this->loadFixture($fixture);

        $upgrader = new Upgrade39Users();
        $configuration = new YamlConfiguration($em->getConnection());

        $configuration->load(__DIR__.'/../../../../../../../lib/conf.d/migrations.yml');
        $configuration->setMigrationsDirectory(__DIR__.'/../../../../../../../lib/Alchemy/Phrasea/Setup/DoctrineMigration');

        $upgrader->apply($em, $this->createAppboxMock(), $configuration);

        $this->assertUsrTableIsSanitized($em);
        // check usr_ids are preserved
        $this->assertUsridsArePreserved($em);

        $this->checkThatNewConstraintsCanBeApplied($em);

        $this->assertLastAppliedModelAreOk($em);
        $this->assertTemplateAreOwnedByValidUser($em);

        $this->assertEquals('onner316269684', $this->loadUser($em, 30)->getLogin());
        $this->assertTrue($this->loadUser($em, 30)->isDeleted());
    }

    private function assertLastAppliedModelAreOk(EntityManager $em)
    {
        // check update
        $this->assertEquals(176, $this->loadUser($em, 188)->getLastAppliedTemplate()->getId());
        // last_model does not exist
        $this->assertNull($this->loadUser($em, 105)->getLastAppliedTemplate());
        // last_model is a deleted user
        $this->assertNull($this->loadUser($em, 36)->getLastAppliedTemplate());
        // no last_model
        $this->assertNull($this->loadUser($em, 4)->getLastAppliedTemplate());
    }

    private function assertTemplateAreOwnedByValidUser(EntityManager $em)
    {
        // check update
        $this->assertEquals(109, $this->loadUser($em, 160)->getTemplateOwner()->getId());
        // owner does not exist
        $this->assertNull($this->loadUser($em, 12));
        // owner has been deleted
        $this->assertNull($this->loadUser($em, 31));
        // no owner
        $this->assertNull($this->loadUser($em, 11)->getTemplateOwner());
    }

    /**
     * @return User
     */
    private function loadUser(EntityManager $em, $id)
    {
        return $em->find('Phraseanet:User', $id);
    }

    private function checkThatNewConstraintsCanBeApplied(EntityManager $em)
    {
        $tool = new SchemaTool($em);
        $metas = $em->getMetadataFactory()->getAllMetadata();

        // Bad ordering of Indexes changes in Doctrine DBAL-2.5
        $sqls = $tool->getUpdateSchemaSql($metas, true);
        foreach ($sqls as $sql) {
            if ('DROP INDEX usr_id ON Sessions' === $sql) {
                $em->getConnection()->executeQuery($sql);
            }
        }
        // End patching DBAL-2.5

        $tool->updateSchema($metas, true);
    }

    private function assertUsrTableIsSanitized(EntityManager $em)
    {
        $rs = $em->createNativeQuery(
            "SHOW FIELDS FROM usr_backup WHERE Field = 'usr_login';",
            (new ResultSetMapping())->addScalarResult('Type', 'Type')
        )->getSingleResult();

        $this->assertSame('varchar(128)', strtolower($rs['Type']));
    }

    private function assertUsridsArePreserved(EntityManager $em)
    {
        $sql = 'SELECT usr_id FROM usr_backup o, Users n WHERE o.usr_login = n.login AND o.usr_id = n.id AND o.usr_login NOT LIKE "(#deleted_%" AND n.deleted = 0';
        $total = $em->getConnection()->executeQuery($sql)->rowCount();

        $sql = 'SELECT usr_id FROM usr_backup WHERE usr_login NOT LIKE "(#deleted_%"';
        $expected = $em->getConnection()->executeQuery($sql)->rowCount();

        $this->assertGreaterThan(0, $total);
        // 1 template has lost its owner
        $this->assertEquals($total, $expected - 2);
    }

    public function provideVariousFixtures()
    {
        return [
            ['tests/fixtures/update39_fixtureFrom38.sql'],
            ['tests/fixtures/update39_fixtureFrom31.sql'],
        ];
    }

    private function loadFixture($fixture)
    {
        $em = $this->createEntityManager(null);
        try {
            $em->getConnection()->executeQuery('DROP database '.self::DB_NAME);
        } catch (DBALException $e) {

        }

        $em->getConnection()->executeQuery('CREATE DATABASE '.self::DB_NAME.' CHARACTER SET utf8 COLLATE utf8_general_ci');
        $em = $this->createEntityManager();
        $em->getConnection()->executeQuery(file_get_contents(self::$DI['cli']['root.path'].'/'.$fixture));
        return $em;
    }

    private function createEntityManager($dbname = self::DB_NAME)
    {
        $app = self::$DI['cli'];
        $params = $app['db.appbox.info'];
        $params['dbname'] = $dbname;

        $info = $app['db.info']($params);
        $key = $app['orm.add']($info);

        $em = $app['orm.ems'][$key];

        return $em;
    }
}