+++ /dev/null
-<?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);
- }
-}