[DoctrineBridge] Map entities by single-key when disabling auto-mapping

nicolas-grekas · nicolas-grekas · commit ec6d2254a759 · 2024-04-02T11:47:57.000+02:00
diff --git a/src/Symfony/Bridge/Doctrine/ArgumentResolver/EntityValueResolver.php b/src/Symfony/Bridge/Doctrine/ArgumentResolver/EntityValueResolver.php
@@ -152,11 +152,13 @@ private function getIdentifier(Request $request, MapEntity $options, string $nam
 
     private function getCriteria(Request $request, MapEntity $options, ObjectManager $manager): array
     {
-        if (null === $mapping = $options->mapping) {
+        $singleKey = [] === $options->mapping;
+
+        if (!$mapping = $options->mapping) {
             $mapping = $request->attributes->keys();
         }
 
-        if ($mapping && \is_array($mapping) && array_is_list($mapping)) {
+        if ($mapping && array_is_list($mapping)) {
             $mapping = array_combine($mapping, $mapping);
         }
 
@@ -176,13 +178,28 @@ private function getCriteria(Request $request, MapEntity $options, ObjectManager
 
         $criteria = [];
         $metadata = $manager->getClassMetadata($options->class);
+        $usedAttributes = null;
+
+        if ($singleKey) {
+            $usedAttributes ??= $request->attributes->get('_entity_used_attributes') ?? [];
+
+            foreach ($usedAttributes as $attribute) {
+                unset($mapping[$attribute]);
+            }
+        }
 
         foreach ($mapping as $attribute => $field) {
             if (!$metadata->hasField($field) && (!$metadata->hasAssociation($field) || !$metadata->isSingleValuedAssociation($field))) {
                 continue;
             }
 
             $criteria[$field] = $request->attributes->get($attribute);
+
+            if ($singleKey) {
+                $usedAttributes[] = $attribute;
+                $request->attributes->set('_entity_used_attributes', $usedAttributes);
+                break;
+            }
         }
 
         if ($options->stripNull) {
diff --git a/src/Symfony/Bridge/Doctrine/CHANGELOG.md b/src/Symfony/Bridge/Doctrine/CHANGELOG.md
@@ -1,6 +1,11 @@
 CHANGELOG
 =========
 
+7.1
+---
+
+ * Map entities by single-key when disabling auto-mapping
+
 7.0
 ---
 
diff --git a/src/Symfony/Bridge/Doctrine/Tests/ArgumentResolver/EntityValueResolverTest.php b/src/Symfony/Bridge/Doctrine/Tests/ArgumentResolver/EntityValueResolverTest.php
@@ -257,6 +257,60 @@ public function testResolveWithMappingAndExclude()
         $this->assertSame([$object], $resolver->resolve($request, $argument));
     }
 
+    public function testResolveWithMultipleEntitiesAndNoAutoMapping()
+    {
+        $manager = $this->getMockBuilder(ObjectManager::class)->getMock();
+        $registry = $this->createRegistry($manager);
+        $resolver = new EntityValueResolver($registry);
+
+        $request = new Request();
+        $request->attributes->set('foo', 1);
+        $request->attributes->set('bar', 2);
+
+        $argument1 = $this->createArgument('Foo', new MapEntity('Foo', mapping: []));
+        $argument2 = $this->createArgument('Bar', new MapEntity('Bar', mapping: []));
+
+        $metadata1 = $this->getMockBuilder(ClassMetadata::class)->getMock();
+        $metadata1->expects($this->once())
+            ->method('hasField')
+            ->with('foo')
+            ->willReturn(true);
+
+        $metadata2 = $this->getMockBuilder(ClassMetadata::class)->getMock();
+        $metadata2->expects($this->once())
+            ->method('hasField')
+            ->with('bar')
+            ->willReturn(true);
+
+        $manager->expects($this->any())
+            ->method('getClassMetadata')
+            ->willReturnCallback(static fn ($v) => match ($v) {
+                'Foo' => $metadata1,
+                'Bar' => $metadata2,
+            });
+
+        $object1 = new \stdClass();
+        $object2 = new \stdClass();
+
+        $repository = $this->getMockBuilder(ObjectRepository::class)->getMock();
+        $repository->expects($this->any())
+            ->method('findOneBy')
+            ->willReturnCallback(static fn ($v) => match ($v) {
+                ['foo' => 1] => $object1,
+                ['bar' => 2] => $object2,
+            });
+
+        $manager->expects($this->any())
+            ->method('getRepository')
+            ->willReturn($repository);
+
+        $this->assertSame([$object1], $resolver->resolve($request, $argument1));
+        $this->assertSame(['foo'], $request->attributes->get('_entity_used_attributes'));
+
+        $this->assertSame([$object2], $resolver->resolve($request, $argument2));
+        $this->assertSame(['foo', 'bar'], $request->attributes->get('_entity_used_attributes'));
+    }
+
     public function testExceptionWithExpressionIfNoLanguageAvailable()
     {
         $manager = $this->getMockBuilder(ObjectManager::class)->getMock();
