archived-data-fixtures/tests/Common/DataFixtures/BaseTestCase.php

<?php

declare(strict_types=1);

namespace Doctrine\Tests\Common\DataFixtures;

use Doctrine\DBAL\DriverManager;
use Doctrine\ORM\EntityManager;
use Doctrine\ORM\ORMSetup;
use PHPUnit\Framework\TestCase;

use function method_exists;

use const PHP_VERSION_ID;

abstract class BaseTestCase extends TestCase
{
    /**
     * EntityManager mock object together with
     * annotation mapping driver and pdo_sqlite
     * database in memory
     */
    protected function getMockSqliteEntityManager(string $fixtureSet = 'TestEntity'): EntityManager
    {
        $dbParams = ['driver' => 'sqlite3', 'memory' => true];
        if (PHP_VERSION_ID >= 80400 && method_exists(ORMSetup::class, 'createAttributeMetadataConfig')) {
            $config = ORMSetup::createAttributeMetadataConfig([__DIR__ . '/' . $fixtureSet], true);
            $config->enableNativeLazyObjects(true);
        } else {
            $config = ORMSetup::createAttributeMetadataConfiguration([__DIR__ . '/' . $fixtureSet], true);
            $config->setLazyGhostObjectEnabled(true);
        }

        $connection = DriverManager::getConnection($dbParams, $config);
        $platform   = $connection->getDatabasePlatform();
        if (method_exists($platform, 'disableSchemaEmulation')) {
            $platform->disableSchemaEmulation();
        }

        $connection->executeStatement('ATTACH DATABASE \':memory:\' AS readers');

        return new EntityManager($connection, $config);
    }
}