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

Merge a13528e5e2 into eae8ed7f3c

Browse source
This commit is contained in:
willbird1936-dev 2026-09-29 00:31:22 +01:00 • committed by GitHub
commit c227d6eb11
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 27 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
tests/test_litellm/proxy/guardrails/test_guardrail_endpoints.py
View file

@ -997,6 +997,33 @@ async def test_create_guardrail_endpoint(
assert "Failed to initialize guardrail" in str(mock_logger.warning.call_args)
@pytest.mark.asyncio
async def test_create_guardrail_config_error_returns_400_not_500(
mocker, mock_guardrail_registry, mock_in_memory_handler
):
mock_prisma_client = mocker.Mock()
mock_in_memory_handler.initialize_guardrail.side_effect = ValueError("invalid config")
mocker.patch("litellm.proxy.proxy_server.prisma_client", mock_prisma_client)
mocker.patch(
"litellm.proxy.guardrails.guardrail_endpoints.GUARDRAIL_REGISTRY",
mock_guardrail_registry,
)
mocker.patch(
"litellm.proxy.guardrails.guardrail_registry.IN_MEMORY_GUARDRAIL_HANDLER",
mock_in_memory_handler,
)
with pytest.raises(HTTPException) as exc_info:
await create_guardrail(MOCK_CREATE_REQUEST, user_api_key_dict=MOCK_ADMIN_USER)
assert exc_info.value.status_code == 400
assert "invalid config" in str(exc_info.value.detail)
mock_guardrail_registry.add_guardrail_to_db.assert_called_once_with(
guardrail=MOCK_CREATE_REQUEST.guardrail, prisma_client=mocker.ANY
)
@pytest.mark.parametrize(
"scenario,expected_result,expected_exception",
[