diff --git a/src/NodeChanger/MethodNameChanger.php b/src/NodeChanger/MethodNameChanger.php index 68c180a16ad..434ef000bd7 100644 --- a/src/NodeChanger/MethodNameChanger.php +++ b/src/NodeChanger/MethodNameChanger.php @@ -7,13 +7,15 @@ use PhpParser\Node\Identifier; final class MethodNameChanger { - /** - * @param string[] $renameMethodMap - */ - public function renameNode(MethodCall $node, array $renameMethodMap): void - { - $oldNodeMethodName = $node->name->toString(); - $node->name = new Identifier($renameMethodMap[$oldNodeMethodName]); + public function renameNode(MethodCall $node, $oldToNewMethods): void + { + if (is_array($oldToNewMethods)) { + $oldNodeMethodName = $node->name->toString(); + + $node->name = new Identifier($oldToNewMethods[$oldNodeMethodName]); + } elseif (is_string($oldToNewMethods)) { + $node->name = new Identifier($oldToNewMethods); + } } }