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\FinalClassPass;
Chris@0: 
Chris@0: class FinalClassPassTest extends CodeCleanerTestCase
Chris@0: {
Chris@0:     public function setUp()
Chris@0:     {
Chris@0:         $this->pass      = new FinalClassPass();
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\FatalErrorException
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:         $stmts = array(
Chris@0:             array('final class A {} class B extends A {}'),
Chris@0:             array('class A {} final class B extends A {} class C extends B {}'),
Chris@0:             // array('namespace A { final class B {} } namespace C { class D extends \\A\\B {} }'),
Chris@0:         );
Chris@0: 
Chris@0:         if (!defined('HHVM_VERSION')) {
Chris@0:             // For some reason Closure isn't final in HHVM?
Chris@0:             $stmts[] = array('class A extends \\Closure {}');
Chris@0:         }
Chris@0: 
Chris@0:         return $stmts;
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('class A extends \\stdClass {}'),
Chris@0:             array('final class A extends \\stdClass {}'),
Chris@0:             array('class A {} class B extends A {}'),
Chris@0:         );
Chris@0:     }
Chris@0: }