From 045be5e6a91925046d4d21d37acf1a8b08330daa Mon Sep 17 00:00:00 2001 From: Santhosh Date: Tue, 22 Apr 2025 10:31:15 +0530 Subject: [PATCH 1/2] test: add pytest coverage for SSE transport functions --- tests/shared/test_sse.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/tests/shared/test_sse.py b/tests/shared/test_sse.py index f5158c3c3..b49c60b91 100644 --- a/tests/shared/test_sse.py +++ b/tests/shared/test_sse.py @@ -13,7 +13,7 @@ from starlette.routing import Mount, Route from mcp.client.session import ClientSession -from mcp.client.sse import sse_client +from mcp.client.sse import sse_client, remove_request_params from mcp.server import Server from mcp.server.sse import SseServerTransport from mcp.shared.exceptions import McpError @@ -250,3 +250,14 @@ async def test_sse_client_timeout( return pytest.fail("the client should have timed out and returned an error already") + + +def test_remove_request_params(): + # Removes query parameters + assert remove_request_params('http://example.com/test?foo=bar') == 'http://example.com/test' + # Removes fragment + assert remove_request_params('http://example.com/test#section') == 'http://example.com/test' + # Leaves clean URL unchanged + assert remove_request_params('http://example.com/test') == 'http://example.com/test' + # Works with path only + assert remove_request_params('/test/path?x=1') == '/test/path' From c5c7a6afb7536ab2a8c80cbf5937b8dc9819e5b2 Mon Sep 17 00:00:00 2001 From: Santhosh Date: Tue, 22 Apr 2025 10:42:40 +0530 Subject: [PATCH 2/2] style: auto-fix Ruff formatting issues --- tests/shared/test_sse.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/shared/test_sse.py b/tests/shared/test_sse.py index b49c60b91..b9299b86b 100644 --- a/tests/shared/test_sse.py +++ b/tests/shared/test_sse.py @@ -13,7 +13,7 @@ from starlette.routing import Mount, Route from mcp.client.session import ClientSession -from mcp.client.sse import sse_client, remove_request_params +from mcp.client.sse import remove_request_params, sse_client from mcp.server import Server from mcp.server.sse import SseServerTransport from mcp.shared.exceptions import McpError