oterm/tests/tools/test_mcp_transports.py
2025-06-12 10:46:32 +02:00

26 lines
746 B
Python

from fastmcp.client.transports import (
StdioTransport,
StreamableHttpTransport,
WSTransport,
)
from oterm.tools.mcp.client import MCPClient
def test_stdio_transport(mcp_server_config):
"""
Test the MCP client with a StdioServerParameters.
"""
client = MCPClient("test_stdio", mcp_server_config["stdio"])
assert isinstance(client.transport, StdioTransport)
def test_streamable_http(mcp_server_config):
client = MCPClient("test_streamable_http", mcp_server_config["streamable_http"])
assert isinstance(client.transport, StreamableHttpTransport)
def test_ws_transport(mcp_server_config):
client = MCPClient("test_ws", mcp_server_config["ws"])
assert isinstance(client.transport, WSTransport)