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])