First commit

vendor/psy/psysh/test/Psy/Test/CodeCleaner/CodeCleanerTestCase.php

@@ -0,0 +1,97 @@
+<?php
+
+/*
+ * This file is part of Psy Shell.
+ *
+ * (c) 2012-2017 Justin Hileman
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Psy\Test\CodeCleaner;
+
+use PhpParser\NodeTraverser;
+use PhpParser\PrettyPrinter\Standard as Printer;
+use Psy\CodeCleaner\CodeCleanerPass;
+use Psy\Exception\ParseErrorException;
+use Psy\ParserFactory;
+
+class CodeCleanerTestCase extends \PHPUnit\Framework\TestCase
+{
+    protected $pass;
+    protected $traverser;
+    private $parser;
+    private $printer;
+
+    protected function setPass(CodeCleanerPass $pass)
+    {
+        $this->pass = $pass;
+        if (!isset($this->traverser)) {
+            $this->traverser = new NodeTraverser();
+        }
+        $this->traverser->addVisitor($this->pass);
+    }
+
+    protected function parse($code, $prefix = '<?php ')
+    {
+        $code = $prefix . $code;
+        try {
+            return $this->getParser()->parse($code);
+        } catch (\PhpParser\Error $e) {
+            if (!$this->parseErrorIsEOF($e)) {
+                throw ParseErrorException::fromParseError($e);
+            }
+
+            try {
+                // Unexpected EOF, try again with an implicit semicolon
+                return $this->getParser()->parse($code . ';');
+            } catch (\PhpParser\Error $e) {
+                return false;
+            }
+        }
+    }
+
+    protected function traverse(array $stmts)
+    {
+        return $this->traverser->traverse($stmts);
+    }
+
+    protected function prettyPrint(array $stmts)
+    {
+        return $this->getPrinter()->prettyPrint($stmts);
+    }
+
+    protected function assertProcessesAs($from, $to)
+    {
+        $stmts = $this->parse($from);
+        $stmts = $this->traverse($stmts);
+        $this->assertEquals($to, $this->prettyPrint($stmts));
+    }
+
+    private function getParser()
+    {
+        if (!isset($this->parser)) {
+            $parserFactory = new ParserFactory();
+            $this->parser  = $parserFactory->createParser();
+        }
+
+        return $this->parser;
+    }
+
+    private function getPrinter()
+    {
+        if (!isset($this->printer)) {
+            $this->printer = new Printer();
+        }
+
+        return $this->printer;
+    }
+
+    private function parseErrorIsEOF(\PhpParser\Error $e)
+    {
+        $msg = $e->getRawMessage();
+
+        return ($msg === 'Unexpected token EOF') || (strpos($msg, 'Syntax error, unexpected EOF') !== false);
+    }
+}