Unit test for Mongo Active Relation added.

tests/unit/extensions/mongo/ActiveRelationTest.php

@@ -0,0 +1,83 @@
+<?php
+
+namespace yiiunit\extensions\mongo;
+
+use yiiunit\data\ar\mongo\ActiveRecord;
+use yiiunit\data\ar\mongo\Customer;
+use yiiunit\data\ar\mongo\CustomerOrder;
+
+/**
+ * @group mongo
+ */
+class ActiveRelationTest extends MongoTestCase
+{
+	protected function setUp()
+	{
+		parent::setUp();
+		ActiveRecord::$db = $this->getConnection();
+		$this->setUpTestRows();
+	}
+
+	protected function tearDown()
+	{
+		$this->dropCollection(Customer::collectionName());
+		$this->dropCollection(CustomerOrder::collectionName());
+		parent::tearDown();
+	}
+
+	/**
+	 * Sets up test rows.
+	 */
+	protected function setUpTestRows()
+	{
+		$customerCollection = $this->getConnection()->getCollection('customer');
+
+		$customers = [];
+		for ($i = 1; $i <= 5; $i++) {
+			$customers[] = [
+				'name' => 'name' . $i,
+				'email' => 'email' . $i,
+				'address' => 'address' . $i,
+				'status' => $i,
+			];
+		}
+		$customerCollection->batchInsert($customers);
+
+		$customerOrderCollection = $this->getConnection()->getCollection('customer_order');
+		$customerOrders = [];
+		foreach ($customers as $customer) {
+			$customerOrders[] = [
+				'customer_id' => $customer['_id'],
+				'number' => $customer['status'],
+			];
+			$customerOrders[] = [
+				'customer_id' => $customer['_id'],
+				'number' => $customer['status'] + 1,
+			];
+		}
+		$customerOrderCollection->batchInsert($customerOrders);
+	}
+
+	// Tests :
+
+	public function testFindLazy()
+	{
+		/** @var CustomerOrder $order */
+		$order = CustomerOrder::find(['number' => 2]);
+		$this->assertFalse($order->isRelationPopulated('customer'));
+		$index = $order->customer;
+		$this->assertTrue($order->isRelationPopulated('customer'));
+		$this->assertTrue($index instanceof Customer);
+		$this->assertEquals(1, count($order->populatedRelations));
+	}
+
+	public function testFindEager()
+	{
+		$orders = CustomerOrder::find()->with('customer')->all();
+		$this->assertEquals(10, count($orders));
+		$this->assertTrue($orders[0]->isRelationPopulated('customer'));
+		$this->assertTrue($orders[1]->isRelationPopulated('customer'));
+		$this->assertTrue($orders[0]->index instanceof ArticleIndex);
+		$this->assertTrue($orders[1]->index instanceof ArticleIndex);
+	}
+}