test: assert setup_database called with correct args

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-10 18:31:04 +05:30 · 2026-03-10 18:31:04 +05:30 · 2f5a553a7d
commit 2f5a553a7d
parent aae2deb839
1 changed files with 1 additions and 0 deletions
--- 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 ---