Allow custom services to be lazy

Author: goetas

DependencyInjection/DoctrineMigrationsExtension.php

@@ -9,6 +9,7 @@
 use InvalidArgumentException;
 use RuntimeException;
 use Symfony\Component\Config\FileLocator;
+use Symfony\Component\DependencyInjection\Argument\ServiceClosureArgument;
 use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\DependencyInjection\Definition;
 use Symfony\Component\DependencyInjection\Loader\XmlFileLoader;
@@ -67,7 +68,7 @@ public function load(array $configs, ContainerBuilder $container) : void
         $diDefinition = $container->getDefinition('doctrine.migrations.dependency_factory');
 
         foreach ($config['services'] as $doctrineId => $symfonyId) {
-            $diDefinition->addMethodCall('setService', [$doctrineId, new Reference($symfonyId)]);
+            $diDefinition->addMethodCall('setDefinition', [$doctrineId, new ServiceClosureArgument(new Reference($symfonyId))]);
         }
 
         if (! isset($config['services'][MetadataStorage::class])) {

Tests/DependencyInjection/DoctrineMigrationsExtensionTest.php

@@ -16,6 +16,7 @@
 use Doctrine\Migrations\Version\Comparator;
 use Doctrine\Migrations\Version\Version;
 use Doctrine\ORM\EntityManager;
+use Exception;
 use InvalidArgumentException;
 use PHPUnit\Framework\TestCase;
 use Symfony\Component\Config\Definition\Exception\InvalidConfigurationException;
@@ -25,6 +26,7 @@
 use Symfony\Component\DependencyInjection\Definition;
 use Symfony\Component\DependencyInjection\Loader\XmlFileLoader;
 use Symfony\Component\DependencyInjection\ParameterBag\ParameterBag;
+use Symfony\Component\DependencyInjection\Reference;
 use function assert;
 use function method_exists;
 use function sys_get_temp_dir;
@@ -175,6 +177,34 @@ public function compare(Version $a, Version $b) : int
         self::assertSame($sorter, $di->getVersionComparator());
     }
 
+    public function testServicesAreLazy() : void
+    {
+        $config    = [
+            'services' => [Comparator::class => 'my_sorter'],
+        ];
+        $container = $this->getContainer($config);
+
+        $conn = $this->createMock(Connection::class);
+        $container->set('doctrine.dbal.default_connection', $conn);
+
+        $sorterFactory = new class() {
+            public function __invoke() : void
+            {
+                throw new Exception('This method should not be invoked.');
+            }
+        };
+        $container->set('my_sorter_factory', $sorterFactory);
+
+        $sorterDefinition = new Definition(Comparator::class);
+        $sorterDefinition->setFactory(new Reference('my_sorter_factory'));
+        $container->setDefinition('my_sorter', $sorterDefinition);
+
+        $container->compile();
+
+        $di = $container->get('doctrine.migrations.dependency_factory');
+        self::assertInstanceOf(DependencyFactory::class, $di);
+    }
+
     public function testCustomConnection() : void
     {
         $config    = [

composer.json

@@ -43,6 +43,11 @@
     "autoload-dev": {
         "psr-4": { "Doctrine\\Bundle\\MigrationsBundle\\Tests\\": "Tests" }
     },
+    "config": {
+        "platform": {
+            "php": "7.2.5"
+        }
+    },
     "extra": {
         "branch-alias": {
           "dev-master": "3.0.x-dev"