def test__aws_credentials_with_short_lived_credentials_and_ec2_metadata_service_having_credentials(mock):
credentials = Credentials(access_key=_random_string(), secret_key=_random_string(), token=_random_string())
session = Mock()
session.get_credentials.return_value = credentials
mock.return_value = True
aws_credentials = _aws_credentials(session)
assert aws_credentials is None
def test__aws_credentials_with_long_lived_credentials():
credentials = Credentials(access_key=_random_string(), secret_key=_random_string(), token=None)
session = Mock()
session.get_credentials.return_value = credentials
aws_credentials = _aws_credentials(session)
assert aws_credentials == [
"AWS_ACCESS_KEY_ID=%s" % credentials.access_key,
"AWS_SECRET_ACCESS_KEY=%s" % credentials.secret_key,
]
def test__aws_credentials_with_short_lived_credentials_and_ec2_metadata_service_having_no_credentials(mock):
credentials = Credentials(access_key=_random_string(), secret_key=_random_string(), token=_random_string())
session = Mock()
session.get_credentials.return_value = credentials
mock.return_value = False
aws_credentials = _aws_credentials(session)
assert aws_credentials == [
'AWS_ACCESS_KEY_ID=%s' % credentials.access_key,
'AWS_SECRET_ACCESS_KEY=%s' % credentials.secret_key,
'AWS_SESSION_TOKEN=%s' % credentials.token
]
