[PHPUnit100] Add willReturnOnConsecutiveCalls() variables to closure use() in WithConsecutiveRector

rules-tests/PHPUnit100/Rector/StmtsAwareInterface/WithConsecutiveRector/Fixture/combine_with_will_return_on_consecutive_calls_variable.php.inc

@@ -0,0 +1,49 @@
+<?php
+
+namespace Rector\PHPUnit\Tests\PHPUnit100\Rector\StmtsAwareInterface\WithConsecutiveRector\Fixture;
+
+use PHPUnit\Framework\TestCase;
+
+final class CombineWithWillReturnOnConsecutiveCallsVariable extends TestCase
+{
+    public function test()
+    {
+        $title = new \stdClass();
+
+        $this->manager->expects($this->exactly(2))
+            ->method('getFieldDefinitions')
+            ->withConsecutive(['first'], ['second'])
+            ->willReturnOnConsecutiveCalls(['title' => $title], []);
+    }
+}
+
+?>
+-----
+<?php
+
+namespace Rector\PHPUnit\Tests\PHPUnit100\Rector\StmtsAwareInterface\WithConsecutiveRector\Fixture;
+
+use PHPUnit\Framework\TestCase;
+
+final class CombineWithWillReturnOnConsecutiveCallsVariable extends TestCase
+{
+    public function test()
+    {
+        $title = new \stdClass();
+        $matcher = $this->exactly(2);
+
+        $this->manager->expects($matcher)
+            ->method('getFieldDefinitions')->willReturnCallback(function (...$parameters) use ($matcher, $title) {
+            if ($matcher->numberOfInvocations() === 1) {
+                $this->assertSame('first', $parameters[0]);
+                return ['title' => $title];
+            }
+            if ($matcher->numberOfInvocations() === 2) {
+                $this->assertSame('second', $parameters[0]);
+                return [];
+            }
+        });
+    }
+}
+
+?>

rules/PHPUnit100/NodeFactory/WillReturnCallbackFactory.php

@@ -36,9 +36,14 @@ public function createClosure(
         MethodCall $withConsecutiveMethodCall,
         ?Stmt $returnStmt,
         Variable|Expr|null $referenceVariable,
+        ?MethodCall $willReturnOnConsecutiveMethodCall = null,
     ): Closure {
         $matcherVariable = new Variable(ConsecutiveVariable::MATCHER);
-        $usedVariables = $this->usedVariablesResolver->resolveUsedVariables($withConsecutiveMethodCall, $returnStmt);
+        $usedVariables = $this->usedVariablesResolver->resolveUsedVariables(
+            $withConsecutiveMethodCall,
+            $returnStmt,
+            $willReturnOnConsecutiveMethodCall
+        );
 
         $closureStmts = $this->createParametersMatch($withConsecutiveMethodCall);
         if ($returnStmt instanceof Stmt) {

rules/PHPUnit100/Rector/StmtsAwareInterface/WithConsecutiveRector.php

@@ -242,6 +242,7 @@ private function refactorToWillReturnCallback(
             $withConsecutiveMethodCall,
             $returnStmt,
             $referenceVariable,
+            $willReturnOnConsecutiveMethodCall,
         );
 
         $withConsecutiveMethodCall->name = new Identifier(ConsecutiveMethodName::WILL_RETURN_CALLBACK);

src/NodeFactory/UsedVariablesResolver.php

@@ -23,10 +23,17 @@ public function __construct(
     /**
      * @return Variable[]
      */
-    public function resolveUsedVariables(MethodCall $withConsecutiveMethodCall, ?Stmt $returnStmt): array
-    {
+    public function resolveUsedVariables(
+        MethodCall $withConsecutiveMethodCall,
+        ?Stmt $returnStmt,
+        ?MethodCall $willReturnOnConsecutiveMethodCall = null
+    ): array {
         $consecutiveArgs = $withConsecutiveMethodCall->getArgs();
 
+        if ($willReturnOnConsecutiveMethodCall instanceof MethodCall) {
+            $consecutiveArgs = array_merge($consecutiveArgs, $willReturnOnConsecutiveMethodCall->getArgs());
+        }
+
         $stmtVariables = $returnStmt instanceof Stmt ? $this->resolveUniqueVariables([$returnStmt]) : [];
 
         return $this->resolveUniqueVariables(array_merge($consecutiveArgs, $stmtVariables));