Chris@0: <?php
Chris@0: 
Chris@0: /*
Chris@0:  * This file is part of Psy Shell.
Chris@0:  *
Chris@0:  * (c) 2012-2017 Justin Hileman
Chris@0:  *
Chris@0:  * For the full copyright and license information, please view the LICENSE
Chris@0:  * file that was distributed with this source code.
Chris@0:  */
Chris@0: 
Chris@0: namespace Psy\Test\CodeCleaner;
Chris@0: 
Chris@0: use PhpParser\NodeTraverser;
Chris@0: use Psy\CodeCleaner\CalledClassPass;
Chris@0: 
Chris@0: class CalledClassPassTest extends CodeCleanerTestCase
Chris@0: {
Chris@0:     public function setUp()
Chris@0:     {
Chris@0:         $this->pass      = new CalledClassPass();
Chris@0:         $this->traverser = new NodeTraverser();
Chris@0:         $this->traverser->addVisitor($this->pass);
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * @dataProvider invalidStatements
Chris@0:      * @expectedException \Psy\Exception\ErrorException
Chris@0:      */
Chris@0:     public function testProcessStatementFails($code)
Chris@0:     {
Chris@0:         $stmts = $this->parse($code);
Chris@0:         $this->traverser->traverse($stmts);
Chris@0:     }
Chris@0: 
Chris@0:     public function invalidStatements()
Chris@0:     {
Chris@0:         return array(
Chris@0:             array('get_class()'),
Chris@0:             array('get_class(null)'),
Chris@0:             array('get_called_class()'),
Chris@0:             array('get_called_class(null)'),
Chris@0:             array('function foo() { return get_class(); }'),
Chris@0:             array('function foo() { return get_class(null); }'),
Chris@0:             array('function foo() { return get_called_class(); }'),
Chris@0:             array('function foo() { return get_called_class(null); }'),
Chris@0:         );
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * @dataProvider validStatements
Chris@0:      */
Chris@0:     public function testProcessStatementPasses($code)
Chris@0:     {
Chris@0:         $stmts = $this->parse($code);
Chris@0:         $this->traverser->traverse($stmts);
Chris@0:     }
Chris@0: 
Chris@0:     public function validStatements()
Chris@0:     {
Chris@0:         return array(
Chris@0:             array('get_class($foo)'),
Chris@0:             array('get_class(bar())'),
Chris@0:             array('get_called_class($foo)'),
Chris@0:             array('get_called_class(bar())'),
Chris@0:             array('function foo($bar) { return get_class($bar); }'),
Chris@0:             array('function foo($bar) { return get_called_class($bar); }'),
Chris@0:             array('class Foo { function bar() { return get_class(); } }'),
Chris@0:             array('class Foo { function bar() { return get_class(null); } }'),
Chris@0:             array('class Foo { function bar() { return get_called_class(); } }'),
Chris@0:             array('class Foo { function bar() { return get_called_class(null); } }'),
Chris@0:             array('$foo = function () {}; $foo()'),
Chris@0:         );
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * @dataProvider validTraitStatements
Chris@0:      */
Chris@0:     public function testProcessTraitStatementPasses($code)
Chris@0:     {
Chris@0:         if (version_compare(PHP_VERSION, '5.4', '<')) {
Chris@0:             $this->markTestSkipped();
Chris@0:         }
Chris@0: 
Chris@0:         $stmts = $this->parse($code);
Chris@0:         $this->traverser->traverse($stmts);
Chris@0:     }
Chris@0: 
Chris@0:     public function validTraitStatements()
Chris@0:     {
Chris@0:         return array(
Chris@0:             array('trait Foo { function bar() { return get_class(); } }'),
Chris@0:             array('trait Foo { function bar() { return get_class(null); } }'),
Chris@0:             array('trait Foo { function bar() { return get_called_class(); } }'),
Chris@0:             array('trait Foo { function bar() { return get_called_class(null); } }'),
Chris@0:         );
Chris@0:     }
Chris@0: }