add expects call

rectorphp · Sep 25, 2024 · a49d8fc · a49d8fc
1 parent b4acf4b
commit a49d8fc
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/...terface/WithConsecutiveRector/Fixture/combine_with_on_consecutive_without_expects.php.inc b/...terface/WithConsecutiveRector/Fixture/combine_with_on_consecutive_without_expects.php.inc
@@ -30,7 +30,7 @@ final class CombineWithOnConsecutiveWithoutExcepts extends TestCase
     public function test()
     {
         $matcher = $this->exactly(2);
-        $this->userServiceMock->expects($matcher)->method('prepare')->willReturnCallback(function ($parameters) use ($matcher) {
+        $this->userServiceMock->method('prepare')->expects($matcher)->willReturnCallback(function ($parameters) use ($matcher) {
             match ($matcher->numberOfInvocations()) {
                 1 => self::assertEquals([1, 2], $parameters),
                 2 => self::assertEquals([3, 4], $parameters),

diff --git a/rules/PHPUnit100/Rector/StmtsAwareInterface/WithConsecutiveRector.php b/rules/PHPUnit100/Rector/StmtsAwareInterface/WithConsecutiveRector.php
@@ -386,6 +386,16 @@ private function refactorToWillReturnCallback(
             )),
         ];
 
+        $hasExpects = $this->findMethodCall($expression, 'expects') instanceof MethodCall;
+        if ($hasExpects === false) {
+            /** @var MethodCall $mockMethodCall */
+            $mockMethodCall = $expression->expr;
+
+            $mockMethodCall->var = new MethodCall($mockMethodCall->var, 'expects', [
+                new Arg(new Variable('matcher'))
+            ]);
+        }
+
         $matcherAssign = new Assign(new Variable('matcher'), $expectsCall);
         return [new Expression($matcherAssign), $expression];
     }