From 479fbf77ef874798c42d811cdaa85159e81270ca Mon Sep 17 00:00:00 2001
From: Paul Klimov <klimov.paul@gmail.com>
Date: Fri, 29 Nov 2013 17:12:13 +0200
Subject: [PATCH] Unit test for Mongo ActiveDataProvider added.

---
 .../extensions/mongo/ActiveDataProviderTest.php    | 91 ++++++++++++++++++++++
 1 file changed, 91 insertions(+)
 create mode 100644 tests/unit/extensions/mongo/ActiveDataProviderTest.php

diff --git a/tests/unit/extensions/mongo/ActiveDataProviderTest.php b/tests/unit/extensions/mongo/ActiveDataProviderTest.php
new file mode 100644
index 0000000..3660516
--- /dev/null
+++ b/tests/unit/extensions/mongo/ActiveDataProviderTest.php
@@ -0,0 +1,91 @@
+<?php
+
+namespace yiiunit\extensions\mongo;
+
+use yii\data\ActiveDataProvider;
+use yii\mongo\Query;
+use yiiunit\data\ar\mongo\ActiveRecord;
+use yiiunit\data\ar\mongo\Customer;
+
+/**
+ * @group mongo
+ */
+class ActiveDataProviderTest extends MongoTestCase
+{
+	protected function setUp()
+	{
+		parent::setUp();
+		ActiveRecord::$db = $this->getConnection();
+		$this->setUpTestRows();
+	}
+
+	protected function tearDown()
+	{
+		$this->dropCollection(Customer::collectionName());
+		parent::tearDown();
+	}
+
+	/**
+	 * Sets up test rows.
+	 */
+	protected function setUpTestRows()
+	{
+		$collection = $this->getConnection()->getCollection('customer');
+		$rows = [];
+		for ($i = 1; $i <= 10; $i++) {
+			$rows[] = [
+				'name' => 'name' . $i,
+				'email' => 'email' . $i,
+				'address' => 'address' . $i,
+				'status' => $i,
+			];
+		}
+		$collection->batchInsert($rows);
+	}
+
+	// Tests :
+
+	public function testQuery()
+	{
+		$query = new Query;
+		$query->from('customer');
+
+		$provider = new ActiveDataProvider([
+			'query' => $query,
+			'db' => $this->getConnection(),
+		]);
+		$models = $provider->getModels();
+		$this->assertEquals(10, count($models));
+
+		$provider = new ActiveDataProvider([
+			'query' => $query,
+			'db' => $this->getConnection(),
+			'pagination' => [
+				'pageSize' => 5,
+			]
+		]);
+		$models = $provider->getModels();
+		$this->assertEquals(5, count($models));
+	}
+
+	public function testActiveQuery()
+	{
+		$provider = new ActiveDataProvider([
+			'query' => Customer::find()->orderBy('id ASC'),
+		]);
+		$models = $provider->getModels();
+		$this->assertEquals(10, count($models));
+		$this->assertTrue($models[0] instanceof Customer);
+		$keys = $provider->getKeys();
+		$this->assertTrue($keys[0] instanceof \MongoId);
+
+		$provider = new ActiveDataProvider([
+			'query' => Customer::find(),
+			'pagination' => [
+				'pageSize' => 5,
+			]
+		]);
+		$models = $provider->getModels();
+		$this->assertEquals(5, count($models));
+	}
+}
\ No newline at end of file