mirror of
https://github.com/PrefectHQ/fastmcp.git
synced 2026-09-01 20:13:19 +02:00
e9aad2eacb
Co-authored-by: Jeremiah Lowin <jlowin@users.noreply.github.com> Co-authored-by: marvin-context-protocol[bot] <225465937+marvin-context-protocol[bot]@users.noreply.github.com>
26 lines
837 B
Python
26 lines
837 B
Python
from fastmcp.server import FastMCP
|
|
from fastmcp.server.http import create_sse_app, create_streamable_http_app
|
|
|
|
|
|
def test_http_app_sets_mcp_server_state():
|
|
server = FastMCP(name="StateTest")
|
|
app = server.http_app()
|
|
assert app.state.fastmcp_server is server
|
|
|
|
|
|
def test_http_app_sse_sets_mcp_server_state():
|
|
server = FastMCP(name="StateTest")
|
|
app = server.http_app(transport="sse")
|
|
assert app.state.fastmcp_server is server
|
|
|
|
|
|
def test_create_streamable_http_app_sets_state():
|
|
server = FastMCP(name="StateTest")
|
|
app = create_streamable_http_app(server, "/mcp/")
|
|
assert app.state.fastmcp_server is server
|
|
|
|
|
|
def test_create_sse_app_sets_state():
|
|
server = FastMCP(name="StateTest")
|
|
app = create_sse_app(server, message_path="/message", sse_path="/sse/")
|
|
assert app.state.fastmcp_server is server
|