covert topmost call

Author: TomasVotruba

rules-tests/CodeQuality/Rector/ClassMethod/RemoveStandaloneCreateMockRector/Fixture/handle_multiples.php.inc

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\PHPUnit\Tests\CodeQuality\Rector\ClassMethod\RemoveStandaloneCreateMockRector\Fixture;
+
+final class HandleMultiples extends \PHPUnit\Framework\TestCase
+{
+    public function testSomething(): void
+    {
+        $this->createMock('someClass');
+        $this->createMock('someClass');
+        $this->createMock('someClass');
+        $this->createMock('someClass');
+    }
+}
+
+?>
+-----
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\PHPUnit\Tests\CodeQuality\Rector\ClassMethod\RemoveStandaloneCreateMockRector\Fixture;
+
+final class HandleMultiples extends \PHPUnit\Framework\TestCase
+{
+    public function testSomething(): void
+    {
+    }
+}
+
+?>

rules-tests/CodeQuality/Rector/ClassMethod/RemoveStandaloneCreateMockRector/Fixture/handle_nested_call.php.inc

@@ -0,0 +1,32 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\PHPUnit\Tests\CodeQuality\Rector\ClassMethod\RemoveStandaloneCreateMockRector\Fixture;
+
+final class HandleNestedCall extends \PHPUnit\Framework\TestCase
+{
+    public function testSomething(): void
+    {
+        $this->createMock('someClass')
+            ->method('some')
+            ->willReturn(100);
+    }
+}
+
+?>
+-----
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\PHPUnit\Tests\CodeQuality\Rector\ClassMethod\RemoveStandaloneCreateMockRector\Fixture;
+
+final class HandleNestedCall extends \PHPUnit\Framework\TestCase
+{
+    public function testSomething(): void
+    {
+    }
+}
+
+?>

rules/CodeQuality/Rector/ClassMethod/RemoveStandaloneCreateMockRector.php

@@ -84,8 +84,8 @@ public function refactor(Node $node): ?Node
                 continue;
             }
 
-            $methodCall = $stmt->expr;
-            if (! $this->isName($methodCall->name, 'createMock')) {
+            $topmostCall = $this->resolveTopmostCall($stmt->expr);
+            if (! $this->isName($topmostCall->name, 'createMock')) {
                 continue;
             }
 
@@ -99,4 +99,14 @@ public function refactor(Node $node): ?Node
 
         return null;
     }
+
+    private function resolveTopmostCall(MethodCall $methodCall): MethodCall
+    {
+        $currentMethodCall = $methodCall;
+        while ($currentMethodCall->var instanceof MethodCall) {
+            $currentMethodCall = $currentMethodCall->var;
+        }
+
+        return $currentMethodCall;
+    }
 }