gnu-social/vendor/psy/psysh/test/CodeCleaner/CodeCleanerTestCase.php
2020-08-07 23:42:38 +01:00

42 lines
909 B
PHP

<?php
/*
* This file is part of Psy Shell.
*
* (c) 2012-2018 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 Psy\CodeCleaner\CodeCleanerPass;
use Psy\Test\ParserTestCase;
class CodeCleanerTestCase extends ParserTestCase
{
protected $pass;
public function tearDown()
{
$this->pass = null;
parent::tearDown();
}
protected function setPass(CodeCleanerPass $pass)
{
$this->pass = $pass;
if (!isset($this->traverser)) {
$this->traverser = new NodeTraverser();
}
$this->traverser->addVisitor($this->pass);
}
protected function parseAndTraverse($code, $prefix = '<?php ')
{
return $this->traverse($this->parse($code, $prefix));
}
}