Fix proxy interface for ProxyManager or Symfony LazyGhost

Author: GromNaN

src/DependencyInjection/DoctrineMongoDBExtension.php

@@ -487,7 +487,7 @@ public function load(array $configs, ContainerBuilder $container): void
         }
 
         $container->getDefinition('doctrine_mongodb')
-            ->setArgument(5, $config['enable_lazy_ghost_objects'] ? LazyLoadingInterface::class : Proxy::class);
+            ->setArgument(5, $config['enable_lazy_ghost_objects'] ? Proxy::class : LazyLoadingInterface::class);
 
         // load the connections
         $this->loadConnections($config['connections'], $container, $config);

tests/DependencyInjection/DoctrineMongoDBExtensionTest.php

@@ -10,10 +10,14 @@
 use Doctrine\Bundle\MongoDBBundle\Attribute\MapDocument;
 use Doctrine\Bundle\MongoDBBundle\Command\Encryption\DiagnosticCommand;
 use Doctrine\Bundle\MongoDBBundle\Command\Encryption\DumpFieldsMapCommand;
+use Doctrine\Bundle\MongoDBBundle\DependencyInjection\Compiler\CreateProxyDirectoryPass;
 use Doctrine\Bundle\MongoDBBundle\DependencyInjection\Compiler\ServiceRepositoryCompilerPass;
 use Doctrine\Bundle\MongoDBBundle\DependencyInjection\DoctrineMongoDBExtension;
+use Doctrine\Bundle\MongoDBBundle\ManagerRegistry;
+use Doctrine\Bundle\MongoDBBundle\Tests\DependencyInjection\Fixtures\Bundles\AttributesBundle\Document\TestDocument;
 use Doctrine\Bundle\MongoDBBundle\Tests\DependencyInjection\Fixtures\Bundles\DocumentListenerBundle\EventListener\TestAttributeListener;
 use Doctrine\ODM\MongoDB\Configuration;
+use Doctrine\ODM\MongoDB\DocumentManager;
 use Doctrine\ODM\MongoDB\Mapping\Annotations;
 use InvalidArgumentException;
 use MongoDB\Client;
@@ -22,6 +26,7 @@
 use stdClass;
 use Symfony\Component\DependencyInjection\Alias;
 use Symfony\Component\DependencyInjection\ChildDefinition;
+use Symfony\Component\DependencyInjection\Compiler\ResolveParameterPlaceHoldersPass;
 use Symfony\Component\DependencyInjection\Container;
 use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\DependencyInjection\Definition;
@@ -716,6 +721,55 @@ public function testAutoEncryptionMinimumODMVersion(): void
         $loader->load([$config], $container);
     }
 
+    #[DataProvider('provideLazyObjectConfigurations')]
+    public function testRegistryGetManagerForClass(array $config): void
+    {
+        $container = $this->getContainer('AttributesBundle');
+        $loader    = new DoctrineMongoDBExtension();
+        $loader->load(self::buildConfiguration($config + [
+            'connections' => [
+                'default' => [],
+            ],
+            'document_managers' => [
+                'default' => [
+                    'mappings' => [
+                        'AttributesBundle' => ['type' => 'attribute'],
+                    ],
+                ],
+            ],
+        ]), $container);
+        (new ResolveParameterPlaceHoldersPass())->process($container);
+        (new ServiceRepositoryCompilerPass())->process($container);
+        (new CreateProxyDirectoryPass())->process($container);
+
+        $container->compile();
+        $registry = $container->get('doctrine_mongodb');
+        $dm       = $container->get('doctrine_mongodb.odm.document_manager');
+
+        self::assertInstanceOf(ManagerRegistry::class, $registry);
+        self::assertInstanceOf(DocumentManager::class, $dm);
+
+        // Create a lazy object
+        $ref = $dm->getReference(TestDocument::class, 'some');
+        self::assertInstanceOf(TestDocument::class, $ref);
+        self::assertSame($dm, $registry->getManagerForClass($ref::class));
+    }
+
+    public static function provideLazyObjectConfigurations(): iterable
+    {
+        yield 'Proxy Manager' => [
+            ['enable_lazy_ghost_objects' => false, 'enable_native_lazy_objects' => false],
+        ];
+
+        yield 'Symfony Lazy Objects' => [
+            ['enable_lazy_ghost_objects' => true, 'enable_native_lazy_objects' => false],
+        ];
+
+        yield 'Native Lazy Objects' => [
+            ['enable_lazy_ghost_objects' => false, 'enable_native_lazy_objects' => true],
+        ];
+    }
+
     private static function requireAutoEncryptionSupportInODM(): void
     {
         if (! InstalledVersions::satisfies(new VersionParser(), 'doctrine/mongodb-odm', '>=2.12@dev')) {

tests/DependencyInjection/Fixtures/Bundles/AttributesBundle/Document/TestDocument.php

@@ -9,4 +9,9 @@
 #[MongoDB\Document]
 class TestDocument
 {
+    #[MongoDB\Id]
+    public ?string $id = null;
+
+    #[MongoDB\Field(type: 'string')]
+    public ?string $name = null;
 }