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)
