rector-src/tests/PhpParser/NodeTraverser/RectorNodeTraverserTest.php at 7cc5bc7da66f953e5929decc818f1127ca39e595 · rectorphp/rector-src · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
<?php

declare(strict_types=1);

namespace Rector\Tests\PhpParser\NodeTraverser;

use PhpParser\Node\Stmt\Class_;
use Rector\PhpParser\NodeTraverser\RectorNodeTraverser;
use Rector\Testing\PHPUnit\AbstractLazyTestCase;
use Rector\Tests\PhpParser\NodeTraverser\Class_\RuleUsingClassRector;
use Rector\Tests\PhpParser\NodeTraverser\ClassLike\RuleUsingClassLikeRector;
use Rector\Tests\PhpParser\NodeTraverser\Function_\RuleUsingFunctionRector;

/**
 * @see \Rector\PhpParser\NodeTraverser\AbstractImmutableNodeTraverser
 */
final class RectorNodeTraverserTest extends AbstractLazyTestCase
{
    private RectorNodeTraverser $rectorNodeTraverser;

    private RuleUsingFunctionRector $ruleUsingFunctionRector;

    private RuleUsingClassRector $ruleUsingClassRector;

    private RuleUsingClassLikeRector $ruleUsingClassLikeRector;

    protected function setUp(): void
    {
        parent::setUp();

        $this->rectorNodeTraverser = $this->make(RectorNodeTraverser::class);
        $this->rectorNodeTraverser->refreshPhpRectors([]);

        $this->ruleUsingFunctionRector = new RuleUsingFunctionRector();
        $this->ruleUsingClassRector = new RuleUsingClassRector();
        $this->ruleUsingClassLikeRector = new RuleUsingClassLikeRector();
    }

    public function testGetVisitorsForNodeWhenNoVisitorsAvailable(): void
    {
        $class = new Class_('test');
        $visitorsForNode = $this->rectorNodeTraverser->getVisitorsForNode($class);

        $this->assertSame([], $visitorsForNode);
    }

    public function testGetVisitorsForNodeWhenNoVisitorsMatch(): void
    {
        $class = new Class_('test');
        $this->rectorNodeTraverser->refreshPhpRectors([$this->ruleUsingFunctionRector]);

        $visitorsForNode = $this->rectorNodeTraverser->getVisitorsForNode($class);
        $this->assertSame([], $visitorsForNode);
    }

    public function testGetVisitorsForNodeWhenSomeVisitorsMatch(): void
    {
        $class = new Class_('test');

        $this->rectorNodeTraverser->refreshPhpRectors([new RuleUsingFunctionRector(), new RuleUsingClassRector()]);

        $visitorsForNode = $this->rectorNodeTraverser->getVisitorsForNode($class);
        $this->assertEquals([$this->ruleUsingClassRector], $visitorsForNode);
    }

    public function testGetVisitorsForNodeWhenAllVisitorsMatch(): void
    {
        $class = new Class_('test');
        $this->rectorNodeTraverser->refreshPhpRectors([
            $this->ruleUsingClassRector,
            $this->ruleUsingClassLikeRector,
        ]);

        $visitorsForNode = $this->rectorNodeTraverser->getVisitorsForNode($class);
        $this->assertEquals([$this->ruleUsingClassRector, $this->ruleUsingClassLikeRector], $visitorsForNode);
    }
}