Allow resetting lazy managers on Symfony 6.2

Registry.php

@@ -8,6 +8,7 @@
 use ProxyManager\Proxy\LazyLoadingInterface;
 use Psr\Container\ContainerInterface;
 use Symfony\Bridge\Doctrine\ManagerRegistry;
+use Symfony\Component\VarExporter\LazyGhostObjectInterface;
 use Symfony\Contracts\Service\ResetInterface;
 
 use function array_keys;
@@ -75,7 +76,7 @@ private function resetOrClearManager(string $managerName, string $serviceId): vo
 
         assert($manager instanceof EntityManagerInterface);
 
-        if (! $manager instanceof LazyLoadingInterface || $manager->isOpen()) {
+        if ((! $manager instanceof LazyLoadingInterface && ! $manager instanceof LazyGhostObjectInterface) || $manager->isOpen()) {
             $manager->clear();
 
             return;

Tests/LazyGhostObjectEntityManagerInterface.php

@@ -0,0 +1,10 @@
+<?php
+
+namespace Doctrine\Bundle\DoctrineBundle\Tests;
+
+use Doctrine\ORM\EntityManagerInterface;
+use Symfony\Component\VarExporter\LazyGhostObjectInterface;
+
+interface LazyGhostObjectEntityManagerInterface extends LazyGhostObjectInterface, EntityManagerInterface
+{
+}

Tests/RegistryTest.php

@@ -13,6 +13,7 @@
 use ProxyManager\Proxy\ProxyInterface;
 use stdClass;
 use Symfony\Component\DependencyInjection\ContainerInterface;
+use Symfony\Component\VarExporter\LazyGhostObjectInterface;
 
 use function assert;
 use function interface_exists;
@@ -169,6 +170,33 @@ public function testReset(): void
         $registry->reset();
     }
 
+    public function testResetGhostObject(): void
+    {
+        if (! interface_exists(EntityManagerInterface::class) || ! interface_exists(LazyGhostObjectInterface::class)) {
+            self::markTestSkipped('This test requires ORM and VarExporter 6.2+');
+        }
+
+        /** @psalm-suppress MissingDependency https://github.com/vimeo/psalm/issues/8258 */
+        $ghostManager = $this->createMock(LazyGhostObjectEntityManagerInterface::class);
+        /** @psalm-suppress MissingDependency https://github.com/vimeo/psalm/issues/8258 */
+        $ghostManager->expects($this->once())->method('resetLazyGhostObject')->willReturn(true);
+
+        $container = $this->createMock(ContainerInterface::class);
+        $container->method('initialized')
+            ->withConsecutive(['doctrine.orm.uninitialized_entity_manager'], ['doctrine.orm.ghost_entity_manager'])
+            ->willReturnOnConsecutiveCalls(false, true, true);
+        $container->method('get')
+            ->withConsecutive(['doctrine.orm.ghost_entity_manager'], ['doctrine.orm.ghost_entity_manager'], ['doctrine.orm.ghost_entity_manager'])
+            ->willReturnOnConsecutiveCalls($ghostManager, $ghostManager, $ghostManager);
+
+        $entityManagers = [
+            'uninitialized' => 'doctrine.orm.uninitialized_entity_manager',
+            'ghost' => 'doctrine.orm.ghost_entity_manager',
+        ];
+
+        (new Registry($container, [], $entityManagers, 'default', 'default'))->reset();
+    }
+
     public function testIdentityMapsStayConsistentAfterReset(): void
     {
         if (! interface_exists(EntityManagerInterface::class)) {
@@ -183,7 +211,7 @@ public function testIdentityMapsStayConsistentAfterReset(): void
         $entityManager = $container->get('doctrine.orm.default_entity_manager');
         $repository    = $entityManager->getRepository(TestCustomClassRepoEntity::class);
 
-        $this->assertInstanceOf(ProxyInterface::class, $entityManager);
+        $this->assertInstanceOf(interface_exists(LazyGhostObjectInterface::class) ? LazyGhostObjectInterface::class : ProxyInterface::class, $entityManager);
         assert($entityManager instanceof EntityManagerInterface);
         assert($registry instanceof Registry);
         assert($repository instanceof TestCustomClassRepoRepository);

psalm.xml.dist

@@ -43,6 +43,7 @@
             <errorLevel type="suppress">
                 <!-- We use the "Foo" namespace in unit tests. We are aware that those classes don't exist. -->
                 <referencedClass name="Foo\*"/>
+                <referencedClass name="Symfony\Component\VarExporter\LazyGhostObjectInterface"/>
             </errorLevel>
         </UndefinedClass>
         <UndefinedDocblockClass>