def test_ssh_connection_with_key_string(ssh_mock):
ssh_key = generate_ssh_key()
ssh_resource = SSHResource(
remote_host='remote_host',
remote_port=12345,
username='******',
password=None,
timeout=10,
key_string=six.ensure_str(ssh_key),
keepalive_interval=30,
compress=True,
no_host_key_check=False,
allow_host_key_change=False,
logger=logging.root.getChild('test_resources'),
)
with ssh_resource.get_connection():
ssh_mock.return_value.connect.assert_called_once_with(
hostname='remote_host',
username='******',
key_filename=None,
pkey=key_from_str(ssh_key),
timeout=10,
compress=True,
port=12345,
sock=None,
)
def test_tunnel_with_string_key(ssh_mock):
ssh_key = generate_ssh_key()
ssh_resource = SSHResource(
remote_host='remote_host',
remote_port=12345,
username='******',
password=None,
timeout=10,
key_string=six.ensure_str(ssh_key),
keepalive_interval=30,
compress=True,
no_host_key_check=False,
allow_host_key_change=False,
logger=logging.root.getChild('test_resources'),
)
with ssh_resource.get_tunnel(1234):
ssh_mock.assert_called_once_with(
'remote_host',
ssh_port=12345,
ssh_username='******',
ssh_pkey=key_from_str(ssh_key),
ssh_proxy=None,
local_bind_address=('localhost', ),
remote_bind_address=('localhost', 1234),
host_pkey_directories=[],
logger=ssh_resource.log,
)