import pytest

from mcp.server import Server

from mcp.types import (

ListPromptsRequest,

ListPromptsResult,

ListResourcesRequest,

ListResourcesResult,

ListToolsRequest,

ListToolsResult,

PaginatedRequestParams,

ServerResult,

)

@pytest.mark.anyio

async def test_list_prompts_pagination() -> None:

server = Server("test")

test_cursor = "test-cursor-123"

# Track what request was received

received_request: ListPromptsRequest | None = None

@server.list_prompts()

async def handle_list_prompts(request: ListPromptsRequest) -> ListPromptsResult:

nonlocal received_request

received_request = request

return ListPromptsResult(prompts=[], nextCursor="next")

handler = server.request_handlers[ListPromptsRequest]

# Test: No cursor provided -> handler receives request with None params

request = ListPromptsRequest(method="prompts/list", params=None)

result = await handler(request)

assert received_request is not None

assert received_request.params is None

assert isinstance(result, ServerResult)

# Test: Cursor provided -> handler receives request with cursor in params

request_with_cursor = ListPromptsRequest(method="prompts/list", params=PaginatedRequestParams(cursor=test_cursor))

result2 = await handler(request_with_cursor)

assert received_request is not None

assert received_request.params is not None

assert received_request.params.cursor == test_cursor

assert isinstance(result2, ServerResult)

@pytest.mark.anyio

async def test_list_resources_pagination() -> None:

server = Server("test")

test_cursor = "resource-cursor-456"

# Track what request was received

received_request: ListResourcesRequest | None = None

@server.list_resources()

async def handle_list_resources(request: ListResourcesRequest) -> ListResourcesResult:

nonlocal received_request

received_request = request

return ListResourcesResult(resources=[], nextCursor="next")

handler = server.request_handlers[ListResourcesRequest]

# Test: No cursor provided -> handler receives request with None params

request = ListResourcesRequest(method="resources/list", params=None)

result = await handler(request)

assert received_request is not None

assert received_request.params is None

assert isinstance(result, ServerResult)

# Test: Cursor provided -> handler receives request with cursor in params

request_with_cursor = ListResourcesRequest(

method="resources/list", params=PaginatedRequestParams(cursor=test_cursor)

)

result2 = await handler(request_with_cursor)

assert received_request is not None

assert received_request.params is not None

assert received_request.params.cursor == test_cursor

assert isinstance(result2, ServerResult)

@pytest.mark.anyio

async def test_list_tools_pagination() -> None:

server = Server("test")

test_cursor = "tools-cursor-789"

# Track what request was received

received_request: ListToolsRequest | None = None

@server.list_tools()

async def handle_list_tools(request: ListToolsRequest) -> ListToolsResult:

nonlocal received_request

received_request = request

return ListToolsResult(tools=[], nextCursor="next")

handler = server.request_handlers[ListToolsRequest]

# Test: No cursor provided -> handler receives request with None params

request = ListToolsRequest(method="tools/list", params=None)

result = await handler(request)

assert received_request is not None

assert received_request.params is None

assert isinstance(result, ServerResult)

# Test: Cursor provided -> handler receives request with cursor in params

request_with_cursor = ListToolsRequest(method="tools/list", params=PaginatedRequestParams(cursor=test_cursor))

result2 = await handler(request_with_cursor)

assert received_request is not None

assert received_request.params is not None

assert received_request.params.cursor == test_cursor

assert isinstance(result2, ServerResult)