Aggregation uses CursorInterface instead of Cursor

doctrine · Sep 6, 2023 · 64ec92e · 64ec92e
1 parent 69e64e7
commit 64ec92e
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/lib/Doctrine/ODM/MongoDB/Aggregation/Aggregation.php b/lib/Doctrine/ODM/MongoDB/Aggregation/Aggregation.php
@@ -12,10 +12,9 @@
 use Doctrine\ODM\MongoDB\Mapping\ClassMetadata;
 use IteratorAggregate;
 use MongoDB\Collection;
-use MongoDB\Driver\Cursor;
+use MongoDB\Driver\CursorInterface;
 
 use function array_merge;
-use function assert;
 
 /** @psalm-import-type PipelineExpression from Builder */
 final class Aggregation implements IteratorAggregate
@@ -35,12 +34,11 @@ public function getIterator(): Iterator
         $options = array_merge($this->options, ['cursor' => true]);
 
         $cursor = $this->collection->aggregate($this->pipeline, $options);
-        assert($cursor instanceof Cursor);
 
         return $this->prepareIterator($cursor);
     }
 
-    private function prepareIterator(Cursor $cursor): Iterator
+    private function prepareIterator(CursorInterface&Iterator $cursor): Iterator
     {
         if ($this->classMetadata) {
             $cursor = new HydratingIterator($cursor, $this->dm->getUnitOfWork(), $this->classMetadata);