scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

test(e2e): assert the server record round-trips in the ingress rejection tests

Browse source
This commit is contained in:
Tin Chi Lo 2026-07-16 13:03:24 -07:00
parent 9b144f1311
commit 5880135418
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/e2e/mcp/test_mcp_tool_access_e2e.py
View file

@ -44,6 +44,11 @@ class TestMcpToolAccess:
created = client.create_server(McpServerCreateBody(alias=alias, url=MCP_STUB_URL, allow_all_keys=True))
resources.defer(lambda: client.delete_server(created.server_id))
stored = client.server_info(created.server_id)
assert stored.alias == alias
assert stored.url == MCP_STUB_URL
assert stored.allow_all_keys is True
key = client.gateway.generate_key(KeyGenerateBody())
resources.defer(lambda: client.gateway.delete_key(key))
_ = client.poll_tool_names(alias, {"x-litellm-api-key": f"Bearer {key}"})
@ -62,6 +67,11 @@ class TestMcpToolAccess:
created = client.create_server(McpServerCreateBody(alias=alias, url=MCP_STUB_URL, allow_all_keys=True))
resources.defer(lambda: client.delete_server(created.server_id))
stored = client.server_info(created.server_id)
assert stored.alias == alias
assert stored.url == MCP_STUB_URL
assert stored.allow_all_keys is True
key = client.gateway.generate_key(KeyGenerateBody())
resources.defer(lambda: client.gateway.delete_key(key))
_ = client.poll_tool_names(alias, {"Authorization": f"Bearer {key}"})