MongoDB Active Relation lazy load fixed.

extensions/yii/mongodb/ActiveRelation.php

@@ -19,4 +19,35 @@ use yii\db\ActiveRelationTrait;
 class ActiveRelation extends ActiveQuery implements ActiveRelationInterface
 {
 	use ActiveRelationTrait;
+
+	/**
+	 * @inheritdoc
+	 */
+	protected function buildCursor($db = null)
+	{
+		if ($this->primaryModel !== null) {
+			// lazy loading
+			if ($this->via instanceof self) {
+				// via pivot collection
+				$viaModels = $this->via->findPivotRows([$this->primaryModel]);
+				$this->filterByModels($viaModels);
+			} elseif (is_array($this->via)) {
+				// via relation
+				/** @var ActiveRelation $viaQuery */
+				list($viaName, $viaQuery) = $this->via;
+				if ($viaQuery->multiple) {
+					$viaModels = $viaQuery->all();
+					$this->primaryModel->populateRelation($viaName, $viaModels);
+				} else {
+					$model = $viaQuery->one();
+					$this->primaryModel->populateRelation($viaName, $model);
+					$viaModels = $model === null ? [] : [$model];
+				}
+				$this->filterByModels($viaModels);
+			} else {
+				$this->filterByModels([$this->primaryModel]);
+			}
+		}
+		return parent::buildCursor($db);
+	}
 }

tests/unit/extensions/mongodb/ActiveRelationTest.php

@@ -52,7 +52,7 @@ class ActiveRelationTest extends MongoDbTestCase
 			];
 			$customerOrders[] = [
 				'customer_id' => $customer['_id'],
-				'number' => $customer['status'] + 1,
+				'number' => $customer['status'] + 100,
 			];
 		}
 		$customerOrderCollection->batchInsert($customerOrders);
@@ -65,9 +65,10 @@ class ActiveRelationTest extends MongoDbTestCase
 		/** @var CustomerOrder $order */
 		$order = CustomerOrder::find(['number' => 2]);
 		$this->assertFalse($order->isRelationPopulated('customer'));
-		$index = $order->customer;
+		$customer = $order->customer;
 		$this->assertTrue($order->isRelationPopulated('customer'));
-		$this->assertTrue($index instanceof Customer);
+		$this->assertTrue($customer instanceof Customer);
+		$this->assertEquals((string)$customer->_id, (string)$order->customer_id);
 		$this->assertEquals(1, count($order->populatedRelations));
 	}
 
@@ -78,6 +79,8 @@ class ActiveRelationTest extends MongoDbTestCase
 		$this->assertTrue($orders[0]->isRelationPopulated('customer'));
 		$this->assertTrue($orders[1]->isRelationPopulated('customer'));
 		$this->assertTrue($orders[0]->customer instanceof Customer);
+		$this->assertEquals((string)$orders[0]->customer->_id, (string)$orders[0]->customer_id);
 		$this->assertTrue($orders[1]->customer instanceof Customer);
+		$this->assertEquals((string)$orders[1]->customer->_id, (string)$orders[1]->customer_id);
 	}
 }