From f16e304bc96f8d2ab6d6f79a8fda07f1e4527b5d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=86=AF=E5=9F=BA=E9=AD=81?= <1412414664@qq.com>
Date: Sat, 11 Jul 2026 03:25:25 +0800
Subject: [PATCH] test(proxy): refresh MCP header fixtures

---
 .../_experimental/mcp_server/test_mcp_sigv4_auth.py  | 12 ++++++++++++
 .../_experimental/mcp_server/test_rest_endpoints.py  |  4 ++--
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_sigv4_auth.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_sigv4_auth.py
index 0164d2d217a..7186bc5da7e 100644
--- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_sigv4_auth.py
+++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_sigv4_auth.py
@@ -995,6 +995,12 @@ class TestSigV4BuildFromTable:
         table_record.instructions = None
         table_record.source_url = None
         table_record.submitted_by = None
+        table_record.token_exchange_endpoint = None
+        table_record.audience = None
+        table_record.subject_token_type = None
+        table_record.token_exchange_profile = None
+        table_record.timeout = None
+        table_record.max_concurrent_requests = None
 
         manager = MCPServerManager()
 
@@ -1058,6 +1064,12 @@ class TestSigV4BuildFromTable:
         table_record.instructions = None
         table_record.source_url = None
         table_record.submitted_by = "user-submitter"
+        table_record.token_exchange_endpoint = None
+        table_record.audience = None
+        table_record.subject_token_type = None
+        table_record.token_exchange_profile = None
+        table_record.timeout = None
+        table_record.max_concurrent_requests = None
 
         manager = MCPServerManager()
 
diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_rest_endpoints.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_rest_endpoints.py
index e92647a2feb..024d7a4302e 100644
--- a/tests/test_litellm/proxy/_experimental/mcp_server/test_rest_endpoints.py
+++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_rest_endpoints.py
@@ -2570,7 +2570,7 @@ class TestPreviewOpenAPITools:
 
         registered_summary_to_name: dict = {}
 
-        def fake_create_tool_function(path, method, operation, base_url):  # noqa: ANN001
+        def fake_create_tool_function(path, method, operation, base_url):
             def _f():
                 return None
 
@@ -2583,7 +2583,7 @@ class TestPreviewOpenAPITools:
         )
 
         class _StubRegistry:
-            def register_tool(self, name, description, input_schema, handler):  # noqa: ANN001
+            def register_tool(self, name, description, input_schema, handler):
                 registered_summary_to_name[description] = name
 
         monkeypatch.setattr(