codecept\Helper\CleanDoctrine2

Complexity

Total Complexity

5

Size/Duplication

Total Lines	71
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	8

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	_before()	0	18	2
A	_after()	0	13	2
A	retrieveEntityManager()	0	7	1

<?php
declare(strict_types=1);

namespace codecept\Helper;

use Codeception\TestInterface;
use Doctrine\ORM\Tools\SchemaTool;
use SlayerBirden\DataFlowServer\Doctrine\Persistence\EntityManagerRegistry;
use SlayerBirden\DataFlowServer\Doctrine\SimpleRegistry;

class CleanDoctrine2 extends \Codeception\Module\Doctrine2
{
    protected $dependencyMessage = <<<EOF
Set a dependent module:

modules:
    enabled:
        - CleanDoctrine2:
            depends: ZendExpressive3
EOF;

    /**
     * @var EntityManagerRegistry
     */
    public $registry;

    /**
     * @inheritdoc
     * @param TestInterface $test
     * @throws \Doctrine\DBAL\DBALException
     * @throws \Doctrine\ORM\ORMException
     * @throws \Codeception\Exception\ModuleConfigException
     */
    public function _before(TestInterface $test)
    {
        $this->retrieveEntityManager();
        if ($this->config['cleanup']) {
            $schemaManager = $this->em->getConnection()->getSchemaManager();
            $currentSchema = $schemaManager->createSchema();

            $metadatas = $this->em->getMetadataFactory()->getAllMetadata();
            $expectedSchema = (new SchemaTool($this->em))->getSchemaFromMetadata($metadatas);

            $sql = $currentSchema->getMigrateToSql($expectedSchema, $this->em->getConnection()->getDatabasePlatform());
            array_walk($sql, function ($script) {
                $this->em->getConnection()->executeUpdate($script);
            });

            $this->debugSection('Database', 'Database Created');
        }
    }

    /**
     * @inheritdoc
     * @throws \Codeception\Exception\ModuleConfigException
     */
    public function _after(TestInterface $test)
    {
        $this->retrieveEntityManager();
        if ($this->config['cleanup']) {
            $metadatas = $this->em->getMetadataFactory()->getAllMetadata();
            (new SchemaTool($this->em))->dropSchema($metadatas);

            $this->debugSection('Database', 'Database Is Dropped!');
        }

        $this->clean();
        $this->em->getConnection()->close();
    }

    /**
     * @throws \Codeception\Exception\ModuleConfigException
     */
    protected function retrieveEntityManager()
    {
        parent::retrieveEntityManager();
        $this->registry = new SimpleRegistry([], [
            SimpleRegistry::DEFAULT_MANAGER_NAME => $this->em,
        ]);
    }
}