Spaces:

purecipher
/

xsecuremcp

Sleeping

xsecuremcp / tests /server /test_app_state.py

Jeremiah Lowin

Update docs and test

7b03b73 over 1 year ago

837 Bytes

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

	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