remove broken node visitor from phpstan

Author: TomasVotruba

tests/PhpParser/Printer/PHPStanPrinterTest.php

@@ -7,6 +7,8 @@
 use PhpParser\PrettyPrinter\Standard;
 use PHPStan\DependencyInjection\MemoizingContainer;
 use PHPStan\DependencyInjection\Nette\NetteContainer;
+use PHPStan\Parser\AnonymousClassVisitor;
+use PHPStan\Parser\ArrayMapArgVisitor;
 use PHPStan\Parser\Parser;
 use PHPStan\Parser\RichParser;
 use Rector\Testing\PHPUnit\AbstractLazyTestCase;
@@ -15,6 +17,8 @@
 /**
  * Test case for: https://github.com/rectorphp/rector/issues/9492
  * Most likely caused by https://github.com/phpstan/phpstan-src/pull/3763
+ *
+ * @see https://github.com/phpstan/phpstan-src/blob/2.1.x/src/Parser/ArrayMapArgVisitor.php
  */
 final class PHPStanPrinterTest extends AbstractLazyTestCase
 {
@@ -23,6 +27,30 @@ public function testAddingCommentOnSomeNodesFail(): void
         /** @var RichParser $phpstanParser */
         $phpstanParser = $this->make(Parser::class);
 
+        $this->removeNodeVisitorFromPHPStanParser($phpstanParser, [ArrayMapArgVisitor::class]);
+
+        $stmts = $phpstanParser->parseFile(__DIR__ . '/Fixture/some_array_map.php');
+
+        // get private property "parser"
+        $parserReflectionProperty = new \ReflectionProperty(RichParser::class, 'parser');
+
+        /** @var \PhpParser\Parser $innerParser */
+        $innerParser = $parserReflectionProperty->getValue($phpstanParser);
+        $tokens = $innerParser->getTokens();
+
+        $standardPrinter = new Standard([
+            'newline' => "\n",
+        ]);
+        $printerContents = $standardPrinter->printFormatPreserving($stmts, $stmts, $tokens);
+
+        $this->assertStringEqualsFile(__DIR__ . '/Fixture/some_array_map.php', $printerContents);
+    }
+
+    /**
+     * @param class-string[] $nodeVisitorsToRemove
+     */
+    private function removeNodeVisitorFromPHPStanParser(RichParser $phpstanParser, array $nodeVisitorsToRemove): void
+    {
         // the only way now seems to access container early and remove unwanted services
         // here https://github.com/phpstan/phpstan-src/blob/522421b007cbfc674bebb93e823c774167ac78cd/src/Parser/RichParser.php#L90-L92
         $privatesAccessor = new PrivatesAccessor();
@@ -36,31 +64,15 @@ public function testAddingCommentOnSomeNodesFail(): void
         /** @var NetteContainer $originalContainer */
         $deeperContainer = $privatesAccessor->getPrivateProperty($originalContainer, 'container');
 
-        // @todo
-
         // get tags property
         $tags = $privatesAccessor->getPrivateProperty($deeperContainer, 'tags');
 
-        dump($tags);
-        die;
-
-        dump($container->getServicesByTag(RichParser::VISITOR_SERVICE_TAG));
-        die;
-
-        $stmts = $phpstanParser->parseFile(__DIR__ . '/Fixture/some_array_map.php');
-
-        // get private property "parser"
-        $parserReflectionProperty = new \ReflectionProperty(RichParser::class, 'parser');
-
-        /** @var \PhpParser\Parser $innerParser */
-        $innerParser = $parserReflectionProperty->getValue($phpstanParser);
-        $tokens = $innerParser->getTokens();
+        // keep only the anonymous class visitor
+        // remove all the rest, https://github.com/phpstan/phpstan-src/tree/1d86de8bb9371534983a8dbcd879e057d2ff028f/src/Parser
+        $tags[RichParser::VISITOR_SERVICE_TAG] = [
+            $container->findServiceNamesByType(AnonymousClassVisitor::class)[0] => true,
+        ];
 
-        $standardPrinter = new Standard([
-            'newline' => "\n",
-        ]);
-        $printerContents = $standardPrinter->printFormatPreserving($stmts, $stmts, $tokens);
-
-        $this->assertStringEqualsFile(__DIR__ . '/Fixture/some_array_map.php', $printerContents);
+        $privatesAccessor->setPrivateProperty($deeperContainer, 'tags', $tags);
     }
 }