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

78
tests/local_testing/test_aws_secret_manager.py
View file

@ -50,26 +50,36 @@ 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"
# Write secret try:
write_response = await secret_manager.async_write_secret( # Write secret
secret_name=test_secret_name, write_response = await secret_manager.async_write_secret(
secret_value=test_secret_value, secret_name=test_secret_name,
description="LiteLLM Test Secret", secret_value=test_secret_value,
) description="LiteLLM Test Secret",
)
print("Write Response:", write_response) print("Write Response:", write_response)
assert write_response is not None assert write_response is not None
assert "ARN" in write_response assert "ARN" in write_response
assert "Name" in write_response assert "Name" in write_response
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,24 +96,34 @@ async def test_write_and_read_json_secret():
"metadata": {"team": "ml", "project": "litellm"}, "metadata": {"team": "ml", "project": "litellm"},
} }
# Write JSON secret try:
write_response = await secret_manager.async_write_secret( # Write JSON secret
secret_name=test_secret_name, write_response = await secret_manager.async_write_secret(
secret_value=json.dumps(test_secret_value), secret_name=test_secret_name,
description="LiteLLM JSON Test Secret", secret_value=json.dumps(test_secret_value),
) description="LiteLLM JSON Test 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(
parsed_value = json.loads(read_value) secret_name=test_secret_name
)
parsed_value = json.loads(read_value)
print("Read Value:", read_value) print("Read Value:", read_value)
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