From 883cbf259d5891abe54a41ed9b8d1e2455f0ea94 Mon Sep 17 00:00:00 2001
From: "devin-ai-integration[bot]"
 <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Thu, 29 Oct 2026 01:19:57 +0000
Subject: [PATCH] test(proxy): cover the dry run CLI and PrismaManager wrapper
 without a database

---
 tests/unit/proxy/db/test_prisma_client.py | 16 ++++++++++++++++
 tests/unit/proxy/test_proxy_cli.py        | 17 +++++++++++++++++
 2 files changed, 33 insertions(+)

diff --git a/tests/unit/proxy/db/test_prisma_client.py b/tests/unit/proxy/db/test_prisma_client.py
index 4b776bfbb1a..6c82628576f 100644
--- a/tests/unit/proxy/db/test_prisma_client.py
+++ b/tests/unit/proxy/db/test_prisma_client.py
@@ -586,3 +586,19 @@ async def test_a_query_issued_while_the_last_one_finishes_keeps_the_engine_drain
     await second
     await asyncio.wait_for(drained, timeout=1)
     assert engine.queries_finished == 2
+
+
+def test_migration_dry_run_summary_without_migrations_has_nothing_to_dry_run(unset_database_url):
+    """`prisma db push` applies no migration files, so the dry run reports that instead of connecting."""
+    from litellm.proxy.db.prisma_client import PrismaManager
+
+    assert PrismaManager.migration_dry_run_summary(use_migrate=False) == (
+        True,
+        "The schema is pushed rather than migrated, there is nothing to dry-run.",
+    )
+
+
+def test_migration_dry_run_summary_needs_database_url(unset_database_url):
+    from litellm.proxy.db.prisma_client import PrismaManager
+
+    assert PrismaManager.migration_dry_run_summary(use_migrate=True) == (False, "DATABASE_URL is not set.")
diff --git a/tests/unit/proxy/test_proxy_cli.py b/tests/unit/proxy/test_proxy_cli.py
index d77731aa83b..e240038670f 100644
--- a/tests/unit/proxy/test_proxy_cli.py
+++ b/tests/unit/proxy/test_proxy_cli.py
@@ -4073,3 +4073,20 @@ def test_granian_flag_notices_are_json_records_in_json_mode(
         "LiteLLM: --ciphers is not applied when using --run_granian.",
     ]
     assert {record["level"] for record in records} == {"INFO"}
+
+
+def test_migration_data_check_exits_zero_when_there_is_nothing_to_dry_run(monkeypatch, capfd):
+    monkeypatch.setenv("DATABASE_URL", "postgresql://user:pass@db.example/litellm")
+
+    assert ProxyInitializationHelpers.run_migration_data_check(use_migrate=False) == 0
+
+    out, err = capfd.readouterr()
+    assert out == "LiteLLM Proxy: The schema is pushed rather than migrated, there is nothing to dry-run.\n"
+    assert err == ""
+
+
+def test_migration_data_check_requires_database_url(monkeypatch):
+    monkeypatch.delenv("DATABASE_URL", raising=False)
+
+    with pytest.raises(click.UsageError, match="requires DATABASE_URL"):
+        ProxyInitializationHelpers.run_migration_data_check(use_migrate=True)