Ynlo\GraphQLBundle\Test\FixtureLoader\DataPopulator\ORMDataLoader

Complexity

Total Complexity

6

Size/Duplication

Total Lines	32
Duplicated Lines	0 %

Test Coverage

Coverage

92.31%

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	supports()	0	3	1
A	__construct()	0	3	1
A	createExecutor()	0	12	4

<?php
/*******************************************************************************
 *  This file is part of the GraphQL Bundle package.
 *
 *  (c) YnloUltratech <support@ynloultratech.com>
 *
 *  For the full copyright and license information, please view the LICENSE
 *  file that was distributed with this source code.
 ******************************************************************************/

namespace Ynlo\GraphQLBundle\Test\FixtureLoader\DataPopulator;

use Doctrine\Bundle\DoctrineBundle\Registry;
use Doctrine\Common\DataFixtures\Executor\AbstractExecutor;
use Doctrine\Common\DataFixtures\Executor\ORMExecutor;
use Doctrine\Common\DataFixtures\Purger\ORMPurger;
use Doctrine\DBAL\Connection;
use Doctrine\DBAL\Driver;

/**
 * Class ORMDataLoader
 */
class ORMDataLoader implements DataLoaderInterface
{
    protected $cacheDir;

    public function __construct($cacheDir = null)
    {
        $this->cacheDir = $cacheDir;
    }

    /**
     * {@inheritdoc}
     */
    public function supports(Registry $registry): bool
    {
        return $registry->getName() === 'ORM';
    }

    /**
     * {@inheritdoc}
     */
    public function createExecutor(Registry $registry): AbstractExecutor
    {
        $om = $registry->getManager();
        $purger = new ORMPurger();

        if (($connection = $registry->getConnection())
            && $connection instanceof Connection
            && $connection->getDriver() instanceof Driver\AbstractSQLiteDriver) {
            return new SQLiteORMExecutor($om, $purger, $this->cacheDir);
        }

        return new ORMExecutor($om, $purger);
    }
}