scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 114

test noma, presidio

Browse source
This commit is contained in:
Ishaan Jaffer 2025-11-06 17:06:03 -08:00
parent 0a4e2a88e3
commit c727a2ad63
2 changed files with 20 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
tests/test_litellm/proxy/guardrails/guardrail_hooks/test_noma.py
View file

@ -267,9 +267,18 @@ class TestNomaGuardrailHooks:
# Verify API call details
call_args = mock_post.call_args
assert call_args[0][0].endswith("/ai-dr/v2/prompt/scan")
assert call_args[1]["headers"]["X-Noma-AIDR-Application-ID"] == "test-app"
assert call_args[1]["headers"]["Authorization"] == "Bearer test-api-key"
# Verify the URL endpoint
assert call_args.args[0].endswith("/ai-dr/v2/prompt/scan")
# Verify headers and JSON payload
if "headers" in call_args.kwargs:
headers = call_args.kwargs["headers"]
assert "Authorization" in headers
assert headers["Authorization"] == "Bearer test-api-key"
# Verify application ID is in JSON payload (not headers)
if "json" in call_args.kwargs:
json_payload = call_args.kwargs["json"]
assert "x-noma-context" in json_payload
assert json_payload["x-noma-context"]["applicationId"] == "test-app"
@pytest.mark.asyncio
async def test_pre_call_hook_blocked(
@ -1128,7 +1137,9 @@ class TestNomaImageProcessing:
mock_response.json.return_value = noma_response
mock_response.raise_for_status = MagicMock()
from litellm.proxy.guardrails.guardrail_hooks.noma.noma import NomaBlockedMessage
from litellm.proxy.guardrails.guardrail_hooks.noma.noma import (
NomaBlockedMessage,
)
with patch.object(
noma_guardrail.async_handler, "post", return_value=mock_response

6
tests/test_litellm/proxy/guardrails/guardrail_hooks/test_presidio.py
View file

@ -577,7 +577,11 @@ async def test_presidio_sets_guardrail_information_in_request_data():
assert "metadata" in request_data
assert "standard_logging_guardrail_information" in request_data["metadata"]
guardrail_info = request_data["metadata"]["standard_logging_guardrail_information"]
guardrail_info_list = request_data["metadata"]["standard_logging_guardrail_information"]
assert isinstance(guardrail_info_list, list)
assert len(guardrail_info_list) > 0
guardrail_info = guardrail_info_list[0]
assert "masked_entity_count" in guardrail_info
assert guardrail_info["masked_entity_count"]["EMAIL_ADDRESS"] == 1
assert guardrail_info["masked_entity_count"]["PERSON"] == 1