This repository has been archived on 2025-01-19. You can view files and clone it, but cannot push or open issues or pull requests.
drupalcampbristol/vendor/psy/psysh/test/CodeCleaner/CodeCleanerTestCase.php
2018-11-23 12:29:20 +00: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));
}
}