mirror of
https://github.com/blazickjp/arxiv-mcp-server.git
synced 2025-07-25 20:38:49 +03:00
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
"""Unit tests for prompt handlers."""
|
|
|
|
import pytest
|
|
from typing import Dict
|
|
from arxiv_mcp_server.prompts.handlers import list_prompts, get_prompt
|
|
from mcp.types import GetPromptResult, PromptMessage, TextContent
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_prompts():
|
|
"""Test listing available prompts."""
|
|
prompts = await list_prompts()
|
|
assert len(prompts) == 1
|
|
|
|
prompt_names = {p.name for p in prompts}
|
|
expected_names = {"deep-paper-analysis"}
|
|
assert prompt_names == expected_names
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_paper_analysis_prompt():
|
|
"""Test getting paper analysis prompt."""
|
|
result = await get_prompt(
|
|
"deep-paper-analysis",
|
|
{"paper_id": "2401.00123"}
|
|
)
|
|
|
|
assert isinstance(result, GetPromptResult)
|
|
assert len(result.messages) == 1
|
|
message = result.messages[0]
|
|
|
|
assert isinstance(message, PromptMessage)
|
|
assert message.role == "user"
|
|
assert isinstance(message.content, TextContent)
|
|
assert "2401.00123" in message.content.text
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_prompt_with_invalid_name():
|
|
"""Test getting prompt with invalid name."""
|
|
with pytest.raises(ValueError, match="Prompt not found"):
|
|
await get_prompt("invalid-prompt", {})
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_prompt_with_no_arguments():
|
|
"""Test getting prompt with no arguments."""
|
|
with pytest.raises(ValueError, match="No arguments provided"):
|
|
await get_prompt("deep-paper-analysis", None)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_prompt_with_missing_required_argument():
|
|
"""Test getting prompt with missing required argument."""
|
|
with pytest.raises(ValueError, match="Missing required argument"):
|
|
await get_prompt("deep-paper-analysis", {}) |