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

fix: restore module import in test_check_migration after merge

Browse source 
The conflict resolution imported only the function but the test body
references check_migration as a module throughout.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
user 2026-04-08 18:03:23 +00:00
parent a733c74a68
commit cc28914d7f
No known key found for this signature in database
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_litellm/proxy/db/test_check_migration.py
View file

@ -14,11 +14,11 @@ def test_check_migration_out_of_sync(mocker):
- 🚨 [IMPORTANT] Does NOT Raise an Exception when the Prisma schema is out of sync with the database.
- logs an error when the Prisma schema is out of sync with the database.
"""
# Import the function first so check_migration module is in sys.modules,
# Import the module first so check_migration is in sys.modules,
# then patch the logger reference in that module directly (not the source
# module) so the patch works regardless of import order or xdist worker
# assignment.
from litellm.proxy.db.check_migration import check_prisma_schema_diff
from litellm.proxy.db import check_migration
mock_logger = mocker.patch("litellm.proxy.db.check_migration.verbose_logger")