From 4d0581e1d4642c5ab00009335599a2aeda2777ae Mon Sep 17 00:00:00 2001 From: Waqar Ahmed Khan Date: Tue, 23 Aug 2022 12:16:01 -0500 Subject: [PATCH] Fix test_default_provider when AWS_SESSION_TOKEN is set (#385) --- test/test_auth.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/test/test_auth.py b/test/test_auth.py index f99c44020..44e9b8aa2 100644 --- a/test/test_auth.py +++ b/test/test_auth.py @@ -27,11 +27,14 @@ def __init__(self, key, value): self.prev_value = os.environ.get(key) def __enter__(self): - os.environ[self.key] = self.value + if self.value is None: + os.environ.pop(self.key, None) + else: + os.environ[self.key] = self.value def __exit__(self, type, value, tb): if self.prev_value is None: - del os.environ[self.key] + os.environ.pop(self.key, None) else: os.environ[self.key] = self.prev_value @@ -98,7 +101,8 @@ def test_static_provider_no_session_token(self): def test_default_provider(self): # Default credentials provider should pick up environment variables. with ScopedEnvironmentVariable('AWS_ACCESS_KEY_ID', EXAMPLE_ACCESS_KEY_ID), \ - ScopedEnvironmentVariable('AWS_SECRET_ACCESS_KEY', EXAMPLE_SECRET_ACCESS_KEY): + ScopedEnvironmentVariable('AWS_SECRET_ACCESS_KEY', EXAMPLE_SECRET_ACCESS_KEY),\ + ScopedEnvironmentVariable('AWS_SESSION_TOKEN', None): event_loop_group = awscrt.io.EventLoopGroup() host_resolver = awscrt.io.DefaultHostResolver(event_loop_group)