def test_to_dict(self):
# This would fail miserably if ``Credentials.request_id`` hadn't been
# explicitly set (no default).
# Default.
self.assertEqual(self.creds.to_dict(), {
'access_key': None,
'expiration': None,
'request_id': None,
'secret_key': None,
'session_token': None
})
# Override.
creds = Credentials()
creds.access_key = 'something'
creds.secret_key = 'crypto'
creds.session_token = 'this'
creds.expiration = 'way'
creds.request_id = 'comes'
self.assertEqual(creds.to_dict(), {
'access_key': 'something',
'expiration': 'way',
'request_id': 'comes',
'secret_key': 'crypto',
'session_token': 'this'
})
def test_to_dict(self):
# This would fail miserably if ``Credentials.request_id`` hadn't been
# explicitly set (no default).
# Default.
self.assertEqual(self.creds.to_dict(), {
'access_key': None,
'expiration': None,
'request_id': None,
'secret_key': None,
'session_token': None
})
# Override.
creds = Credentials()
creds.access_key = 'something'
creds.secret_key = 'crypto'
creds.session_token = 'this'
creds.expiration = 'way'
creds.request_id = 'comes'
self.assertEqual(creds.to_dict(), {
'access_key': 'something',
'expiration': 'way',
'request_id': 'comes',
'secret_key': 'crypto',
'session_token': 'this'
})
def test_should_give_credentials(self, sts_mock):
self.credentials_provider_mock.get_credentials_for_all_roles.return_value = DUMMY_CREDENTIALS
given_credentials = Credentials()
given_credentials.access_key = ANOTHER_KEY
given_credentials.secret_key = ANOTHER_SECRET
given_credentials.session_token = ANOTHER_TOKEN
given_credentials.expiration = ANOTHER_EXPIRATION
given_credentials_string = json.dumps({
"AccessKeyId": ANOTHER_KEY,
"SecretAccessKey": ANOTHER_SECRET,
"Token": ANOTHER_TOKEN,
"Expiration": ANOTHER_EXPIRATION
})
sts_mock.return_value.assume_role.return_value = AssumedRole(credentials=given_credentials)
credentials = self.provider.get_credentials_for_all_roles()
sts_mock.assert_called_with("fo-bar-1",
aws_access_key_id="ACCESS_KEY",
aws_secret_access_key="SECRET",
security_token="MY_TOKEN",
proxy=None,
proxy_port=None)
self.assertEqual(given_credentials_string, credentials[ROLE])
def test_should_give_credentials(self, sts_mock):
self.credentials_provider_mock.get_credentials_for_all_roles.return_value = DUMMY_CREDENTIALS
given_credentials = Credentials()
given_credentials.access_key = ANOTHER_KEY
given_credentials.secret_key = ANOTHER_SECRET
given_credentials.session_token = ANOTHER_TOKEN
given_credentials.expiration = ANOTHER_EXPIRATION
given_credentials_string = json.dumps({
"AccessKeyId": ANOTHER_KEY,
"SecretAccessKey": ANOTHER_SECRET,
"Token": ANOTHER_TOKEN,
"Expiration": ANOTHER_EXPIRATION
})
sts_mock.return_value.assume_role.return_value = AssumedRole(credentials=given_credentials)
credentials = self.provider.get_credentials_for_all_roles()
self.assertEqual(given_credentials_string, credentials[ROLE])
