add unit test

cargomedia · Feb 2, 2015 · 0f21202 · 0f21202
1 parent 3aa3c65
commit 0f21202
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/tests/library/CM/MongoDb/ClientTest.php b/tests/library/CM/MongoDb/ClientTest.php
@@ -146,6 +146,26 @@ public function testFind() {
         $this->assertEquals([['foo' => 1], ['foo' => 1], ['foo' => 2], ['foo' => 3]], $actual);
     }
 
+    public function testFindBatchSize() {
+        $mongoDb = CM_Service_Manager::getInstance()->getMongoDb();
+        $collectionName = 'findBatchSize';
+        CM_Config::get()->CM_MongoDb_Client->batchSize = null;
+
+        $cursor = $mongoDb->find($collectionName);
+        $this->assertSame(0, $cursor->info()['batchSize']);
+        $cursor = $mongoDb->find($collectionName, null, null, ['$match' => ['foo' => 'bar']]);
+        $this->assertSame(0, $cursor->info()['batchSize']);
+        $this->assertSame(101, $cursor->info()['query']['cursor']['batchSize']);
+
+        CM_Config::get()->CM_MongoDb_Client->batchSize = 10;
+
+        $cursor = $mongoDb->find($collectionName);
+        $this->assertSame(10, $cursor->info()['batchSize']);
+        $cursor = $mongoDb->find($collectionName, null, null, ['$match' => ['foo' => 'bar']]);
+        $this->assertSame(10, $cursor->info()['batchSize']);
+        $this->assertSame(10, $cursor->info()['query']['cursor']['batchSize']);
+    }
+
     public function testFindAndModify() {
         $mongoDb = CM_Service_Manager::getInstance()->getMongoDb();
         $collectionName = 'findAndModify';