def test_get_target_region_config_metadata_unavailable(mocker, capsys):
mocker.patch('mount_efs.get_aws_ec2_metadata_token', return_value=None)
mocker.patch('mount_efs.urlopen', side_effect=URLError('test error'))
config = get_config('{fs_id}.efs.{region}.{dns_name_suffix}')
with pytest.raises(SystemExit) as ex:
mount_efs.get_target_region(config)
assert 0 != ex.value.code
out, err = capsys.readouterr()
assert 'Error retrieving region' in err
def get_target_region_helper():
config = get_config('{fs_id}.efs.{region}.{dns_name_suffix}')
return mount_efs.get_target_region(config)
def _test_get_target_region_from_dns_format(mocker, config):
mocker.patch('mount_efs.get_aws_ec2_metadata_token', return_value=None)
mocker.patch('mount_efs.urlopen', side_effect=URLError('test error'))
assert TARGET_REGION == mount_efs.get_target_region(config)
def test_get_target_region_from_config_variable(mocker):
config = get_config('{fs_id}.efs.us-east-2.{dns_name_suffix}',
TARGET_REGION)
assert TARGET_REGION == mount_efs.get_target_region(config)
def test_get_target_region_from_metadata(mocker):
mocker.patch('mount_efs.get_aws_ec2_metadata_token', return_value=None)
mocker.patch('mount_efs.urlopen', return_value=MockUrlLibResponse())
config = get_config('{fs_id}.efs.{region}.{dns_name_suffix}', None)
assert TARGET_REGION == mount_efs.get_target_region(config)
def get_target_region_helper():
config = get_config(DEFAULT_DNS_NAME_FORMAT)
return mount_efs.get_target_region(config)
