Chris@13: <?php
Chris@13: 
Chris@13: /*
Chris@13:  * This file is part of Psy Shell.
Chris@13:  *
Chris@13:  * (c) 2012-2018 Justin Hileman
Chris@13:  *
Chris@13:  * For the full copyright and license information, please view the LICENSE
Chris@13:  * file that was distributed with this source code.
Chris@13:  */
Chris@13: 
Chris@13: namespace Psy\Test\CodeCleaner;
Chris@13: 
Chris@13: use PhpParser\NodeTraverser;
Chris@13: use Psy\CodeCleaner\CodeCleanerPass;
Chris@13: use Psy\Test\ParserTestCase;
Chris@13: 
Chris@13: class CodeCleanerTestCase extends ParserTestCase
Chris@13: {
Chris@13:     protected $pass;
Chris@13: 
Chris@16:     public function tearDown()
Chris@16:     {
Chris@16:         $this->pass = null;
Chris@16:         parent::tearDown();
Chris@16:     }
Chris@16: 
Chris@13:     protected function setPass(CodeCleanerPass $pass)
Chris@13:     {
Chris@13:         $this->pass = $pass;
Chris@13:         if (!isset($this->traverser)) {
Chris@13:             $this->traverser = new NodeTraverser();
Chris@13:         }
Chris@13:         $this->traverser->addVisitor($this->pass);
Chris@13:     }
Chris@13: 
Chris@13:     protected function parseAndTraverse($code, $prefix = '<?php ')
Chris@13:     {
Chris@13:         return $this->traverse($this->parse($code, $prefix));
Chris@13:     }
Chris@13: }