update RectorNodeTraverser test

Author: TomasVotruba

src/PhpParser/NodeTraverser/RectorNodeTraverser.php

@@ -64,7 +64,11 @@ public function refreshPhpRectors(array $rectors): void
      */
     public function getVisitorsForNode(Node $node): array
     {
-        Assert::true($this->areNodeVisitorsPrepared);
+        // only in tests
+        if ($this->areNodeVisitorsPrepared === false) {
+            $this->prepareNodeVisitors();
+        }
+
         return $this->visitorsPerNodeClass[$node::class] ?? [];
     }
 
@@ -81,6 +85,7 @@ private function prepareNodeVisitors(): void
 
         // filer out by version
         $this->visitors = $this->phpVersionedFilter->filter($this->rectors);
+
         // filter by configuration
         $this->visitors = $this->configurationRuleFilter->filter($this->visitors);
 

tests/PhpParser/NodeTraverser/RectorNodeTraverserTest.php

@@ -48,7 +48,9 @@ public function testGetVisitorsForNodeWhenNoVisitorsAvailable(): void
     public function testGetVisitorsForNodeWhenNoVisitorsMatch(): void
     {
         $class = new Class_('test');
-        $this->rectorNodeTraverser->addVisitor($this->ruleUsingFunctionRector);
+        $this->rectorNodeTraverser->refreshPhpRectors([
+            $this->ruleUsingFunctionRector,
+        ]);
 
         $visitors = $this->rectorNodeTraverser->getVisitorsForNode($class);
 
@@ -58,8 +60,11 @@ public function testGetVisitorsForNodeWhenNoVisitorsMatch(): void
     public function testGetVisitorsForNodeWhenSomeVisitorsMatch(): void
     {
         $class = new Class_('test');
-        $this->rectorNodeTraverser->addVisitor($this->ruleUsingFunctionRector);
-        $this->rectorNodeTraverser->addVisitor($this->ruleUsingClassRector);
+
+        $this->rectorNodeTraverser->refreshPhpRectors([
+            new RuleUsingFunctionRector(),
+            new RuleUsingClassRector()
+        ]);
 
         $visitors = $this->rectorNodeTraverser->getVisitorsForNode($class);
 
@@ -69,19 +74,23 @@ public function testGetVisitorsForNodeWhenSomeVisitorsMatch(): void
     public function testGetVisitorsForNodeWhenAllVisitorsMatch(): void
     {
         $class = new Class_('test');
-        $this->rectorNodeTraverser->addVisitor($this->ruleUsingClassRector);
-        $this->rectorNodeTraverser->addVisitor($this->ruleUsingClassLikeRector);
+        $this->rectorNodeTraverser->refreshPhpRectors([
+            $this->ruleUsingClassRector,
+            $this->ruleUsingClassLikeRector
+        ]);
 
-        $visitors = $this->rectorNodeTraverser->getVisitorsForNode($class);
+        $visitorsForNode = $this->rectorNodeTraverser->getVisitorsForNode($class);
 
-        $this->assertEquals([$this->ruleUsingClassRector, $this->ruleUsingClassLikeRector], $visitors);
+        $this->assertEquals([$this->ruleUsingClassRector, $this->ruleUsingClassLikeRector], $visitorsForNode);
     }
 
     public function testGetVisitorsForNodeUsesCachedValue(): void
     {
         $class = new Class_('test');
-        $this->rectorNodeTraverser->addVisitor($this->ruleUsingClassRector);
-        $this->rectorNodeTraverser->addVisitor($this->ruleUsingClassLikeRector);
+        $this->rectorNodeTraverser->refreshPhpRectors([
+            $this->ruleUsingClassRector,
+            $this->ruleUsingClassLikeRector,
+        ]);
 
         $visitors = $this->rectorNodeTraverser->getVisitorsForNode($class);