scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

test(router): patch direct endpoints in failure parity fixture

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
mateo 2026-11-12 01:19:31 +00:00
parent 70ec2b6c3e
commit 025999dc48
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/unit/test_router/core/test_usage_and_failure_counting_parity.py
View file

@ -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