diff --git a/tests/mcp_tests/test_mcp_server.py b/tests/mcp_tests/test_mcp_server.py index a4a28215e16..f2917d21e17 100644 --- a/tests/mcp_tests/test_mcp_server.py +++ b/tests/mcp_tests/test_mcp_server.py @@ -1474,6 +1474,8 @@ async def test_add_update_server_with_alias(): mock_mcp_server.byok_api_key_help_url = None mock_mcp_server.created_at = None mock_mcp_server.updated_at = None + mock_mcp_server.token_validation = None + mock_mcp_server.token_storage_ttl_seconds = None # Add server to manager await test_manager.add_server(mock_mcp_server) @@ -1530,6 +1532,8 @@ async def test_add_update_server_without_alias(): mock_mcp_server.byok_api_key_help_url = None mock_mcp_server.created_at = None mock_mcp_server.updated_at = None + mock_mcp_server.token_validation = None + mock_mcp_server.token_storage_ttl_seconds = None # Add server to manager await test_manager.add_server(mock_mcp_server) @@ -1587,6 +1591,8 @@ async def test_add_update_server_fallback_to_server_id(): mock_mcp_server.byok_api_key_help_url = None mock_mcp_server.created_at = None mock_mcp_server.updated_at = None + mock_mcp_server.token_validation = None + mock_mcp_server.token_storage_ttl_seconds = None # Add server to manager await test_manager.add_server(mock_mcp_server)