Add tests

Author: Steveb-p

tests/Doctrine/ODM/MongoDB/Tests/Aggregation/BuilderTest.php

@@ -7,6 +7,7 @@
 use DateTimeImmutable;
 use Doctrine\ODM\MongoDB\Aggregation\Stage;
 use Doctrine\ODM\MongoDB\Iterator\Iterator;
+use Doctrine\ODM\MongoDB\Iterator\UnrewindableIterator;
 use Doctrine\ODM\MongoDB\Tests\BaseTest;
 use Documents\Article;
 use Documents\BlogPost;
@@ -352,12 +353,29 @@ public function testBuilderWithOutStageReturnsNoData()
     public function testBuilderWithIndexStatsStageDoesNotApplyFilters()
     {
         $builder = $this->dm
-            ->createAggregationBuilder(Project::class)
+            ->createAggregationBuilder(BlogPost::class)
             ->indexStats();
 
         $this->assertSame('$indexStats', array_keys($builder->getPipeline()[0])[0]);
     }
 
+    /**
+     * @expectedException \LogicException
+     */
+    public function testNonRewindableBuilder()
+    {
+        $builder = $this->dm
+            ->createAggregationBuilder(Project::class)
+            ->match()
+            ->rewindable(false);
+
+        $iterator = $builder->execute();
+        $this->assertInstanceOf(UnrewindableIterator::class, $iterator);
+        iterator_to_array($iterator);
+        iterator_to_array($iterator);
+
+    }
+
     private function insertTestData()
     {
         $baseballTag = new Tag('baseball');

tests/Doctrine/ODM/MongoDB/Tests/Query/BuilderTest.php

@@ -5,6 +5,7 @@
 namespace Doctrine\ODM\MongoDB\Tests\Query;
 
 use DateTime;
+use Doctrine\ODM\MongoDB\Iterator\UnrewindableIterator;
 use Doctrine\ODM\MongoDB\Mapping\Annotations as ODM;
 use Doctrine\ODM\MongoDB\Query\Builder;
 use Doctrine\ODM\MongoDB\Query\Expr;
@@ -806,6 +807,20 @@ public function testBitXorUpdateQuery()
         $this->assertEquals($expected, $qb->getNewObj());
     }
 
+    /**
+     * @expectedException \LogicException
+     */
+    public function testNonRewindable()
+    {
+        $query = $this->getTestQueryBuilder()
+            ->setRewindable(false)
+            ->getQuery();
+
+        $this->assertInstanceOf(UnrewindableIterator::class, $query->execute());
+        iterator_to_array($query->execute());
+        iterator_to_array($query->execute());
+    }
+
     private function getTestQueryBuilder()
     {
         return new Builder($this->dm, User::class);

tests/Doctrine/ODM/MongoDB/Tests/QueryTest.php

@@ -5,6 +5,7 @@
 namespace Doctrine\ODM\MongoDB\Tests;
 
 use Doctrine\ODM\MongoDB\Iterator\Iterator;
+use Doctrine\ODM\MongoDB\Iterator\UnrewindableIterator;
 use Doctrine\ODM\MongoDB\Mapping\Annotations as ODM;
 use Doctrine\ODM\MongoDB\Mapping\ClassMetadata;
 use Doctrine\ODM\MongoDB\Query\Query;
@@ -516,6 +517,33 @@ public function testFindOptionInheritance()
         $this->assertInstanceOf(Traversable::class, $query->execute());
     }
 
+    /**
+     * @expectedException \LogicException
+     */
+    public function testNonRewindable(): void
+    {
+        $cursor = $this->createMock(Traversable::class);
+
+        $collection = $this->getMockCollection();
+        $collection->expects($this->once())
+            ->method('find')
+            ->with([], [])
+            ->will($this->returnValue($cursor));
+
+        $queryArray = [
+            'type' => Query::TYPE_FIND,
+            'query' => [],
+        ];
+        $query = new Query($this->dm, new ClassMetadata(User::class), $collection, $queryArray);
+        $query->setRewindable(false);
+
+        $iterator = $query->execute();
+
+        $this->assertInstanceOf(UnrewindableIterator::class, $iterator);
+        iterator_to_array($iterator);
+        iterator_to_array($iterator);
+    }
+
     private function getMockCollection()
     {
         return $this->getMockBuilder(Collection::class)