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