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

test(tags): suppress TQ008 on the new tests' patch calls

Browse source 
These patch litellm.proxy.proxy_server internals the endpoint imports
directly, the same unavoidable pattern every other test in this file
already uses; the new occurrences pushed the file's TQ008 count over
the CI budget.
This commit is contained in:
Arnold Galovics 2026-09-18 09:37:54 +02:00
parent 38eace0342
commit 19a854aeaa
1 changed files with 13 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
tests/unit/proxy/management_endpoints/test_tag_management_endpoints.py
View file

@ -618,11 +618,9 @@ async def test_update_tag_resets_spend():
try:
with (
patch("litellm.proxy.proxy_server.prisma_client") as mock_prisma,
patch(
"litellm.proxy.proxy_server.litellm_proxy_admin_name", "default_user_id"
),
patch("litellm.proxy.proxy_server.spend_counter_cache") as mock_spend_counter_cache,
patch("litellm.proxy.proxy_server.prisma_client") as mock_prisma, # test-quality-ok: the endpoint imports proxy_server.prisma_client itself; no parameter to inject
patch("litellm.proxy.proxy_server.litellm_proxy_admin_name", "default_user_id"), # test-quality-ok: the endpoint imports proxy_server.litellm_proxy_admin_name itself; no parameter to inject
patch("litellm.proxy.proxy_server.spend_counter_cache") as mock_spend_counter_cache, # test-quality-ok: the endpoint imports proxy_server.spend_counter_cache itself; no parameter to inject
):
mock_spend_counter_cache.redis_cache = None
@ -683,11 +681,9 @@ async def test_update_tag_resets_spend_in_redis():
try:
with (
patch("litellm.proxy.proxy_server.prisma_client") as mock_prisma,
patch(
"litellm.proxy.proxy_server.litellm_proxy_admin_name", "default_user_id"
),
patch("litellm.proxy.proxy_server.spend_counter_cache") as mock_spend_counter_cache,
patch("litellm.proxy.proxy_server.prisma_client") as mock_prisma, # test-quality-ok: the endpoint imports proxy_server.prisma_client itself; no parameter to inject
patch("litellm.proxy.proxy_server.litellm_proxy_admin_name", "default_user_id"), # test-quality-ok: the endpoint imports proxy_server.litellm_proxy_admin_name itself; no parameter to inject
patch("litellm.proxy.proxy_server.spend_counter_cache") as mock_spend_counter_cache, # test-quality-ok: the endpoint imports proxy_server.spend_counter_cache itself; no parameter to inject
):
mock_spend_counter_cache.redis_cache = MagicMock()
mock_spend_counter_cache.redis_cache.async_set_cache = AsyncMock()
@ -746,11 +742,9 @@ async def test_update_tag_resets_spend_redis_failure_does_not_fail_request():
try:
with (
patch("litellm.proxy.proxy_server.prisma_client") as mock_prisma,
patch(
"litellm.proxy.proxy_server.litellm_proxy_admin_name", "default_user_id"
),
patch("litellm.proxy.proxy_server.spend_counter_cache") as mock_spend_counter_cache,
patch("litellm.proxy.proxy_server.prisma_client") as mock_prisma, # test-quality-ok: the endpoint imports proxy_server.prisma_client itself; no parameter to inject
patch("litellm.proxy.proxy_server.litellm_proxy_admin_name", "default_user_id"), # test-quality-ok: the endpoint imports proxy_server.litellm_proxy_admin_name itself; no parameter to inject
patch("litellm.proxy.proxy_server.spend_counter_cache") as mock_spend_counter_cache, # test-quality-ok: the endpoint imports proxy_server.spend_counter_cache itself; no parameter to inject
):
mock_spend_counter_cache.redis_cache = MagicMock()
mock_spend_counter_cache.redis_cache.async_set_cache = AsyncMock(
@ -807,11 +801,9 @@ async def test_update_tag_without_spend_does_not_touch_counter_cache():
try:
with (
patch("litellm.proxy.proxy_server.prisma_client") as mock_prisma,
patch(
"litellm.proxy.proxy_server.litellm_proxy_admin_name", "default_user_id"
),
patch("litellm.proxy.proxy_server.spend_counter_cache") as mock_spend_counter_cache,
patch("litellm.proxy.proxy_server.prisma_client") as mock_prisma, # test-quality-ok: the endpoint imports proxy_server.prisma_client itself; no parameter to inject
patch("litellm.proxy.proxy_server.litellm_proxy_admin_name", "default_user_id"), # test-quality-ok: the endpoint imports proxy_server.litellm_proxy_admin_name itself; no parameter to inject
patch("litellm.proxy.proxy_server.spend_counter_cache") as mock_spend_counter_cache, # test-quality-ok: the endpoint imports proxy_server.spend_counter_cache itself; no parameter to inject
):
mock_db = Mock()
mock_prisma.db = mock_db
@ -866,7 +858,7 @@ async def test_update_tag_rejects_non_finite_spend():
)
try:
with patch("litellm.proxy.proxy_server.prisma_client") as mock_prisma:
with patch("litellm.proxy.proxy_server.prisma_client") as mock_prisma: # test-quality-ok: the endpoint imports proxy_server.prisma_client itself; no parameter to inject
mock_db = Mock()
mock_prisma.db = mock_db
mock_db.litellm_tagtable.update = AsyncMock()