X-Git-Url: http://www.aleph1.co.uk/gitweb/?p=yaffs-website;a=blobdiff_plain;f=vendor%2Fnikic%2Fphp-parser%2Ftest%2FPhpParser%2FNodeVisitor%2FFindingVisitorTest.php;fp=vendor%2Fnikic%2Fphp-parser%2Ftest%2FPhpParser%2FNodeVisitor%2FFindingVisitorTest.php;h=2e87600b6975787a27517165c1dde76a65dad40d;hp=0000000000000000000000000000000000000000;hb=af6d1fb995500ae68849458ee10d66abbdcfb252;hpb=680c79a86e3ed402f263faeac92e89fb6d9edcc0

diff --git a/vendor/nikic/php-parser/test/PhpParser/NodeVisitor/FindingVisitorTest.php b/vendor/nikic/php-parser/test/PhpParser/NodeVisitor/FindingVisitorTest.php
new file mode 100644
index 000000000..2e87600b6
--- /dev/null
+++ b/vendor/nikic/php-parser/test/PhpParser/NodeVisitor/FindingVisitorTest.php
@@ -0,0 +1,54 @@
+<?php declare(strict_types=1);
+
+namespace PhpParser\NodeVisitor;
+
+use PhpParser\Node;
+use PhpParser\Node\Expr;
+use PhpParser\NodeTraverser;
+use PHPUnit\Framework\TestCase;
+
+class FindingVisitorTest extends TestCase
+{
+    public function testFindVariables() {
+        $traverser = new NodeTraverser();
+        $visitor = new FindingVisitor(function(Node $node) {
+            return $node instanceof Node\Expr\Variable;
+        });
+        $traverser->addVisitor($visitor);
+
+        $assign = new Expr\Assign(new Expr\Variable('a'), new Expr\BinaryOp\Concat(
+            new Expr\Variable('b'), new Expr\Variable('c')
+        ));
+        $stmts = [new Node\Stmt\Expression($assign)];
+
+        $traverser->traverse($stmts);
+        $this->assertSame([
+            $assign->var,
+            $assign->expr->left,
+            $assign->expr->right,
+        ], $visitor->getFoundNodes());
+    }
+
+    public function testFindAll() {
+        $traverser = new NodeTraverser();
+        $visitor = new FindingVisitor(function(Node $node) {
+            return true; // All nodes
+        });
+        $traverser->addVisitor($visitor);
+
+        $assign = new Expr\Assign(new Expr\Variable('a'), new Expr\BinaryOp\Concat(
+            new Expr\Variable('b'), new Expr\Variable('c')
+        ));
+        $stmts = [new Node\Stmt\Expression($assign)];
+
+        $traverser->traverse($stmts);
+        $this->assertSame([
+            $stmts[0],
+            $assign,
+            $assign->var,
+            $assign->expr,
+            $assign->expr->left,
+            $assign->expr->right,
+        ], $visitor->getFoundNodes());
+    }
+}