mirror of
https://github.com/PrefectHQ/fastmcp.git
synced 2026-08-24 22:44:17 +02:00
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
from mcp.shared.memory import (
|
|
create_connected_server_and_client_session as client_session,
|
|
)
|
|
from fastmcp.server import FastMCPServer
|
|
|
|
|
|
class TestServer:
|
|
async def test_create_server(self):
|
|
server = FastMCPServer()
|
|
assert server.name == "FastMCPServer"
|
|
|
|
|
|
def tool_fn(x: int, y: int) -> int:
|
|
return x + y
|
|
|
|
|
|
class TestServerTools:
|
|
async def test_add_tool(self):
|
|
server = FastMCPServer()
|
|
server.add_tool(tool_fn)
|
|
server.add_tool(tool_fn)
|
|
assert len(server._tool_manager.list_tools()) == 1
|
|
|
|
async def test_list_tools(self):
|
|
server = FastMCPServer()
|
|
server.add_tool(tool_fn)
|
|
async with client_session(server._mcp_server) as client:
|
|
tools = await client.list_tools()
|
|
assert len(tools.tools) == 1
|
|
|
|
async def test_call_tool(self):
|
|
server = FastMCPServer()
|
|
server.add_tool(tool_fn)
|
|
async with client_session(server._mcp_server) as client:
|
|
result = await client.call_tool("my_tool", {"arg1": "value"})
|
|
assert "error" not in result
|
|
assert len(result.content) > 0
|