From f080e0847f0aad0270f5e4beccdda0794235fbbe Mon Sep 17 00:00:00 2001 From: Chris Farhood Date: Mon, 6 Jul 2026 22:20:36 -0400 Subject: [PATCH] test: mock uvicorn.run in streamable-http start test (CORS app path) --- tests/test_server_setup.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/tests/test_server_setup.py b/tests/test_server_setup.py index 0649f40..abf5220 100644 --- a/tests/test_server_setup.py +++ b/tests/test_server_setup.py @@ -61,10 +61,16 @@ def test_start_server_stdio(): m.run.assert_called_once_with() -def test_start_server_streamable_http(): +def test_start_server_streamable_http(monkeypatch): m = _mock_mcp() + calls = {} + monkeypatch.setattr("uvicorn.run", lambda app, **kw: calls.update(app=app, kw=kw)) server_setup.start_server(m, TransportAliases.STREAMABLE_HTTP) - m.run.assert_called_once_with(transport="streamable-http") + # builds the app, adds CORS middleware, and serves it via uvicorn + m.streamable_http_app.assert_called_once_with() + m.streamable_http_app.return_value.add_middleware.assert_called_once() + assert calls["app"] is m.streamable_http_app.return_value + assert calls["kw"]["host"] == "0.0.0.0" and calls["kw"]["port"] == 8080 def test_start_server_sse_uses_mount_path(monkeypatch):