def test_auto_add_policy(): """Test the auto add policy.""" with SshTransport(machine='localhost', timeout=30, load_system_host_keys=True, key_policy='AutoAddPolicy'): pass
def test_no_host_key(self): # Disable logging to avoid output during test logging.disable(logging.ERROR) with self.assertRaises(paramiko.SSHException): with SshTransport(machine='localhost', timeout=30, load_system_host_keys=False): pass # Reset logging level logging.disable(logging.NOTSET)
def test_gotocomputer(): """Test gotocomputer""" with SshTransport(machine='localhost', timeout=30, use_login_shell=False, key_policy='AutoAddPolicy') as transport: cmd_str = transport.gotocomputer_command('/remote_dir/') expected_str = ( """ssh -t localhost "if [ -d '/remote_dir/' ] ;""" """ then cd '/remote_dir/' ; bash ; else echo ' ** The directory' ; """ """echo ' ** /remote_dir/' ; echo ' ** seems to have been deleted, I logout...' ; fi" """ ) assert cmd_str == expected_str
def test_auto_add_policy(self): with SshTransport(machine='localhost', timeout=30, load_system_host_keys=True, key_policy='AutoAddPolicy'): pass
def test_invalid_param(self): with self.assertRaises(ValueError): SshTransport(machine='localhost', invalid_param=True)
def test_closed_connection_sftp(self): with self.assertRaises( aiida.transports.transport.TransportInternalError): t = SshTransport(machine='localhost') t.listdir()
def test_closed_connection_ssh(self): with self.assertRaises( aiida.transports.transport.TransportInternalError): t = SshTransport(machine='localhost') t._exec_command_internal('ls')
""" from __future__ import division from __future__ import print_function from __future__ import absolute_import import unittest import logging import aiida.transports import aiida.transports.transport import paramiko from aiida.transports.plugins.ssh import SshTransport # This will be used by test_all_plugins plugin_transport = SshTransport(machine='localhost', timeout=30, load_system_host_keys=True, key_policy='AutoAddPolicy') class TestBasicConnection(unittest.TestCase): """ Test basic connections. """ def test_closed_connection_ssh(self): with self.assertRaises( aiida.transports.transport.TransportInternalError): t = SshTransport(machine='localhost') t._exec_command_internal('ls') def test_closed_connection_sftp(self): with self.assertRaises(
def test_closed_connection_sftp(self): """Test calling sftp command on a closed connection.""" with self.assertRaises(TransportInternalError): transport = SshTransport(machine='localhost') transport.listdir()
def test_closed_connection_ssh(self): """Test calling command on a closed connection.""" with self.assertRaises(TransportInternalError): transport = SshTransport(machine='localhost') transport._exec_command_internal('ls') # pylint: disable=protected-access