From 3a0f03183cd1f1b83a98dee40eb3fd5dc4634df5 Mon Sep 17 00:00:00 2001
From: David Wilson <dw@botanicus.net>
Date: Sun, 17 Sep 2017 21:19:06 +0530
Subject: [PATCH] Simple working fakessh_test.

---
 tests/fakessh_test.py | 50 +++++++++++++------------------------------
 tests/ssh_test.py     | 26 +++++-----------------
 tests/testlib.py      | 12 +++++++++++
 3 files changed, 32 insertions(+), 56 deletions(-)

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',
+        )