From 545fe29c012f0a079d5dfb92bff2765ef4f01c59 Mon Sep 17 00:00:00 2001 From: Jackson Tian Date: Tue, 11 Jun 2024 18:31:12 +0800 Subject: [PATCH] add test cases for sts provider --- tests/test_providers.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tests/test_providers.py b/tests/test_providers.py index 944f8e6..3e648fc 100644 --- a/tests/test_providers.py +++ b/tests/test_providers.py @@ -356,6 +356,13 @@ def test_EnvironmentVariableCredentialsProvider(self): self.assertEqual('accessKeyIdTest', cred.access_key_id) self.assertEqual('accessKeySecretTest', cred.access_key_secret) + auth_util.environment_security_token = 'token' + cred = prov.get_credentials() + self.assertEqual('accessKeyIdTest', cred.access_key_id) + self.assertEqual('accessKeySecretTest', cred.access_key_secret) + self.assertEqual('token', cred.security_token) + self.assertEqual('sts', cred.credential_type) + auth_util.environment_access_key_id = None self.assertIsNone(prov.get_credentials()) @@ -368,3 +375,4 @@ def test_EnvironmentVariableCredentialsProvider(self): auth_util.environment_access_key_id = None auth_util.environment_access_key_secret = None + auth_util.environment_security_token = None \ No newline at end of file