Merge pull request #4 from kuldippujara/master

added use fixture loader option
Cocolabs-SAS · Mar 3, 2022 · 4447796 · 4447796
2 parents cdf64b5 + b0fe239
commit 4447796
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/VIPSoft/DoctrineDataFixturesExtension/Extension.php b/src/VIPSoft/DoctrineDataFixturesExtension/Extension.php
@@ -55,6 +55,9 @@ public function configure(ArrayNodeDefinition $builder)
                 ->arrayNode('fixtures')
                     ->prototype('scalar')->end()
                 ->end()
+                ->booleanNode('use_fixture_loader')
+                    ->defaultValue(false)
+                ->end()
                 ->scalarNode('lifetime')
                     ->defaultValue('feature')
                     ->validate()
@@ -90,6 +93,7 @@ public function load(ContainerBuilder $container, array $config)
         $container->setParameter('behat.doctrine_data_fixtures.lifetime', $config['lifetime']);
         $container->setParameter('behat.doctrine_data_fixtures.migrations', $config['migrations']);
         $container->setParameter('behat.doctrine_data_fixtures.use_backup', $config['use_backup']);
+        $container->setParameter('behat.doctrine_data_fixtures.use_fixture_loader', $config['use_fixture_loader']);
     }
 
     /**

diff --git a/src/VIPSoft/DoctrineDataFixturesExtension/Service/FixtureService.php b/src/VIPSoft/DoctrineDataFixturesExtension/Service/FixtureService.php
@@ -59,6 +59,7 @@ public function __construct(ContainerInterface $container, Kernel $kernel)
         $this->directories = $container->getParameter('behat.doctrine_data_fixtures.directories');
         $this->migrations = $container->getParameter('behat.doctrine_data_fixtures.migrations');
         $this->useBackup = $container->getParameter('behat.doctrine_data_fixtures.use_backup');
+        $this->useFixtureLoader = $container->getParameter('behat.doctrine_data_fixtures.use_fixture_loader');
         $this->kernel = $kernel;
 
         if ($this->useBackup) {
@@ -100,8 +101,13 @@ private function init()
      */
     private function getFixtureLoader()
     {
+
         $container = $this->kernel->getContainer();
 
+        if ($this->useFixtureLoader && $container->get('test.service_container')->has('doctrine.fixtures.loader')) {
+            return $container->get('test.service_container')->get('doctrine.fixtures.loader');
+        }
+
         $loader = class_exists(FixtureLoader::class)
             ? new FixtureLoader($container)
             : (class_exists('Symfony\Bridge\Doctrine\DataFixtures\ContainerAwareLoader')