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

test(proxy): inject the fake Prisma client through the module boundary instead of patching _setup_prisma_client

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-24 06:07:49 +00:00
parent ffef19f2a0
commit e6a5d32cd9
1 changed files with 5 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/test_litellm/proxy/test_proxy_server.py
View file

@ -1733,7 +1733,7 @@ class _PrismaClientWhoseUserTableCannotHash:
class _Db:
litellm_usertable = None
def __init__(self):
def __init__(self, database_url, proxy_logging_obj):
self.db = self._Db()
self.db.litellm_usertable = self._Table()
self.writer_db = self.db
@ -1744,6 +1744,9 @@ class _PrismaClientWhoseUserTableCannotHash:
async def disconnect(self):
pass
def start_view_setup_task(self):
pass
async def check_view_exists(self):
pass
@ -1761,12 +1764,7 @@ async def test_proxy_startup_surfaces_a_password_migration_crypto_failure(monkey
_boot_with_general_settings(monkeypatch, tmp_path, {"master_key": "sk-a-safe-master-key"})
monkeypatch.setattr("litellm.proxy.proxy_server.prisma_client", None)
monkeypatch.setenv("DATABASE_URL", "postgresql://nobody:nothing@127.0.0.1:1/unreachable")
cannot_hash = _PrismaClientWhoseUserTableCannotHash()
async def connected(**kwargs):
return cannot_hash
monkeypatch.setattr("litellm.proxy.proxy_server.ProxyStartupEvent._setup_prisma_client", connected)
monkeypatch.setattr("litellm.proxy.proxy_server.PrismaClient", _PrismaClientWhoseUserTableCannotHash)
monkeypatch.setattr("litellm.proxy.proxy_server.openssl_enforces_fips", lambda: True)
monkeypatch.setenv("LITELLM_FIPS_MODE", fips_mode)