test: mongodb mock return types

Author: soyuka

tests/Doctrine/Odm/Extension/OrderExtensionTest.php

@@ -126,7 +126,7 @@ public function testApplyToCollectionWithOrderOverriddenWithAssociation(): void
         $lookupProphecy = $this->prophesize(Lookup::class);
         $lookupProphecy->localField('author')->shouldBeCalled()->willReturn($lookupProphecy);
         $lookupProphecy->foreignField('_id')->shouldBeCalled()->willReturn($lookupProphecy);
-        $lookupProphecy->alias('author_lkup')->shouldBeCalled();
+        $lookupProphecy->alias('author_lkup')->shouldBeCalled()->willReturn($lookupProphecy);
         $aggregationBuilderProphecy->lookup(Dummy::class)->shouldBeCalled()->willReturn($lookupProphecy->reveal());
         $aggregationBuilderProphecy->unwind('$author_lkup')->shouldBeCalled();
         $aggregationBuilderProphecy->getStage(0)->willThrow(new \OutOfRangeException('message'));

tests/Doctrine/Odm/Extension/PaginationExtensionTest.php

@@ -414,8 +414,8 @@ private function mockAggregationBuilder(int $expectedOffset, int $expectedLimit)
             $skipProphecy->limit($expectedLimit)->shouldBeCalled();
         } else {
             $matchProphecy = $this->prophesize(AggregationMatch::class);
-            $matchProphecy->field(Paginator::LIMIT_ZERO_MARKER_FIELD)->shouldBeCalled()->willReturn($matchProphecy);
-            $matchProphecy->equals(Paginator::LIMIT_ZERO_MARKER)->shouldBeCalled();
+            $matchProphecy->field(Paginator::LIMIT_ZERO_MARKER_FIELD)->shouldBeCalled()->willReturn($matchProphecy->reveal());
+            $matchProphecy->equals(Paginator::LIMIT_ZERO_MARKER)->shouldBeCalled()->willReturn($matchProphecy->reveal());
             $skipProphecy->match()->shouldBeCalled()->willReturn($matchProphecy->reveal());
         }
 

tests/Doctrine/Odm/State/ItemProviderTest.php

@@ -46,7 +46,7 @@ public function testGetItemSingleIdentifier(): void
 
         $matchProphecy = $this->prophesize(AggregationMatch::class);
         $matchProphecy->field('id')->willReturn($matchProphecy)->shouldBeCalled();
-        $matchProphecy->equals(1)->shouldBeCalled();
+        $matchProphecy->equals(1)->shouldBeCalled()->willReturn($matchProphecy);
 
         $iterator = $this->prophesize(Iterator::class);
         $result = new \stdClass();
@@ -78,8 +78,8 @@ public function testGetItemWithExecuteOptions(): void
         $context = ['foo' => 'bar', 'fetch_data' => true];
 
         $matchProphecy = $this->prophesize(AggregationMatch::class);
-        $matchProphecy->field('id')->willReturn($matchProphecy)->shouldBeCalled();
-        $matchProphecy->equals(1)->shouldBeCalled();
+        $matchProphecy->field('id')->willReturn($matchProphecy->reveal())->shouldBeCalled();
+        $matchProphecy->equals(1)->shouldBeCalled()->willReturn($matchProphecy->reveal());
 
         $iterator = $this->prophesize(Iterator::class);
         $result = new \stdClass();
@@ -112,8 +112,8 @@ public function testGetItemDoubleIdentifier(): void
         $matchProphecy = $this->prophesize(AggregationMatch::class);
         $matchProphecy->field('ida')->willReturn($matchProphecy)->shouldBeCalled();
         $matchProphecy->field('idb')->willReturn($matchProphecy)->shouldBeCalled();
-        $matchProphecy->equals(1)->shouldBeCalled();
-        $matchProphecy->equals(2)->shouldBeCalled();
+        $matchProphecy->equals(1)->shouldBeCalled()->willReturn($matchProphecy);
+        $matchProphecy->equals(2)->shouldBeCalled()->willReturn($matchProphecy);
 
         $iterator = $this->prophesize(Iterator::class);
         $result = new \stdClass();
@@ -147,7 +147,7 @@ public function testAggregationResultExtension(): void
 
         $matchProphecy = $this->prophesize(AggregationMatch::class);
         $matchProphecy->field('id')->willReturn($matchProphecy)->shouldBeCalled();
-        $matchProphecy->equals(1)->shouldBeCalled();
+        $matchProphecy->equals(1)->shouldBeCalled()->willReturn($matchProphecy);
 
         $aggregationBuilderProphecy = $this->prophesize(Builder::class);
         $aggregationBuilderProphecy->match()->willReturn($matchProphecy->reveal())->shouldBeCalled();