def test_create_ssh_command(
mocker, ec2, inst_name, use_inst_id, username, keyfile, port, ssh_options, use_private_ip,
use_gateway, gateway_username, profile_name, expected):
"""create_ssh_command test"""
from jungle.ec2 import create_ssh_command
mocker.patch('boto3.Session', new=lambda profile_name: boto3)
mocker.patch('click.prompt', new=lambda msg, type, default: 0)
ssh_server_instance_id = ec2[
'ssh_target_server'].id if use_inst_id else None
gateway_server_instance_id = ec2[
'gateway_target_server'].id if use_gateway else None
ssh_command = create_ssh_command(
create_session(profile_name),
ssh_server_instance_id, inst_name, username, keyfile, port, ssh_options, use_private_ip,
gateway_server_instance_id, gateway_username)
if use_gateway:
expected_output = expected.format(
ec2['gateway_target_server'].public_ip_address,
ec2['ssh_target_server'].private_ip_address)
elif use_private_ip:
expected_output = expected.format(
ec2['ssh_target_server'].private_ip_address)
else:
expected_output = expected.format(
ec2['ssh_target_server'].public_ip_address)
assert ssh_command == expected_output
def test_create_ssh_command(mocker, ec2, inst_name, use_inst_id, username,
keyfile, port, ssh_options, use_private_ip,
use_gateway, gateway_username, profile_name,
expected):
"""create_ssh_command test"""
from jungle.ec2 import create_ssh_command
mocker.patch('boto3.Session', new=lambda profile_name: boto3)
mocker.patch('click.prompt', new=lambda msg, type, default: 0)
ssh_server_instance_id = ec2[
'ssh_target_server'].id if use_inst_id else None
gateway_server_instance_id = ec2[
'gateway_target_server'].id if use_gateway else None
ssh_command = create_ssh_command(create_session(profile_name),
ssh_server_instance_id, inst_name,
username, keyfile, port, ssh_options,
use_private_ip,
gateway_server_instance_id,
gateway_username)
if use_gateway:
expected_output = expected.format(
ec2['gateway_target_server'].public_ip_address,
ec2['ssh_target_server'].private_ip_address)
elif use_private_ip:
expected_output = expected.format(
ec2['ssh_target_server'].private_ip_address)
else:
expected_output = expected.format(
ec2['ssh_target_server'].public_ip_address)
assert ssh_command == expected_output
def test_create_ssh_command(mocker, ec2, inst_name, use_inst_id, username,
keyfile, keydir, port, use_gateway, expected):
"""create_ssh_command test"""
from jungle.ec2 import create_ssh_command
mocker.patch('click.prompt', new=lambda msg, type, default: 0)
ssh_server_instance_id = ec2['ssh_target_server'].id if use_inst_id else None
gateway_server_instance_id = ec2['gateway_target_server'].id if use_gateway else None
ssh_command = create_ssh_command(
ssh_server_instance_id, inst_name, username, keyfile, keydir, port, gateway_server_instance_id)
if use_gateway:
expected_output = expected.format(
ec2['gateway_target_server'].public_ip_address,
ec2['ssh_target_server'].private_ip_address)
else:
expected_output = expected.format(ec2['ssh_target_server'].public_ip_address)
assert ssh_command == expected_output
