aleph1.co.uk Git - yaffs-website/blob - vendor/psy/psysh/test/CodeCleaner/CodeCleanerTestCase.php

   1 <?php
   2
   3 /*
   4  * This file is part of Psy Shell.
   5  *
   6  * (c) 2012-2018 Justin Hileman
   7  *
   8  * For the full copyright and license information, please view the LICENSE
   9  * file that was distributed with this source code.
  10  */
  11
  12 namespace Psy\Test\CodeCleaner;
  13
  14 use PhpParser\NodeTraverser;
  15 use Psy\CodeCleaner\CodeCleanerPass;
  16 use Psy\Test\ParserTestCase;
  17
  18 class CodeCleanerTestCase extends ParserTestCase
  19 {
  20     protected $pass;
  21
  22     protected function setPass(CodeCleanerPass $pass)
  23     {
  24         $this->pass = $pass;
  25         if (!isset($this->traverser)) {
  26             $this->traverser = new NodeTraverser();
  27         }
  28         $this->traverser->addVisitor($this->pass);
  29     }
  30
  31     protected function parseAndTraverse($code, $prefix = '<?php ')
  32     {
  33         return $this->traverse($this->parse($code, $prefix));
  34     }
  35 }