def test_put(self): in_five_min_epoch = int(time.time()) + 5 * 60 aws_credentials = AWSCredentials( access_key_id="test_aki", secret_access_key="test_sak", session_token="test_st", expiration=in_five_min_epoch, ) cache = AWSCredentialsCache() cache.put( credentials=aws_credentials, account_id="123456789012", role_name="test_rn", role_session_name="test_rsn", ) keys_values = cache.cache.items() self.assertEqual(len(keys_values), 1) for key, value in keys_values: self.assertEqual( key, AWSCredentialsCache.build_cache_key( account_id="123456789012", role_name="test_rn", role_session_name="test_rsn"), ) self.assertEqual( value, AWSCredentials( access_key_id="test_aki", secret_access_key="test_sak", session_token="test_st", expiration=in_five_min_epoch, ), )
def test(self): key = AWSCredentialsCache.build_cache_key( account_id="1234", role_name="test_role", role_session_name="test_role_session") expected_str = "1234:test_role:test_role_session" self.assertEqual(str(key), expected_str)