Revert return to null if manager is not found; add bool to throw exce…

…ption instead
laravel-doctrine · Nov 1, 2024 · faf63df · faf63df
1 parent 936cadc
commit faf63df
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 3 deletions.
diff --git a/src/IlluminateRegistry.php b/src/IlluminateRegistry.php
@@ -302,7 +302,7 @@ public function getRepository(string $persistentObject, string|null $persistentM
      *
      * @param class-string $className A persistent object class name.
      */
-    public function getManagerForClass(string $className): ObjectManager|null
+    public function getManagerForClass(string $className, bool $throwExceptionIfNotFound = false): ObjectManager|null
     {
         // Check for namespace alias
         if (strpos($className, ':') !== false) {
@@ -335,7 +335,11 @@ public function getManagerForClass(string $className): ObjectManager|null
             }
         }
 
-        throw new RuntimeException('No manager found for class ' . $className);
+        if ($throwExceptionIfNotFound) {
+            throw new RuntimeException('No manager found for class ' . $className);
+        }
+
+        return null;
     }
 
     /**

diff --git a/tests/Feature/IlluminateRegistryTest.php b/tests/Feature/IlluminateRegistryTest.php
@@ -485,6 +485,37 @@ public function testGetManagerForClassWithNamespace(): void
         $this->assertEquals($entityManager, $this->registry->getManagerForClass('Alias:Scientist'));
     }
 
+    public function testGetManagerForClassReturnsNullWhenNotFound(): void
+    {
+        $this->container->shouldReceive('singleton');
+        $this->registry->addManager('default');
+
+        $entityManager = m::mock(EntityManagerInterface::class);
+        $this->container->shouldReceive('make')
+            ->with('doctrine.managers.default')
+            ->andReturn($entityManager);
+
+        $metadataFactory = m::mock(ClassMetadataFactory::class);
+        $metadataFactory->shouldReceive('isTransient')
+            ->with('LaravelDoctrineTest\ORM\Assets\Entity\Scientist')
+            ->once()
+            ->andReturnFalse();
+
+        $metadata = m::mock(ClassMetadata::class);
+        $metadata->shouldReceive('getName')
+            ->once()
+            ->andReturn('LaravelDoctrineTest\ORM\Assets\Entity\Theory');
+
+        $metadataFactory->shouldReceive('getAllMetadata')
+            ->once()
+            ->andReturn([$metadata]);
+
+        $entityManager->shouldReceive('getMetadataFactory')
+            ->andReturn($metadataFactory);
+
+        $this->assertNull($this->registry->getManagerForClass('LaravelDoctrineTest\ORM\Assets\Entity\Scientist'));
+    }
+
     public function testGetManagerForClassThrowsExceptionWhenNotFound(): void
     {
         $this->expectException(RuntimeException::class);
@@ -515,7 +546,7 @@ public function testGetManagerForClassThrowsExceptionWhenNotFound(): void
         $entityManager->shouldReceive('getMetadataFactory')
             ->andReturn($metadataFactory);
 
-        $this->assertEquals($entityManager, $this->registry->getManagerForClass('LaravelDoctrineTest\ORM\Assets\Entity\Scientist'));
+        $this->assertEquals($entityManager, $this->registry->getManagerForClass('LaravelDoctrineTest\ORM\Assets\Entity\Scientist', true));
     }
 
     public function testGetManagerForClassInvalidClass(): void