From e54528096fadcefb0e625e46940b0577f8343f20 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Mon, 18 May 2026 20:36:25 +0530 Subject: [PATCH] fix code qa --- tests/test_litellm/test_router.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/tests/test_litellm/test_router.py b/tests/test_litellm/test_router.py index 1757fae69a0..d8be527689e 100644 --- a/tests/test_litellm/test_router.py +++ b/tests/test_litellm/test_router.py @@ -3842,3 +3842,28 @@ def test_get_deployment_credentials_with_provider_returns_none_for_blocked_deplo router = _router_with_two_deployments([True, False]) assert router.get_deployment_credentials_with_provider(model_id="dep-0") is None assert router.get_deployment_credentials_with_provider(model_id="dep-1") is not None + + +def test_is_deployment_blocked_static_helper_reflects_blocked_flag(): + """ + Exercises Router._is_deployment_blocked so router_code_coverage.py (AST call graph) + marks the helper as covered by router-named tests. + """ + import types + + import litellm + + router = _router_with_two_deployments([True, False]) + blocked_dep = router.get_deployment("dep-0") + unblocked_dep = router.get_deployment("dep-1") + assert blocked_dep is not None and unblocked_dep is not None + assert litellm.Router._is_deployment_blocked(blocked_dep) is True + assert litellm.Router._is_deployment_blocked(unblocked_dep) is False + + # No model_info on deployment object → treated as not blocked + assert litellm.Router._is_deployment_blocked(object()) is False + missing_blocked = types.SimpleNamespace() + assert litellm.Router._is_deployment_blocked(types.SimpleNamespace(model_info=missing_blocked)) is False + assert litellm.Router._is_deployment_blocked( + types.SimpleNamespace(model_info=types.SimpleNamespace(blocked=True)) + ) is True