SSHConnectionTest.php

<?php

use DivineOmega\SSHConnection\SSHConnection;
use PHPUnit\Framework\TestCase;

final class SSHConnectionTest extends TestCase
{
    public function testUpload()
    {
        $connection = (new SSHConnection())
            ->to('localhost')
            ->onPort(22)
            ->as('travis')
            ->withPrivateKey('/home/travis/.ssh/id_rsa')
            ->connect();

        $remotePath =  __DIR__ . '/../fixtures/upload.txt';
        $localPath = __DIR__ . '/../fixtures/file.txt';

        $this->assertTrue($connection->upload($localPath, $remotePath));
        $this->assertFileExists($remotePath);
    }

    public function testDownload()
    {
        $connection = (new SSHConnection())
            ->to('localhost')
            ->onPort(22)
            ->as('travis')
            ->withPrivateKey('/home/travis/.ssh/id_rsa')
            ->connect();

        $remotePath =  __DIR__ . '/../fixtures/file.txt';
        $localPath = __DIR__ . '/../fixtures/download.txt';

        $this->assertTrue($connection->download($remotePath, $localPath));
        $this->assertFileExists($localPath);
    }

    public function testSSHConnectionWithKeyPair()
    {
        $connection = (new SSHConnection())
            ->to('localhost')
            ->onPort(22)
            ->as('travis')
            ->withPrivateKey('/home/travis/.ssh/id_rsa')
            ->connect();

        $command = $connection->run('echo "Hello world!"');

        $this->assertEquals('Hello world!', $command->getOutput());
        $this->assertEquals('Hello world!'."\n", $command->getRawOutput());

        $this->assertEquals('', $command->getError());
        $this->assertEquals('', $command->getRawError());
    }

    public function testSSHConnectionWithKeyContents()
    {
        $privateKeyContents = file_get_contents('/home/travis/.ssh/id_rsa');

        $connection = (new SSHConnection())
            ->to('localhost')
            ->onPort(22)
            ->as('travis')
            ->withPrivateKeyString($privateKeyContents)
            ->connect();

        $command = $connection->run('echo "Hello world!"');

        $this->assertEquals('Hello world!', $command->getOutput());
        $this->assertEquals('Hello world!'."\n", $command->getRawOutput());

        $this->assertEquals('', $command->getError());
        $this->assertEquals('', $command->getRawError());
    }

    public function testSSHConnectionWithPassword()
    {
        $connection = (new SSHConnection())
            ->to('test.rebex.net')
            ->onPort(22)
            ->as('demo')
            ->withPassword('password')
            ->connect();

        $command = $connection->run('echo "Hello world!"');

        $this->assertEquals('Hello world!', $command->getOutput());
        $this->assertEquals('Hello world!'."\n", $command->getRawOutput());

        $this->assertEquals('', $command->getError());
        $this->assertEquals('', $command->getRawError());
    }

    public function testMd5Fingerprint()
    {
        $connection1 = (new SSHConnection())
            ->to('localhost')
            ->onPort(22)
            ->as('travis')
            ->withPrivateKey('/home/travis/.ssh/id_rsa')
            ->connect();

        $connection2 = (new SSHConnection())
            ->to('localhost')
            ->onPort(22)
            ->as('travis')
            ->withPrivateKey('/home/travis/.ssh/id_rsa')
            ->connect();

        $this->assertEquals(
            $connection1->fingerprint(SSHConnection::FINGERPRINT_MD5),
            $connection2->fingerprint(SSHConnection::FINGERPRINT_MD5)
        );
    }

    public function testSha1Fingerprint()
    {
        $connection1 = (new SSHConnection())
            ->to('localhost')
            ->onPort(22)
            ->as('travis')
            ->withPrivateKey('/home/travis/.ssh/id_rsa')
            ->connect();

        $connection2 = (new SSHConnection())
            ->to('localhost')
            ->onPort(22)
            ->as('travis')
            ->withPrivateKey('/home/travis/.ssh/id_rsa')
            ->connect();

        $this->assertEquals(
            $connection1->fingerprint(SSHConnection::FINGERPRINT_SHA1),
            $connection2->fingerprint(SSHConnection::FINGERPRINT_SHA1)
        );
    }

    public function testMd5FingerprintFailure()
    {
        $connection1 = (new SSHConnection())
            ->to('localhost')
            ->onPort(22)
            ->as('travis')
            ->withPrivateKey('/home/travis/.ssh/id_rsa')
            ->connect();

        $connection2 = (new SSHConnection())
            ->to('test.rebex.net')
            ->onPort(22)
            ->as('demo')
            ->withPassword('password')
            ->connect();

        $this->assertNotEquals(
            $connection1->fingerprint(SSHConnection::FINGERPRINT_MD5),
            $connection2->fingerprint(SSHConnection::FINGERPRINT_MD5)
        );
    }

    public function testSha1FingerprintFailure()
    {
        $connection1 = (new SSHConnection())
            ->to('localhost')
            ->onPort(22)
            ->as('travis')
            ->withPrivateKey('/home/travis/.ssh/id_rsa')
            ->connect();

        $connection2 = (new SSHConnection())
            ->to('test.rebex.net')
            ->onPort(22)
            ->as('demo')
            ->withPassword('password')
            ->connect();

        $this->assertNotEquals(
            $connection1->fingerprint(SSHConnection::FINGERPRINT_SHA1),
            $connection2->fingerprint(SSHConnection::FINGERPRINT_SHA1)
        );
    }
}