From 5ea4fe620fdefe92d7674c7d3a985919fa3dbc37 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Mon, 21 Sep 2026 11:55:58 -0700
Subject: [PATCH] test(router): give each prompt caching check test a fresh
 callback registry

---
 .../test_prompt_caching_deployment_check.py   | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/tests/unit/router_utils/pre_call_checks/test_prompt_caching_deployment_check.py b/tests/unit/router_utils/pre_call_checks/test_prompt_caching_deployment_check.py
index 3a3ed2c45f4..a87b24656f3 100644
--- a/tests/unit/router_utils/pre_call_checks/test_prompt_caching_deployment_check.py
+++ b/tests/unit/router_utils/pre_call_checks/test_prompt_caching_deployment_check.py
@@ -1,7 +1,7 @@
 import asyncio
 import copy
 import functools
-from typing import cast
+from typing import Final, cast
 
 import pytest
 
@@ -20,6 +20,23 @@ from litellm.utils import get_prompt_cache_min_tokens, is_prompt_caching_valid_p
 
 MODEL_GROUP_ALIAS = "my-claude-group"
 OPUS_4_6_MIN_TOKENS = 4096
+CALLBACK_REGISTRIES: Final = (
+    "input_callback",
+    "success_callback",
+    "failure_callback",
+    "_async_success_callback",
+    "_async_failure_callback",
+    "callbacks",
+)
+
+
+@pytest.fixture(autouse=True)
+def _fresh_callback_registries(monkeypatch):
+    """`litellm.logging_callback_manager` keeps one callback per class, so a
+    `PromptCachingDeploymentCheck` or `_SentMessagesCapture` left behind by an
+    earlier test would swallow the next test's success events."""
+    for registry in CALLBACK_REGISTRIES:
+        monkeypatch.setattr(litellm, registry, [])
 
 
 @pytest.fixture