scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-13 08:52:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 174

test(integration): point database tests at the moved litellm/proxy-extras package (#51453)

Browse source 
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:
devin-ai-integration[bot] 2026-11-12 20:27:00 +00:00 • committed by GitHub
parent 9e45c3d00e
commit 5ebc284053
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/integration/database/migrations/test_request_log_indexes.py
View file

@ -39,7 +39,7 @@ requires_db: Final = pytest.mark.skipif(
)
REPO: Final = Path(__file__).resolve().parents[4]
PACKAGE: Final = REPO / "litellm-proxy-extras" / "litellm_proxy_extras"
PACKAGE: Final = REPO / "litellm" / "proxy-extras" / "litellm_proxy_extras"
PARTITION_SCRIPT: Final = REPO / "litellm/scripts" / "db" / "partition_spend_logs.sql"
API_KEY_INDEX_MIGRATION: Final = "20260823000000_add_spend_logs_api_key_starttime_index"
CALL_ID_INDEX_MIGRATION: Final = "20260831120001_spend_logs_litellm_call_id_index"

6
tests/integration/database/test_managed_file_flat_ids_index.py
View file

@ -19,7 +19,7 @@ from psycopg import sql
from pydantic import JsonValue, TypeAdapter
REPO_ROOT: Final = Path(__file__).resolve().parents[3]
PRISMA_DIR: Final = REPO_ROOT / "litellm-proxy-extras" / "litellm_proxy_extras"
PRISMA_DIR: Final = REPO_ROOT / "litellm" / "proxy-extras" / "litellm_proxy_extras"
GIN_MIGRATION: Final = "20261003000000_add_managed_file_flat_ids_gin_index"
INDEX_NAME: Final = "LiteLLM_ManagedFileTable_flat_model_file_ids_idx"
SHIPPED_MIGRATIONS: Final = tuple(sorted(path.name for path in (PRISMA_DIR / "migrations").iterdir() if path.is_dir()))
@ -57,8 +57,8 @@ def _leg_python_path() -> str:
return os.pathsep.join(
(
str(REPO_ROOT),
str(REPO_ROOT / "litellm-proxy-extras"),
str(REPO_ROOT / "enterprise"),
str(REPO_ROOT / "litellm" / "proxy-extras"),
str(REPO_ROOT / "litellm" / "enterprise"),
os.environ.get("PYTHONPATH", ""),
)
)

2
tests/integration/database/test_proxy_boot_without_general_settings.py
View file

@ -12,7 +12,7 @@ from integration._support.process import DB_PUSH, LEGACY_MIGRATE_DEPLOY, MIGRATE
pytestmark: Final = pytest.mark.timeout(300)
REPO_ROOT: Final = Path(__file__).resolve().parents[3]
MIGRATIONS_DIR: Final = REPO_ROOT / "litellm-proxy-extras" / "litellm_proxy_extras" / "migrations"
MIGRATIONS_DIR: Final = REPO_ROOT / "litellm" / "proxy-extras" / "litellm_proxy_extras" / "migrations"
SHIPPED_MIGRATIONS: Final = frozenset(path.name for path in MIGRATIONS_DIR.iterdir() if path.is_dir())
CANNED_REPLY: Final = "Hello! This is a mock response from the fake OpenAI endpoint."
PROVIDER_KEY: Final = "integration-provider-key"

2
tests/integration/database/test_proxy_database_lifespan.py
View file

@ -121,7 +121,7 @@ def test_failed_startup_then_retry_releases_database_resources(tmp_path: Path, m
"LITELLM_LOCAL_MODEL_COST_MAP": "True",
"LITELLM_TELEMETRY_ENABLED": "false",
"PRISMA_HEALTH_WATCHDOG_ENABLED": "true",
"PYTHONPATH": os.pathsep.join((str(ROOT), str(ROOT / "enterprise"))),
"PYTHONPATH": os.pathsep.join((str(ROOT), str(ROOT / "litellm" / "enterprise"))),
**({"DATABASE_URL_READ_REPLICA": database_url} if replica else {}),
}
schema: Final = subprocess.run(

2
tests/integration/database/test_request_log_indexes_at_boot.py
View file

@ -20,7 +20,7 @@ from psycopg.rows import class_row
pytestmark: Final = pytest.mark.timeout(240)
REPO_ROOT: Final = Path(__file__).resolve().parents[3]
PRISMA_DIR: Final = REPO_ROOT / "litellm-proxy-extras" / "litellm_proxy_extras"
PRISMA_DIR: Final = REPO_ROOT / "litellm" / "proxy-extras" / "litellm_proxy_extras"
PARTITION_SCRIPT: Final = REPO_ROOT / "litellm/scripts" / "db" / "partition_spend_logs.sql"
SHIPPED_MIGRATIONS: Final = tuple(sorted(path.name for path in (PRISMA_DIR / "migrations").iterdir() if path.is_dir()))
API_KEY_INDEX_MIGRATION: Final = "20260823000000_add_spend_logs_api_key_starttime_index"

2
tests/integration/database/test_scheduled_job_runs.py
View file

@ -29,7 +29,7 @@ _MIGRATION: Final = next(
Path(__file__)
.resolve()
.parents[3]
.joinpath("litellm-proxy-extras/litellm_proxy_extras/migrations")
.joinpath("litellm/proxy-extras/litellm_proxy_extras/migrations")
.glob("*_scheduled_job_runs/migration.sql")
)
_PODS: Final = 4

2
tests/integration/database/test_v1_migration_error_logging.py
View file

@ -29,7 +29,7 @@ from tests.integration._support.ports import allocate_port
from tests.integration._support.process import stop_process_group
REPO_ROOT: Final = Path(__file__).resolve().parents[3]
MIGRATIONS_DIR: Final = REPO_ROOT / "litellm-proxy-extras" / "litellm_proxy_extras" / "migrations"
MIGRATIONS_DIR: Final = REPO_ROOT / "litellm" / "proxy-extras" / "litellm_proxy_extras" / "migrations"
MIGRATION_NAME: Final = "20260921190000_agent_identity"
pytestmark: Final = pytest.mark.timeout(300)
PASSWORD: Final = "wr ong'pw9"

2
tests/integration/security/test_hide_secrets_tool_traffic.py
View file

@ -161,7 +161,7 @@ def tool_rig(
"ANTHROPIC_API_KEY": "synthetic-provider-key",
"LITELLM_LOCAL_ANTHROPIC_BETA_HEADERS": "True",
"PYTHONPATH": os.pathsep.join(
(str(product_root / "enterprise"), str(product_root), os.environ.get("PYTHONPATH", ""))
(str(product_root / "litellm" / "enterprise"), str(product_root), os.environ.get("PYTHONPATH", ""))
),
**(environment or {}),
}