Chris@13: <?php declare(strict_types=1);
Chris@13: 
Chris@13: namespace PhpParser\NodeVisitor;
Chris@13: 
Chris@13: use PhpParser\Node;
Chris@13: use PhpParser\Node\Expr;
Chris@13: use PhpParser\NodeTraverser;
Chris@13: 
Chris@17: class FirstFindingVisitorTest extends \PHPUnit\Framework\TestCase
Chris@13: {
Chris@13:     public function testFindFirstVariable() {
Chris@13:         $traverser = new NodeTraverser();
Chris@13:         $visitor = new FirstFindingVisitor(function(Node $node) {
Chris@13:             return $node instanceof Node\Expr\Variable;
Chris@13:         });
Chris@13:         $traverser->addVisitor($visitor);
Chris@13: 
Chris@13:         $assign = new Expr\Assign(new Expr\Variable('a'), new Expr\Variable('b'));
Chris@13:         $stmts = [new Node\Stmt\Expression($assign)];
Chris@13: 
Chris@13:         $traverser->traverse($stmts);
Chris@13:         $this->assertSame($assign->var, $visitor->getFoundNode());
Chris@13:     }
Chris@13: 
Chris@13:     public function testFindNone() {
Chris@13:         $traverser = new NodeTraverser();
Chris@13:         $visitor = new FirstFindingVisitor(function(Node $node) {
Chris@13:             return $node instanceof Node\Expr\BinaryOp;
Chris@13:         });
Chris@13:         $traverser->addVisitor($visitor);
Chris@13: 
Chris@13:         $assign = new Expr\Assign(new Expr\Variable('a'), new Expr\Variable('b'));
Chris@13:         $stmts = [new Node\Stmt\Expression($assign)];
Chris@13: 
Chris@13:         $traverser->traverse($stmts);
Chris@16:         $this->assertNull($visitor->getFoundNode());
Chris@13:     }
Chris@13: }