Marcelo Trylesinski
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
from typing import Any
|
|
|
|
import pytest
|
|
|
|
import mcp.types as types
|
|
from mcp import Client
|
|
from mcp.server import Server
|
|
from mcp.server.session import ServerSession
|
|
from mcp.shared.context import RequestContext
|
|
from mcp.types import EmptyResult, Resource
|
|
|
|
|
|
@pytest.fixture
|
|
def mcp_server() -> Server:
|
|
async def handle_list_resources(
|
|
ctx: RequestContext[ServerSession, Any, Any],
|
|
params: types.PaginatedRequestParams | None,
|
|
) -> types.ListResourcesResult: # pragma: no cover
|
|
return types.ListResourcesResult(
|
|
resources=[
|
|
Resource(
|
|
uri="memory://test",
|
|
name="Test Resource",
|
|
description="A test resource",
|
|
)
|
|
]
|
|
)
|
|
|
|
return Server(name="test_server", on_list_resources=handle_list_resources)
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_memory_server_and_client_connection(mcp_server: Server):
|
|
"""Shows how a client and server can communicate over memory streams."""
|
|
async with Client(mcp_server) as client:
|
|
response = await client.send_ping()
|
|
assert isinstance(response, EmptyResult)
|