From 191313e756fe606db5dceac80bed72618ff6679c Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Mon, 31 Aug 2026 22:22:05 -0700
Subject: [PATCH] test(websearch): register configured search tool in
 pre-request hook test

PR #38113 made a configured search_tool_name fail fast when the router
does not carry a matching search tool, which broke
test_pre_request_hook_modifies_request_body: it names test-search-tool
but never registers it. Stub the proxy router with that tool so the test
exercises the conversion path again.
---
 .../test_websearch_interception_e2e.py                 | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/tests/pass_through_unit_tests/test_websearch_interception_e2e.py b/tests/pass_through_unit_tests/test_websearch_interception_e2e.py
index 091ea106b91..cc7901b1710 100644
--- a/tests/pass_through_unit_tests/test_websearch_interception_e2e.py
+++ b/tests/pass_through_unit_tests/test_websearch_interception_e2e.py
@@ -937,6 +937,14 @@ async def test_pre_request_hook_modifies_request_body():
 
     print("✅ WebSearchInterceptionLogger initialized")
 
+    mock_router = MagicMock()
+    mock_router.search_tools = [
+        {
+            "search_tool_name": "test-search-tool",
+            "litellm_params": {"search_provider": "tavily"},
+        }
+    ]
+
     # Track what actually gets sent to the API
     captured_request = {}
 
@@ -987,7 +995,7 @@ async def test_pre_request_hook_modifies_request_body():
     with patch(
         "litellm.llms.anthropic.experimental_pass_through.messages.handler.anthropic_messages_handler",
         side_effect=mock_anthropic_messages_handler,
-    ):
+    ), patch("litellm.proxy.proxy_server.llm_router", mock_router):
 
         print(
             "\n📝 Making request with native web_search_20250305 tool (stream=True)..."