scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-01 02:02:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 91

test(mcp): verify conformance bridge fidelity and runner failure

Browse source
This commit is contained in:
Joshua Valluru 2026-09-26 11:35:48 -07:00
parent ddb048d51f
commit 3d14ca9063
1 changed files with 48 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
tests/integration/mcp/test_mcp_official_conformance.py
View file

@ -1,12 +1,15 @@
import json
import os
import uuid
from pathlib import Path
from typing import Final
import httpx
import pytest
from integration._support.client import Gateway
from integration._support.conformance import authenticated_endpoint, reference_server, run_scenario
from integration._support.mcp import official_client_outcomes, register_mcp
from integration._support.wire import Reply, wire_server
@pytest.mark.parametrize("name", ("server-initialize", "tools-list", "tools-call-image"))
@ -26,3 +29,48 @@ def test_official_scenario_through_gateway(gateway: Gateway, tmp_path: Path, unu
listed, called = official_client_outcomes(gateway, key, f"/{alias}/mcp", "test_simple_text", {})
assert f"{alias}-test_simple_text" in listed.tools, listed
assert called.ok and called.text == "This is a simple text response for testing.", called
@pytest.mark.parametrize("status", (200, 403))
def test_conformance_bridge_preserves_headers_payload_and_error_status(status: int) -> None:
body: Final = {"jsonrpc": "2.0", "id": 7, "method": "tools/call", "params": {"name": "test_image_content"}}
response_body: Final = b'data: {"jsonrpc":"2.0","id":7,"error":{"code":-32000,"message":"denied"}}\n\n'
with wire_server(
lambda request: Reply(
status=status, chunks=(response_body[:12], response_body[12:]), content_type="text/event-stream"
)
) as upstream:
with authenticated_endpoint(upstream.url + "/mcp", "test-key", "official") as endpoint:
response: Final = httpx.post(
endpoint,
json=body,
headers={
"Host": "127.0.0.1",
"Origin": "https://untrusted.example",
"Authorization": "Bearer replaced",
"MCP-Protocol-Version": "2025-03-26",
"Mcp-Session-Id": "same-session",
},
)
assert response.status_code == status and response.content == response_body
requests: Final = upstream.drain()
assert len(requests) == 1
received: Final = requests[0]
assert json.loads(received.body) == {**body, "params": {"name": "official-test_image_content"}}
assert {
name: received.headers[name]
for name in ("host", "origin", "authorization", "mcp-protocol-version", "mcp-session-id")
} == {
"host": "127.0.0.1",
"origin": "https://untrusted.example",
"authorization": "Bearer test-key",
"mcp-protocol-version": "2025-03-26",
"mcp-session-id": "same-session",
}
def test_official_runner_rejects_unknown_scenario(tmp_path: Path) -> None:
output: Final = tmp_path / "unknown-scenario"
with pytest.raises(AssertionError):
run_scenario(Path(os.environ["MCP_CONFORMANCE_ROOT"]), "http://127.0.0.1:1/mcp", "missing-scenario", output)
assert "missing-scenario" in (output / "runner.log").read_text()