mitogen/tests/ssh_test.py

import unittest

import mitogen
import mitogen.ssh
import mitogen.utils

import testlib
import plain_old_module


class FakeSshTest(testlib.RouterMixin, unittest.TestCase):
    def test_okay(self):
        context = self.router.ssh(
                hostname='hostname',
                username='has-sudo',
                ssh_path=testlib.data_path('fakessh.py'),
        )
        #context.call(mitogen.utils.log_to_file, '/tmp/log')
        #context.call(mitogen.utils.disable_site_packages)
        self.assertEquals(3, context.call(plain_old_module.add, 1, 2))


class SshTest(testlib.DockerMixin, unittest.TestCase):
    stream_class = mitogen.ssh.Stream

    def test_password_required(self):
        try:
            context = self.router.ssh(
                hostname=self.dockerized_ssh.host,
                port=self.dockerized_ssh.port,
                check_host_keys=False,
                username='has-sudo',
            )
            assert 0, 'exception not thrown'
        except mitogen.ssh.PasswordError, e:
            pass

        assert e[0] == self.stream_class.password_required_msg

    def test_password_incorrect(self):
        try:
            context = self.router.ssh(
                hostname=self.dockerized_ssh.host,
                port=self.dockerized_ssh.port,
                check_host_keys=False,
                username='has-sudo',
                password='badpw',
            )
            assert 0, 'exception not thrown'
        except mitogen.ssh.PasswordError, e:
            pass

        assert e[0] == self.stream_class.password_incorrect_msg

    def test_password_specified(self):
        context = self.router.ssh(
            hostname=self.dockerized_ssh.host,
            port=self.dockerized_ssh.port,
            check_host_keys=False,
            username='has-sudo',
            password='y',
        )

        sentinel = 'i-am-mitogen-test-docker-image\n'
        assert sentinel == context.call(plain_old_module.get_sentinel_value)

    def test_pubkey_required(self):
        try:
            context = self.router.ssh(
                hostname=self.dockerized_ssh.host,
                port=self.dockerized_ssh.port,
                check_host_keys=False,
                username='has-sudo-pubkey',
            )
            assert 0, 'exception not thrown'
        except mitogen.ssh.PasswordError, e:
            pass

        assert e[0] == self.stream_class.password_required_msg

    def test_pubkey_specified(self):
        context = self.router.ssh(
            hostname=self.dockerized_ssh.host,
            port=self.dockerized_ssh.port,
            check_host_keys=False,
            username='has-sudo-pubkey',
            identity_file=testlib.data_path('docker/has-sudo-pubkey.key'),
        )

        sentinel = 'i-am-mitogen-test-docker-image\n'
        assert sentinel == context.call(plain_old_module.get_sentinel_value)