tests: cleanup

This commit is contained in:
David Soria Parra 2025-02-05 11:02:51 +00:00
commit f10665db4c
No known key found for this signature in database

View file

@ -1,4 +1,5 @@
import pytest
from pydantic import AnyUrl
from typing_extensions import AsyncGenerator
from mcp.client.session import ClientSession
@ -8,9 +9,27 @@ from mcp.shared.memory import (
)
from mcp.types import (
EmptyResult,
Resource,
)
@pytest.fixture
def mcp_server() -> Server:
server = Server(name="test_server")
@server.list_resources()
async def handle_list_resources():
return [
Resource(
uri=AnyUrl("memory://test"),
name="Test Resource",
description="A test resource",
)
]
return server
@pytest.fixture
async def client_connected_to_server(
mcp_server: Server,