fix test

apache · Jul 18, 2024 · 328884b · 328884b
1 parent 8410c81
commit 328884b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/clients/client-python/tests/unittests/auth/test_simple_auth_provider.py b/clients/client-python/tests/unittests/auth/test_simple_auth_provider.py
@@ -40,6 +40,9 @@ def test_auth_provider(self):
         ).decode("utf-8")
         self.assertEqual(f"{user}:dummy", token_string)
 
+        original_gravitino_user = (
+            os.environ["GRAVITINO_USER"] if "GRAVITINO_USER" in os.environ else ""
+        )
         os.environ["GRAVITINO_USER"] = "test_auth2"
         provider: AuthDataProvider = SimpleAuthProvider()
         self.assertTrue(provider.has_token_data())
@@ -50,4 +53,4 @@ def test_auth_provider(self):
             token[len(AuthConstants.AUTHORIZATION_BASIC_HEADER) :]
         ).decode("utf-8")
         self.assertEqual(f"{user}:dummy", token_string)
-        os.environ["GRAVITINO_USER"] = ""
+        os.environ["GRAVITINO_USER"] = original_gravitino_user