Add alias when using lookup on an inverse relationship

doctrine · Mar 31, 2020 · 1c6fade · 1c6fade
1 parent 14da812
commit 1c6fade
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 2 deletions.
diff --git a/lib/Doctrine/ODM/MongoDB/Aggregation/Stage/Lookup.php b/lib/Doctrine/ODM/MongoDB/Aggregation/Stage/Lookup.php
@@ -140,7 +140,8 @@ private function fromReference($fieldName)
 
             $this
                 ->localField('_id')
-                ->foreignField($referencedFieldName);
+                ->foreignField($referencedFieldName)
+                ->alias($fieldName);
         }
 
         return $this;

diff --git a/tests/Doctrine/ODM/MongoDB/Tests/Aggregation/Stage/LookupTest.php b/tests/Doctrine/ODM/MongoDB/Tests/Aggregation/Stage/LookupTest.php
@@ -420,4 +420,51 @@ private function insertTestData()
         $this->dm->persist($referenceUser);
         $this->dm->flush();
     }
+
+    public function testLookupStageAndDefaultAlias()
+    {
+        $builder = $this->dm->createAggregationBuilder(\Documents\User::class);
+        $builder
+            ->lookup('simpleReferenceOneInverse');
+
+        $expectedPipeline = [
+            [
+                '$lookup' => [
+                    'from' => 'SimpleReferenceUser',
+                    'localField' => '_id',
+                    'foreignField' => 'userId',
+                    'as' => 'simpleReferenceOneInverse',
+                ],
+            ],
+        ];
+
+        $this->assertEquals($expectedPipeline, $builder->getPipeline());
+
+        $result = $builder->execute()->toArray();
+        $this->assertCount(1, $result[0]['simpleReferenceOneInverse']);
+    }
+
+    public function testLookupStageAndDefaultAliasOverride()
+    {
+        $builder = $this->dm->createAggregationBuilder(\Documents\User::class);
+        $builder
+            ->lookup('simpleReferenceOneInverse')
+                ->alias('override');
+
+        $expectedPipeline = [
+            [
+                '$lookup' => [
+                    'from' => 'SimpleReferenceUser',
+                    'localField' => '_id',
+                    'foreignField' => 'userId',
+                    'as' => 'override',
+                ],
+            ],
+        ];
+
+        $this->assertEquals($expectedPipeline, $builder->getPipeline());
+
+        $result = $builder->execute()->toArray();
+        $this->assertCount(1, $result[0]['override']);
+    }
 }
diff --git a/tests/Documents/SimpleReferenceUser.php b/tests/Documents/SimpleReferenceUser.php
@@ -12,7 +12,7 @@ class SimpleReferenceUser
     /** @ODM\Id */
     public $id;
 
-    /** @ODM\ReferenceOne(targetDocument="Documents\User", simple=true, name="userId") @ODM\Index */
+    /** @ODM\ReferenceOne(targetDocument="Documents\User", simple=true, name="userId", inversedBy="simpleReferenceManyInverse") @ODM\Index */
     public $user;
 
     /** @ODM\ReferenceMany(targetDocument="Documents\User", simple=true) */