Support createPartialMock

rectorphp · Jan 3, 2025 · 9e91436 · 9e91436
1 parent 0f39f20
commit 9e91436
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/PhpParser/NodeVisitor/MockedClassNameCollectingNodeVisitor.php b/src/PhpParser/NodeVisitor/MockedClassNameCollectingNodeVisitor.php
@@ -32,7 +32,14 @@ public function enterNode(Node $node): ?Node
         }
 
         $methodName = $node->name->toString();
-        if (! in_array($methodName, ['getMock', 'createMock', 'mock', 'getMockBuilder'], true)) {
+        $mockMethodNames = [
+            'createMock',
+            'createPartialMock',
+            'getMock',
+            'getMockBuilder',
+            'mock',
+        ];
+        if (! in_array($methodName, $mockMethodNames, true)) {
             return null;
         }