mirror of
https://github.com/BerriAI/litellm.git
synced 2026-11-15 09:12:18 +00:00
test(database): expect no native pool in proxy lifespan cleanup (#51911)
Co-authored-by: mateo <277851410+mateo-berri@users.noreply.github.com> Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
61483d8f04
commit
aeb9ed07aa
1 changed files with 1 additions and 2 deletions
|
|
@ -10,7 +10,6 @@ from typing import Final, Literal, TypeAlias
|
|||
import psutil
|
||||
import pytest
|
||||
|
||||
from litellm.rust_bridge._native import NativeDatabase
|
||||
from tests.integration._support.client import eventually
|
||||
from tests.integration._support.database import read_rows, scratch_database
|
||||
|
||||
|
|
@ -80,7 +79,7 @@ async def _exercise_lifespans(mode: Mode) -> None:
|
|||
async with proxy_server.proxy_startup_event(proxy_server.app) as state:
|
||||
database: Final = state["database"]
|
||||
assert database is not None
|
||||
assert isinstance(database.native, NativeDatabase)
|
||||
assert database.native is None
|
||||
assert await database.prisma.writer_db.query_raw("SELECT 42 AS answer") == [{"answer": 42}]
|
||||
assert await database.prisma.read_db.query_raw("SELECT 43 AS answer") == [{"answer": 43}]
|
||||
assert psutil.Process().children(recursive=True)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue