Add unit test for Query\Builder::select()

doctrine · Mar 29, 2013 · 20bdf16 · 20bdf16
1 parent ac94f9a
commit 20bdf16
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/Doctrine/MongoDB/Tests/Query/BuilderTest.php b/tests/Doctrine/MongoDB/Tests/Query/BuilderTest.php
@@ -514,6 +514,18 @@ public function testWithinPolygonRequiresAtLeastThreePoints()
             ->field('loc')->withinPolygon(array(0, 0), array(1, 1));
     }
 
+    public function testSelect()
+    {
+        $qb = $this->getTestQueryBuilder()
+            ->select('foo', 'bar');
+
+        $expected = array(
+            'foo' => 1,
+            'bar' => 1
+        );
+        $this->assertEquals($expected, $qb->debug('select'));
+    }
+
     private function getTestQueryBuilder()
     {
         return $this->conn->selectCollection('db', 'users')->createQueryBuilder();