Chris@16: <?php
Chris@16: 
Chris@16: /*
Chris@16:  * This file is part of Psy Shell.
Chris@16:  *
Chris@16:  * (c) 2012-2018 Justin Hileman
Chris@16:  *
Chris@16:  * For the full copyright and license information, please view the LICENSE
Chris@16:  * file that was distributed with this source code.
Chris@16:  */
Chris@16: 
Chris@16: namespace Psy\Test\Command\TimeitCommand;
Chris@16: 
Chris@16: use PhpParser\NodeTraverser;
Chris@16: use Psy\Command\TimeitCommand\TimeitVisitor;
Chris@16: use Psy\Test\ParserTestCase;
Chris@16: 
Chris@16: class TimeitVisitorTest extends ParserTestCase
Chris@16: {
Chris@16:     public function setUp()
Chris@16:     {
Chris@16:         $this->traverser = new NodeTraverser();
Chris@16:         $this->traverser->addVisitor(new TimeitVisitor());
Chris@16:     }
Chris@16: 
Chris@16:     /**
Chris@16:      * @dataProvider codez
Chris@16:      */
Chris@16:     public function testProcess($from, $to)
Chris@16:     {
Chris@16:         $this->assertProcessesAs($from, $to);
Chris@16:     }
Chris@16: 
Chris@16:     public function codez()
Chris@16:     {
Chris@16:         $start = '\Psy\Command\TimeitCommand::markStart';
Chris@16:         $end = '\Psy\Command\TimeitCommand::markEnd';
Chris@16:         $noReturn = 'new \Psy\CodeCleaner\NoReturnValue()';
Chris@16: 
Chris@16:         return [
Chris@16:             ['', "$end($start());"], // heh
Chris@16:             ['a()', "$start(); $end(a());"],
Chris@16:             ['$b()', "$start(); $end(\$b());"],
Chris@16:             ['$c->d()', "$start(); $end(\$c->d());"],
Chris@16:             ['e(); f()', "$start(); e(); $end(f());"],
Chris@16:             ['function g() { return 1; }', "$start(); function g() {return 1;} $end($noReturn);"],
Chris@16:             ['return 1', "$start(); return $end(1);"],
Chris@16:             ['return 1; 2', "$start(); return $end(1); $end(2);"],
Chris@16:             ['return 1; function h() {}', "$start(); return $end(1); function h() {} $end($noReturn);"],
Chris@16:         ];
Chris@16:     }
Chris@16: }