From 025999dc485e425754c2be1834a577bd980ed101 Mon Sep 17 00:00:00 2001
From: mateo <277851410+mateo-berri@users.noreply.github.com>
Date: Thu, 12 Nov 2026 01:19:31 +0000
Subject: [PATCH] test(router): patch direct endpoints in failure parity
 fixture

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 .../core/test_usage_and_failure_counting_parity.py       | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/tests/unit/test_router/core/test_usage_and_failure_counting_parity.py b/tests/unit/test_router/core/test_usage_and_failure_counting_parity.py
index 3ef5b0c10c7..da500dca2f0 100644
--- a/tests/unit/test_router/core/test_usage_and_failure_counting_parity.py
+++ b/tests/unit/test_router/core/test_usage_and_failure_counting_parity.py
@@ -18,7 +18,7 @@ from litellm.litellm_core_utils.litellm_logging import Logging as LiteLLMLogging
 from litellm.router_core import RouterCore
 from litellm.router_utils import cooldown_handlers
 from litellm.router_utils.cooldown_handlers import mark_deployment_selection_failure
-from tests.unit.test_router.core.endpoints_helpers import EndpointHost
+from tests.unit.test_router.core.endpoints_helpers import EndpointHost, patch_direct_endpoints
 from tests.unit.test_router.core.retries_health_helpers import (
     MESSAGES,
     ErrorBuilder,
@@ -93,8 +93,10 @@ class SharedLoggingHost(EndpointHost):
 
 
 @pytest.fixture
-def host() -> SharedLoggingHost:
-    return SharedLoggingHost()
+def host(monkeypatch: pytest.MonkeyPatch) -> SharedLoggingHost:
+    host: Final = SharedLoggingHost()
+    patch_direct_endpoints(monkeypatch, host)
+    return host
 
 
 def minute() -> str:
@@ -353,6 +355,7 @@ async def test_a_failed_hops_404_is_the_callers_only_on_a_generic_call(
     else:
         _ = await attempt(core, "a", litellm_logging_obj=logging_obj)
 
+    assert host.deployments_called() == ["a", "b"]
     assert core.health().failures_this_minute("b") == counted