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(organization): assert rejected values write nothing to the DB

Browse source
This commit is contained in:
ryan-crabbe-berri 2026-09-04 13:11:31 -07:00
parent 976ff0a785
commit 7a717740dd
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/test_litellm/proxy/management_endpoints/test_organization_endpoints.py
View file

@ -821,7 +821,8 @@ async def test_v2_rejects_negative_integer_limits(monkeypatch: pytest.MonkeyPatc
from litellm.proxy._types import LitellmUserRoles, OrganizationUpdateRequestV2, UserAPIKeyAuth
from litellm.proxy.management_endpoints.organization_endpoints import update_organization_v2
monkeypatch.setattr("litellm.proxy.proxy_server.prisma_client", AsyncMock())
prisma_mock = AsyncMock()
monkeypatch.setattr("litellm.proxy.proxy_server.prisma_client", prisma_mock)
auth = UserAPIKeyAuth(user_role=LitellmUserRoles.PROXY_ADMIN, user_id="admin-1")
with pytest.raises(HTTPException) as exc:
@ -832,6 +833,9 @@ async def test_v2_rejects_negative_integer_limits(monkeypatch: pytest.MonkeyPatc
)
assert exc.value.status_code == 422
assert field in str(exc.value.detail)
prisma_mock.db.tx.assert_not_called()
prisma_mock.db.litellm_budgettable.update.assert_not_awaited()
prisma_mock.db.litellm_organizationtable.update.assert_not_awaited()
@pytest.mark.asyncio
@ -841,7 +845,8 @@ async def test_v2_rejects_unparseable_budget_duration(monkeypatch: pytest.Monkey
from litellm.proxy._types import LitellmUserRoles, OrganizationUpdateRequestV2, UserAPIKeyAuth
from litellm.proxy.management_endpoints.organization_endpoints import update_organization_v2
monkeypatch.setattr("litellm.proxy.proxy_server.prisma_client", AsyncMock())
prisma_mock = AsyncMock()
monkeypatch.setattr("litellm.proxy.proxy_server.prisma_client", prisma_mock)
auth = UserAPIKeyAuth(user_role=LitellmUserRoles.PROXY_ADMIN, user_id="admin-1")
with pytest.raises(HTTPException) as exc:
@ -852,6 +857,9 @@ async def test_v2_rejects_unparseable_budget_duration(monkeypatch: pytest.Monkey
)
assert exc.value.status_code == 422
assert "budget_duration" in str(exc.value.detail)
prisma_mock.db.tx.assert_not_called()
prisma_mock.db.litellm_budgettable.update.assert_not_awaited()
prisma_mock.db.litellm_organizationtable.update.assert_not_awaited()
@pytest.mark.asyncio