scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-21 00:21:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 74

test(mcp): assert redirect rejection without SDK wording

Browse source
This commit is contained in:
Joshua Valluru 2026-09-19 16:01:42 -07:00
parent 064b7d10df
commit 7cd96e9c2d
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_litellm/experimental_mcp_client/test_mcp_client.py
View file

@ -1273,10 +1273,11 @@ async def test_sdk_cross_origin_redirect_never_contacts_destination(operation: s
if operation == "list"
else client.call_tool(CallToolRequestParams(name="add", arguments={"a": 2, "b": 3}), raise_on_error=True)
)
with anyio.fail_after(10), pytest.raises(MCPError, match=r"Redirect to .*destination.* not followed"):
with anyio.fail_after(10), pytest.raises(MCPError):
await pending_operation
assert responder.call_count == 1
request: Final = responder.call_args.args[0]
assert request.method == "POST"
assert request.url == "https://upstream.example.com/mcp"
assert request.headers["x-upstream-token"] == "Bearer synthetic-token"
assert all(call.args[0].url.host != "destination.example.com" for call in responder.call_args_list)