phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

add test for async_delete_secret

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-13 11:54:00 -08:00
parent cc392e2792
commit 4dab664cdc
1 changed files with 49 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tests/local_testing/test_aws_secret_manager.py
View file

@ -50,6 +50,7 @@ async def test_write_and_read_simple_secret():
test_secret_name = f"litellm_test_{uuid.uuid4().hex[:8]}" test_secret_name = f"litellm_test_{uuid.uuid4().hex[:8]}"
test_secret_value = "test_value_123" test_secret_value = "test_value_123"
try:
# Write secret # Write secret
write_response = await secret_manager.async_write_secret( write_response = await secret_manager.async_write_secret(
secret_name=test_secret_name, secret_name=test_secret_name,
@ -65,11 +66,20 @@ async def test_write_and_read_simple_secret():
assert write_response["Name"] == test_secret_name assert write_response["Name"] == test_secret_name
# Read secret back # Read secret back
read_value = await secret_manager.async_read_secret(secret_name=test_secret_name) read_value = await secret_manager.async_read_secret(
secret_name=test_secret_name
)
print("Read Value:", read_value) print("Read Value:", read_value)
assert read_value == test_secret_value assert read_value == test_secret_value
finally:
# Cleanup: Delete the secret
delete_response = await secret_manager.async_delete_secret(
secret_name=test_secret_name
)
print("Delete Response:", delete_response)
assert delete_response is not None
@pytest.mark.asyncio @pytest.mark.asyncio
@ -86,6 +96,7 @@ async def test_write_and_read_json_secret():
"metadata": {"team": "ml", "project": "litellm"}, "metadata": {"team": "ml", "project": "litellm"},
} }
try:
# Write JSON secret # Write JSON secret
write_response = await secret_manager.async_write_secret( write_response = await secret_manager.async_write_secret(
secret_name=test_secret_name, secret_name=test_secret_name,
@ -96,7 +107,9 @@ async def test_write_and_read_json_secret():
print("Write Response:", write_response) print("Write Response:", write_response)
# Read and parse JSON secret # Read and parse JSON secret
read_value = await secret_manager.async_read_secret(secret_name=test_secret_name) read_value = await secret_manager.async_read_secret(
secret_name=test_secret_name
)
parsed_value = json.loads(read_value) parsed_value = json.loads(read_value)
print("Read Value:", read_value) print("Read Value:", read_value)
@ -104,6 +117,13 @@ async def test_write_and_read_json_secret():
assert parsed_value == test_secret_value assert parsed_value == test_secret_value
assert parsed_value["api_key"] == "test_key" assert parsed_value["api_key"] == "test_key"
assert parsed_value["metadata"]["team"] == "ml" assert parsed_value["metadata"]["team"] == "ml"
finally:
# Cleanup: Delete the secret
delete_response = await secret_manager.async_delete_secret(
secret_name=test_secret_name
)
print("Delete Response:", delete_response)
assert delete_response is not None
@pytest.mark.asyncio @pytest.mark.asyncio