Make manager registry resettable

src/ManagerRegistry.php

@@ -6,12 +6,16 @@
 
 use Doctrine\ODM\MongoDB\DocumentManager;
 use Doctrine\ODM\MongoDB\MongoDBException;
+use ProxyManager\Proxy\LazyLoadingInterface;
 use Psr\Container\ContainerInterface;
 use Symfony\Bridge\Doctrine\ManagerRegistry as BaseManagerRegistry;
+use Symfony\Component\VarExporter\LazyObjectInterface;
+use Symfony\Contracts\Service\ResetInterface;
 
 use function array_keys;
+use function assert;
 
-class ManagerRegistry extends BaseManagerRegistry
+class ManagerRegistry extends BaseManagerRegistry implements ResetInterface
 {
     public function __construct(string $name, array $connections, array $managers, string $defaultConnection, string $defaultManager, string $proxyInterfaceName, ?ContainerInterface $container = null)
     {
@@ -42,4 +46,37 @@ public function getAliasNamespace(string $alias): string
 
         throw MongoDBException::unknownDocumentNamespace($alias);
     }
+
+    /**
+     * Clears all document managers.
+     */
+    public function reset(): void
+    {
+        foreach ($this->getManagerNames() as $managerName => $serviceId) {
+            $this->resetOrClearManager($managerName, $serviceId);
+        }
+    }
+
+    private function resetOrClearManager(string $managerName, string $serviceId): void
+    {
+        if (! $this->container->initialized($serviceId)) {
+            return;
+        }
+
+        $manager = $this->container->get($serviceId);
+
+        if ($manager instanceof LazyLoadingInterface || $manager instanceof LazyObjectInterface) {
+            $this->resetManager($managerName);
+
+            return;
+        }
+
+        assert($manager instanceof DocumentManager);
+
+        if (! $manager->isOpen()) {
+            return;
+        }
+
+        $manager->clear();
+    }
 }

tests/ManagerRegistryTest.php

@@ -0,0 +1,71 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Doctrine\Bundle\MongoDBBundle\Tests;
+
+use Doctrine\Bundle\MongoDBBundle\ManagerRegistry;
+use Doctrine\ODM\MongoDB\DocumentManager;
+use Doctrine\Persistence\ObjectManager;
+use MongoDBManagerRepositoryTestResetContainer;
+use Symfony\Component\DependencyInjection\ContainerBuilder;
+use Symfony\Component\DependencyInjection\Dumper\PhpDumper;
+
+class ManagerRegistryTest extends TestCase
+{
+    public function testReset(): void
+    {
+        $container = new ContainerBuilder();
+        $container->register('manager.default', DocumentManagerStub::class)
+            ->setPublic(true);
+        $container->register('manager.lazy', DocumentManagerStub::class)
+            ->setPublic(true)
+            ->setLazy(true)
+            ->addTag('proxy', ['interface' => ObjectManager::class]);
+        $container->compile();
+
+        $dumper = new PhpDumper($container);
+        eval('?' . '>' . $dumper->dump(['class' => 'MongoDBManagerRepositoryTestResetContainer']));
+
+        $container  = new MongoDBManagerRepositoryTestResetContainer();
+        $repository = new ManagerRegistry('MongoDB', [], [
+            'default' => 'manager.default',
+            'lazy' => 'manager.lazy',
+        ], '', '', '', $container);
+
+        DocumentManagerStub::$clearCount = 0;
+
+        $repository->reset();
+
+        // Service was not initialized, so reset should not be called
+        $this->assertSame(0, DocumentManagerStub::$clearCount);
+
+        // The lazy service is reinitialized instead of being cleared
+        $container->get('manager.lazy')->flush();
+        $repository->reset();
+        $this->assertSame(0, DocumentManagerStub::$clearCount);
+
+        // The default service is cleared when initialized
+        $container->get('manager.default')->flush();
+        $repository->reset();
+        $this->assertSame(1, DocumentManagerStub::$clearCount);
+    }
+}
+
+class DocumentManagerStub extends DocumentManager
+{
+    public static int $clearCount;
+
+    public function __construct()
+    {
+    }
+
+    public function clear($objectName = null): void
+    {
+        self::$clearCount++;
+    }
+
+    public function flush(array $options = []): void
+    {
+    }
+}