Move StreamingASGITransport into the library code, so MCP integrations can use this in their tests

praboud-ant · praboud-ant · commit 56f694e16aa0 · 2025-03-19T15:11:31.000-07:00
diff --git a/src/mcp/server/streaming_asgi_transport.py b/src/mcp/server/streaming_asgi_transport.py
@@ -4,6 +4,8 @@
 This transport runs the ASGI app as a separate anyio task, allowing it to
 handle streaming responses like SSE where the app doesn't terminate until
 the connection is closed.
+
+This is only intended for writing tests for the SSE transport.
 """
 
 import typing
diff --git a/tests/server/fastmcp/auth/test_auth_integration.py b/tests/server/fastmcp/auth/test_auth_integration.py
@@ -39,7 +39,7 @@
 )
 from mcp.types import JSONRPCRequest
 
-from .streaming_asgi_transport import StreamingASGITransport
+from mcp.server.streaming_asgi_transport import StreamingASGITransport
 
 
 # Mock client store for testing

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,7 @@`
`39`	`39`	`)`
`40`	`40`	`from mcp.types import JSONRPCRequest`
`41`	`41`
`42`		`-from .streaming_asgi_transport import StreamingASGITransport`
	`42`	`+from mcp.server.streaming_asgi_transport import StreamingASGITransport`
`43`	`43`
`44`	`44`
`45`	`45`	`# Mock client store for testing`