phpseclib/tests/Unit/Net/SSH2Test.php

<?php

/**
 * @author    Marc Scholten <marc@pedigital.de>
 * @copyright MMXIII Marc Scholten
 * @license   http://www.opensource.org/licenses/mit-license.html  MIT License
 */

class Unit_Net_SSH2Test extends PhpseclibTestCase
{
    public function formatLogDataProvider()
    {
        return array(
            array(
                array('hello world'),
                array('<--'),
                "<--\r\n00000000  68:65:6c:6c:6f:20:77:6f:72:6c:64                 hello world\r\n\r\n"
            ),
            array(
                array('hello', 'world'),
                array('<--', '<--'),
                "<--\r\n00000000  68:65:6c:6c:6f                                   hello\r\n\r\n" .
                "<--\r\n00000000  77:6f:72:6c:64                                   world\r\n\r\n"
            ),
        );
    }

    /**
     * @dataProvider formatLogDataProvider
     */
    public function testFormatLog(array $message_log, array $message_number_log, $expected)
    {
        $ssh = $this->createSSHMock();

        $result = $ssh->_format_log($message_log, $message_number_log);
        $this->assertEquals($expected, $result);
    }

    public function generateIdentifierProvider()
    {
        return array(
            array('SSH-2.0-phpseclib_0.3', array()),
            array('SSH-2.0-phpseclib_0.3 (gmp)', array('gmp')),
            array('SSH-2.0-phpseclib_0.3 (bcmath)', array('bcmath')),
            array('SSH-2.0-phpseclib_0.3 (mcrypt)', array('mcrypt')),
            array('SSH-2.0-phpseclib_0.3 (mcrypt, gmp)', array('mcrypt', 'gmp')),
            array('SSH-2.0-phpseclib_0.3 (mcrypt, bcmath)', array('mcrypt', 'bcmath')),
        );
    }

    /**
     * @dataProvider generateIdentifierProvider
     */
    public function testGenerateIdentifier($expected, array $requiredExtensions)
    {
        $notAllowed = array('gmp', 'bcmath', 'mcrypt', 'gmp');
        foreach ($notAllowed as $notAllowedExtension) {
            if (in_array($notAllowedExtension, $requiredExtensions)) {
                continue;
            }

            if (extension_loaded($notAllowedExtension)) {
                $this->markTestSkipped('Extension ' . $notAllowedExtension . ' is not allowed for this data-set');
            }
        }

        $ssh = $this->createSSHMock();
        $identifier = $ssh->_generate_identifier();

        $this->assertEquals($expected, $identifier);
    }

    public function testGetExitStatusIfNotConnected()
    {
        $ssh = $this->createSSHMock();

        $this->assertFalse($ssh->getExitStatus());
    }

    public function testPTYIDefaultValue()
    {
        $ssh = $this->createSSHMock();
        $this->assertFalse($ssh->isPTYEnabled());
    }

    public function testEnablePTY()
    {
        $ssh = $this->createSSHMock();

        $ssh->enablePTY();
        $this->assertTrue($ssh->isPTYEnabled());

        $ssh->disablePTY();
        $this->assertFalse($ssh->isPTYEnabled());
    }

    public function testQuietModeDefaultValue()
    {
        $ssh = $this->createSSHMock();

        $this->assertFalse($ssh->isQuietModeEnabled());
    }

    public function testEnableQuietMode()
    {
        $ssh = $this->createSSHMock();

        $ssh->enableQuietMode();
        $this->assertTrue($ssh->isQuietModeEnabled());

        $ssh->disableQuietMode();
        $this->assertFalse($ssh->isQuietModeEnabled());
    }

    /**
     * @return Net_SSH2
     */
    protected function createSSHMock()
    {
        return $this->getMockBuilder('Net_SSH2')
            ->disableOriginalConstructor()
            ->setMethods(array('__destruct'))
            ->getMock();
    }
}