From b9b5d60a38d1a3e3a630ece9ee89a948d1f12311 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 13 Nov 2024 14:19:37 -0800 Subject: [PATCH] test_aws_secret_manager --- tests/local_testing/test_secret_manager.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_secret_manager.py b/tests/local_testing/test_secret_manager.py index 7066996a2..698812fbb 100644 --- a/tests/local_testing/test_secret_manager.py +++ b/tests/local_testing/test_secret_manager.py @@ -22,15 +22,19 @@ from litellm.secret_managers.aws_secret_manager_v2 import AWSSecretsManagerV2 from litellm.secret_managers.main import get_secret -@pytest.mark.skip(reason="AWS Suspended Account") def test_aws_secret_manager(): + import json + AWSSecretsManagerV2.load_aws_secret_manager(use_aws_secret_manager=True) secret_val = get_secret("litellm_master_key") print(f"secret_val: {secret_val}") - assert secret_val == "sk-1234" + # cast json to dict + secret_val = json.loads(secret_val) + + assert secret_val["litellm_master_key"] == "sk-1234" def redact_oidc_signature(secret_val):