fix unittest; will skip test when cannot build ssh connections

Author: felix5572

tests/test_ssh_context.py

@@ -1,28 +1,47 @@
 import os,sys,json,glob,shutil,uuid,getpass
 import unittest
-from pathlib import Path
+import pathlib
 from paramiko.ssh_exception import NoValidConnectionsError
+from paramiko.ssh_exception import SSHException
 
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 __package__ = 'tests'
 from .context import SSHContext, SSHSession
+from .context import Machine
+from .sample_class import SampleClass
 
 class TestSSHContext(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        with open('jsons/machine_ali_ehpc.json', 'r') as f:
+            mdata = json.load(f)
+        try:
+            cls.machine = Machine.load_from_dict(mdata['machine'])
+        except SSHException:
+            self.skipTest("SSHException cannot connect")
+        cls.submission = SampleClass.get_sample_submission()
+        cls.submission.bind_machine(cls.machine)
+        cls.submission_hash = cls.submission.submission_hash
+        file_list = ['bct-1/log.lammps', 'bct-2/log.lammps', 'bct-3/log.lammps', 'bct-4/log.lammps']
+        for file in file_list:
+            cls.machine.context.write_file(file, '# mock log')
+    
     def setUp(self):
-        self.tmp_local_root = 'test_context_dir/'
-        self.tmp_remote_root = 'tmp_ssh_context_dir/'
+        self.context = self.__class__.machine.context
 
-        self.username = getpass.getuser()
-        # try:
-        self.ssh_session = SSHSession(
-            hostname='localhost',
-            username=self.username
+    def test_ssh_session(self):
+        self.assertIsInstance(
+            self.__class__.machine.context.ssh_session, SSHSession
         )
-        # except NoValidConnectionsError:
-        #     self.skipTest("skip ssh tests due to ssh connection errors")
 
-    def test_ssh_session(self):
-        pass
-        # with self.
+    def test_upload(self):
+        self.context.upload(self.__class__.submission)
+        check_file_list = ['graph.pb', 'bct-1/conf.lmp', 'bct-4/input.lammps']
+        for file in check_file_list:
+            self.assertTrue(self.context.check_file_exists(os.path.join(self.context.remote_root, file)))
+
+    def test_download(self):
+        self.context.download(self.__class__.submission)
+