From 2f5a553a7d8e8cad9035f3b3c8e22cd6b16d1732 Mon Sep 17 00:00:00 2001 From: Harshit28j Date: Tue, 10 Mar 2026 18:31:04 +0530 Subject: [PATCH] test: assert setup_database called with correct args Co-Authored-By: Claude Opus 4.6 --- tests/test_litellm/proxy/test_proxy_cli.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_litellm/proxy/test_proxy_cli.py b/tests/test_litellm/proxy/test_proxy_cli.py index bed77b43c1..642d21a42f 100644 --- a/tests/test_litellm/proxy/test_proxy_cli.py +++ b/tests/test_litellm/proxy/test_proxy_cli.py @@ -720,6 +720,7 @@ class TestHealthAppFactory: ["--local", "--skip_server_startup"], standalone_mode=False ) assert exc_info.value.code == 1 + mock_setup_database.assert_called_once_with(use_migrate=True) # --- Module-level helpers for worker startup hook tests ---