Loading fixtures registered as service (#32)

contributte · Jan 10, 2025 · 2d05568 · 2d05568
1 parent 3dba4a3
commit 2d05568
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/Command/LoadDataFixturesCommand.php b/src/Command/LoadDataFixturesCommand.php
@@ -95,8 +95,8 @@ protected function execute(InputInterface $input, OutputInterface $output): int
 		}
 
 		$fixtures = $this->fixturesLoader->getFixtures();
-		if ($fixtures === []) {
 
+		if ($fixtures === []) {
 			if ($paths === []) {
 				throw new LogicalException('Could not find any fixtures to load.');
 			} else {

diff --git a/src/DI/FixturesExtension.php b/src/DI/FixturesExtension.php
@@ -2,7 +2,9 @@
 
 namespace Nettrine\Fixtures\DI;
 
+use Doctrine\Common\DataFixtures\FixtureInterface;
 use Nette\DI\CompilerExtension;
+use Nette\DI\Definitions\ServiceDefinition;
 use Nette\Schema\Expect;
 use Nette\Schema\Schema;
 use Nettrine\Fixtures\Command\LoadDataFixturesCommand;
@@ -35,4 +37,18 @@ public function loadConfiguration(): void
 			->addTag('console.command', 'doctrine:fixtures:load');
 	}
 
+	public function beforeCompile(): void
+	{
+		$builder = $this->getContainerBuilder();
+
+		$fixtures = $builder->findByType(FixtureInterface::class);
+
+		/** @var ServiceDefinition $fixtureLoader */
+		$fixtureLoader = $builder->getDefinitionByType(FixturesLoader::class);
+
+		foreach ($fixtures as $fixture) {
+			$fixtureLoader->addSetup('addFixture', [$fixture]);
+		}
+	}
+
 }