diff --git a/tests/fakessh_test.py b/tests/fakessh_test.py index 8fc83c22..8ff26e18 100644 --- a/tests/fakessh_test.py +++ b/tests/fakessh_test.py @@ -1,42 +1,22 @@ +import os +import shutil +import unittest + import mitogen.utils import mitogen.fakessh - -@mitogen.utils.run_with_router -def main(router): - import logging - mitogen.utils.log_to_file(level=logging.INFO) - router.enable_debug() - #mitogen.fakessh.run_with_fake_ssh(router, ['bash', '-c', 'echo $PATH']) - #mitogen.fakessh.run_with_fake_ssh(router, ['bash', '-c', 'ls -Gl $SSH_PATH']) - # k3 = router.ssh(hostname='k3') - k3 = router.local() - print 'GOT HERE' - import os - k3.call(os.system, 'hostname') - return - #sud = router.sudo(via=k3, username='root') - sud = k3 - - mitogen.fakessh.run_with_fake_ssh(sud, router, ['rsync', '--progress', '-vvvr', 'h:/var/lib/docker']) - #mitogen.fakessh.run_with_fake_ssh(router, ['bash', '-c', 'ssh h rsync --server . foo']) - # mitogen.fakessh.run_with_fake_ssh(router, ['bash', '-c', 'ssh k3.botanicus.net -t screen -dr']) - print 'end of t.py' +import testlib -import logging -import mitogen.utils -import mitogen.fakessh - -@mitogen.utils.run_with_router -def main(router): - mitogen.utils.log_to_file(level=logging.DEBUG) - #router.enable_debug() - #router.enable_debug() - - k3 = router.ssh(hostname='k3') - sudo = router.sudo(via=k3, username='root') - - mitogen.fakessh.run(sudo, router, ['rsync', '--progress', '-r', 'h:/var/lib/docker']) +class RsyncTest(testlib.DockerMixin, unittest.TestCase): + def test_rsync_from_master(self): + context = self.docker_ssh_any() + context.call(shutil.rmtree, '/tmp/data', ignore_errors=True) + mitogen.fakessh.run(context, self.router, [ + 'rsync', '--progress', '-vvva', + testlib.data_path('.'), 'target:/tmp/data' + ]) + assert context.call(os.path.exists, '/tmp/data') + assert context.call(os.path.exists, '/tmp/data/simple_pkg/a.py') diff --git a/tests/ssh_test.py b/tests/ssh_test.py index dabe0729..c18a5229 100644 --- a/tests/ssh_test.py +++ b/tests/ssh_test.py @@ -26,10 +26,7 @@ class SshTest(testlib.DockerMixin, unittest.TestCase): def test_password_required(self): try: - context = self.router.ssh( - hostname=self.dockerized_ssh.host, - port=self.dockerized_ssh.port, - check_host_keys=False, + context = self.docker_ssh( username='has-sudo', ) assert 0, 'exception not thrown' @@ -40,10 +37,7 @@ class SshTest(testlib.DockerMixin, unittest.TestCase): def test_password_incorrect(self): try: - context = self.router.ssh( - hostname=self.dockerized_ssh.host, - port=self.dockerized_ssh.port, - check_host_keys=False, + context = self.docker_ssh( username='has-sudo', password='badpw', ) @@ -54,10 +48,7 @@ class SshTest(testlib.DockerMixin, unittest.TestCase): 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, + context = self.docker_ssh( username='has-sudo', password='y', ) @@ -67,10 +58,7 @@ class SshTest(testlib.DockerMixin, unittest.TestCase): def test_pubkey_required(self): try: - context = self.router.ssh( - hostname=self.dockerized_ssh.host, - port=self.dockerized_ssh.port, - check_host_keys=False, + context = self.docker_ssh( username='has-sudo-pubkey', ) assert 0, 'exception not thrown' @@ -80,13 +68,9 @@ class SshTest(testlib.DockerMixin, unittest.TestCase): 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, + context = self.docker_ssh( 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) diff --git a/tests/testlib.py b/tests/testlib.py index 5f3f62d1..caf7297d 100644 --- a/tests/testlib.py +++ b/tests/testlib.py @@ -76,3 +76,15 @@ class DockerMixin(RouterMixin): def tearDownClass(cls): cls.dockerized_ssh.close() super(DockerMixin, cls).tearDownClass() + + def docker_ssh(self, **kwargs): + kwargs.setdefault('hostname', self.dockerized_ssh.host) + kwargs.setdefault('port', self.dockerized_ssh.port) + kwargs.setdefault('check_host_keys', False) + return self.router.ssh(**kwargs) + + def docker_ssh_any(self, **kwargs): + return self.docker_ssh( + username='has-sudo-nopw', + password='y', + )