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

test(e2e): call the prefixed tool, require two gateways, cite the Linear tool name

Browse source 
Co-Authored-By: bot_apk <apk@cognition.ai>
This commit is contained in:
Devin AI 2026-09-19 02:03:12 +00:00
parent 890e5feabe
commit fc4a11ac53
2 changed files with 16 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/e2e/e2e_config.py
View file

@ -52,7 +52,7 @@ CHEAP_OPENAI_MODEL = os.environ.get("E2E_CHEAP_OPENAI_MODEL", "gpt-5.5")
LINEAR_MCP_URL = os.environ.get("E2E_LINEAR_MCP_URL", "https://mcp.linear.app/mcp")
LINEAR_STORAGE_STATE = os.environ.get("E2E_LINEAR_STORAGE_STATE", "")
LINEAR_READONLY_TOOL: Final = "list_teams"
LINEAR_READONLY_TOOL: Final = "list_teams" # Linear MCP tool name as listed by tools/list on mcp.linear.app when PR #33787 landed
# Jaeger query API of the compose stack's OTEL trace destination (the `jaeger`
# service in docker-compose.yml maps it to host 16686). Trace-completeness tests

25
tests/e2e/mcp/test_mcp_oauth_happy_path_e2e.py
View file

@ -2,10 +2,10 @@
The test creates a JWT-authorized user, completes real Linear authorization
consent, lists and calls a tool immediately through the per-server MCP route,
and verifies the canonical per-user credential row. It then uses a fresh SDK
client against one gateway URL or a configured replica URL. With one gateway
URL, that second run proves fresh-client reuse only. With replica URLs, it
proves that a process which did not run consent resolves the stored token.
and verifies the canonical per-user credential row. The first run targets the
first configured gateway replica, and a fresh SDK client then targets a
different replica to prove that a process which did not run consent resolves
the stored token.
"""
from __future__ import annotations
@ -18,7 +18,6 @@ from e2e_config import (
LINEAR_MCP_URL,
LINEAR_READONLY_TOOL,
LINEAR_STORAGE_STATE,
PROXY_BASE_URL,
PROXY_REPLICA_URLS,
unique_marker,
)
@ -61,6 +60,11 @@ class TestMcpOauthHappyPath:
)
alias: Final = f"e2elinear{unique_marker()}"
tool: Final = f"{alias}-{LINEAR_READONLY_TOOL}"
assert len(PROXY_REPLICA_URLS) >= 2, (
"set LITELLM_PROXY_REPLICA_URLS to at least two gateway URLs; the persistence cell needs a process "
"that did not run the consent"
)
created: Final = chat_client.create_server(
McpServerCreateBody(
alias=alias,
@ -88,10 +92,11 @@ class TestMcpOauthHappyPath:
headers,
storage,
LINEAR_STORAGE_STATE,
LINEAR_READONLY_TOOL,
tool,
{},
base_url=PROXY_REPLICA_URLS[0],
)
assert f"{alias}-{LINEAR_READONLY_TOOL}" in first_run.tools
assert tool in first_run.tools
assert first_run.is_error is False
assert first_run.text.strip() != ""
@ -106,16 +111,16 @@ class TestMcpOauthHappyPath:
)
)
replica: Final = PROXY_REPLICA_URLS[-1] if len(PROXY_REPLICA_URLS) > 1 else PROXY_BASE_URL
replica: Final = PROXY_REPLICA_URLS[-1]
second_run: Final = chat_client.list_and_call(
alias,
{"x-litellm-api-key": f"Bearer {idp.access_token(jwt_identity)}"},
InMemoryTokenStorage(),
None,
LINEAR_READONLY_TOOL,
tool,
{},
base_url=replica,
)
assert f"{alias}-{LINEAR_READONLY_TOOL}" in second_run.tools
assert tool in second_run.tools
assert second_run.is_error is False
assert second_run.text.strip() != ""