scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

refactor test file and enhance error handling

Browse source
This commit is contained in:
rzeta-10 2026-03-23 14:20:12 +05:30
parent 67fa3f1ee2
commit 0b2bdf0557
1 changed files with 15 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
tests/guardrails_tests/test_akto_guardrails.py
View file

@ -127,16 +127,19 @@ def test_init_from_env():
def test_init_defaults():
g = AktoGuardrail(
akto_base_url="http://localhost:9090",
akto_api_key="test-token",
guardrail_name="default-test",
event_hook="pre_call",
)
assert g.unreachable_fallback == "fail_closed"
assert g.guardrail_timeout == 5
assert g.akto_account_id == "1000000"
assert g.akto_vxlan_id == "0"
with patch.dict(os.environ, {}, clear=False):
os.environ.pop("AKTO_ACCOUNT_ID", None)
os.environ.pop("AKTO_VXLAN_ID", None)
g = AktoGuardrail(
akto_base_url="http://localhost:9090",
akto_api_key="test-token",
guardrail_name="default-test",
event_hook="pre_call",
)
assert g.unreachable_fallback == "fail_closed"
assert g.guardrail_timeout == 5
assert g.akto_account_id == "1000000"
assert g.akto_vxlan_id == "0"
# ── Payload ──
@ -357,6 +360,8 @@ async def test_fail_closed_on_unreachable():
with pytest.raises(HTTPException) as exc_info:
await g.apply_guardrail(inputs=inputs, request_data={}, input_type="request")
assert exc_info.value.status_code == 503
assert exc_info.value.detail == "Akto guardrail service unreachable"
assert "localhost" not in exc_info.value.detail