<?php

/*
 * This file is part of Psy Shell.
 *
 * (c) 2012-2017 Justin Hileman
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Psy\Test\CodeCleaner;

use Psy\CodeCleaner\ExitPass;

class ExitPassTest extends CodeCleanerTestCase
{
    /**
     * @var string
     */
    private $expectedExceptionString = "\\Psy\\Exception\\BreakException::exit_();";

    public function setUp()
    {
        $this->setPass(new ExitPass());
    }

    /**
     * @dataProvider dataProviderExitStatement
     */
    public function testExitStatement($from, $to)
    {
        $this->assertProcessesAs($from, $to);
    }

    /**
     * Data provider for testExitStatement.
     *
     * @return array
     */
    public function dataProviderExitStatement()
    {
        return array(
            array('exit;', $this->expectedExceptionString),
            array('exit();', $this->expectedExceptionString),
            array('die;', $this->expectedExceptionString),
            array('if (true) { exit; }', "if (true) {\n    $this->expectedExceptionString\n}"),
            array('if (false) { exit; }', "if (false) {\n    $this->expectedExceptionString\n}"),
        );
    }
}

<?php

/*
 * This file is part of Psy Shell.
 *
 * (c) 2012-2017 Justin Hileman
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Psy\Test\CodeCleaner;

use Psy\CodeCleaner\ExitPass;

class ExitPassTest extends CodeCleanerTestCase
{
    /**
     * @var string
     */
    private $expectedExceptionString = "\\Psy\\Exception\\BreakException::exit_()";

    public function setUp()
    {
        $this->setPass(new ExitPass());
    }

    /**
     * @dataProvider dataProviderExitStatement
     */
    public function testExitStatement($from, $to)
    {
        $this->assertProcessesAs($from, $to);
    }

    /**
     * Data provider for testExitStatement.
     *
     * @return array
     */
    public function dataProviderExitStatement()
    {
        return array(
            array('exit;', "{$this->expectedExceptionString};"),
            array('exit();', "{$this->expectedExceptionString};"),
            array('die;', "{$this->expectedExceptionString};"),
            array('exit(die(die));', "{$this->expectedExceptionString};"),
            array('if (true) { exit; }', "if (true) {\n    {$this->expectedExceptionString};\n}"),
            array('if (false) { exit; }', "if (false) {\n    {$this->expectedExceptionString};\n}"),
            array('1 and exit();', "1 and {$this->expectedExceptionString};"),
            array('foo() or die', "foo() or {$this->expectedExceptionString};"),
            array('exit and 1;', "{$this->expectedExceptionString} and 1;"),
        );
    }
}

Add test cases for ExitPass
Add test cases for ExitPass
