41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
pytestmark = pytest.mark.smoke
|
|
|
|
|
|
class TestMcpPromptSurface:
|
|
"""Covers smoke-level MCP prompt discovery and retrieval paths."""
|
|
|
|
class TestPromptDiscovery:
|
|
"""Covers MCP prompts/list behavior using native prompt objects."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_lists_prompt_objects(self, mcp_session_factory) -> None:
|
|
"""Ensures prompts/list returns native prompt objects with stable names."""
|
|
async with mcp_session_factory() as mcp_session:
|
|
result = await mcp_session.list_prompts()
|
|
|
|
assert result.prompts
|
|
assert all(prompt.name for prompt in result.prompts)
|
|
|
|
class TestPromptResolution:
|
|
"""Covers MCP prompts/get behavior using native request and response objects."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_gets_prompt_as_native_object(self, mcp_session_factory) -> None:
|
|
"""Ensures prompts/get resolves a listed prompt into structured message objects."""
|
|
async with mcp_session_factory() as mcp_session:
|
|
listed_prompts = await mcp_session.list_prompts()
|
|
prompt = listed_prompts.prompts[0]
|
|
arguments = {arg.name: "test" for arg in (prompt.arguments or []) if arg.required}
|
|
|
|
resolved_prompt = await mcp_session.get_prompt(
|
|
name=prompt.name,
|
|
arguments=arguments or None,
|
|
)
|
|
|
|
assert resolved_prompt.messages
|
|
assert all(message.content for message in resolved_prompt.messages)
|