Support Mockery::mock (#37)

Fixes #36
rectorphp · Jul 2, 2024 · 5c54cf1 · 5c54cf1
1 parent b2c4ffe
commit 5c54cf1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/PhpParser/NodeVisitor/MockedClassNameCollectingNodeVisitor.php b/src/PhpParser/NodeVisitor/MockedClassNameCollectingNodeVisitor.php
@@ -8,6 +8,7 @@
 use PhpParser\Node\Arg;
 use PhpParser\Node\Expr\ClassConstFetch;
 use PhpParser\Node\Expr\MethodCall;
+use PhpParser\Node\Expr\StaticCall;
 use PhpParser\Node\Identifier;
 use PhpParser\Node\Name;
 use PhpParser\NodeVisitorAbstract;
@@ -21,7 +22,7 @@ final class MockedClassNameCollectingNodeVisitor extends NodeVisitorAbstract
 
     public function enterNode(Node $node)
     {
-        if (! $node instanceof MethodCall) {
+        if (! $node instanceof MethodCall && ! $node instanceof StaticCall) {
             return null;
         }
 
@@ -31,7 +32,7 @@ public function enterNode(Node $node)
         }
 
         $methodName = $node->name->toString();
-        if (! in_array($methodName, ['getMock', 'createMock'], true)) {
+        if (! in_array($methodName, ['getMock', 'createMock', 'mock'], true)) {
             return null;
         }