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 Psy\CodeCleaner\FunctionContextPass;
Chris@13: 
Chris@13: class FunctionContextPassTest extends CodeCleanerTestCase
Chris@13: {
Chris@13:     public function setUp()
Chris@13:     {
Chris@13:         $this->setPass(new FunctionContextPass());
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * @dataProvider validStatements
Chris@13:      */
Chris@13:     public function testProcessStatementPasses($code)
Chris@13:     {
Chris@13:         $this->parseAndTraverse($code);
Chris@13:         $this->assertTrue(true);
Chris@13:     }
Chris@13: 
Chris@13:     public function validStatements()
Chris@13:     {
Chris@13:         return [
Chris@13:             ['function foo() { yield; }'],
Chris@13:             ['if (function(){ yield; })'],
Chris@13:         ];
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * @dataProvider invalidYieldStatements
Chris@13:      * @expectedException \Psy\Exception\FatalErrorException
Chris@13:      */
Chris@13:     public function testInvalidYield($code)
Chris@13:     {
Chris@13:         $this->parseAndTraverse($code);
Chris@13:     }
Chris@13: 
Chris@13:     public function invalidYieldStatements()
Chris@13:     {
Chris@13:         return [
Chris@13:             ['yield'],
Chris@13:             ['if (yield)'],
Chris@13:         ];
Chris@13:     }
Chris@13: }