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