mirror of
https://github.com/blazickjp/arxiv-mcp-server.git
synced 2025-07-25 20:38:49 +03:00
format update
This commit is contained in:
@@ -3,6 +3,7 @@
|
||||
import pytest
|
||||
from arxiv_mcp_server.prompts.handlers import list_prompts, get_prompt
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_server_list_prompts():
|
||||
"""Test server list_prompts endpoint."""
|
||||
@@ -15,25 +16,25 @@ async def test_server_list_prompts():
|
||||
assert prompt.description
|
||||
assert prompt.arguments is not None
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_server_get_analysis_prompt():
|
||||
"""Test server get_prompt endpoint with analysis prompt."""
|
||||
result = await get_prompt(
|
||||
"deep-paper-analysis",
|
||||
{"paper_id": "2401.00123"}
|
||||
)
|
||||
result = await get_prompt("deep-paper-analysis", {"paper_id": "2401.00123"})
|
||||
|
||||
assert len(result.messages) == 1
|
||||
message = result.messages[0]
|
||||
assert message.role == "user"
|
||||
assert "2401.00123" in message.content.text
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_server_get_prompt_invalid_name():
|
||||
"""Test server get_prompt endpoint with invalid prompt name."""
|
||||
with pytest.raises(ValueError, match="Prompt not found"):
|
||||
await get_prompt("invalid-prompt", {})
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_server_get_prompt_missing_args():
|
||||
"""Test server get_prompt endpoint with missing required arguments."""
|
||||
|
||||
@@ -5,6 +5,7 @@ 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."""
|
||||
@@ -15,13 +16,11 @@ async def test_list_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"}
|
||||
)
|
||||
result = await get_prompt("deep-paper-analysis", {"paper_id": "2401.00123"})
|
||||
|
||||
assert isinstance(result, GetPromptResult)
|
||||
assert len(result.messages) == 1
|
||||
@@ -32,18 +31,21 @@ async def test_get_paper_analysis_prompt():
|
||||
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."""
|
||||
|
||||
Reference in New Issue
Block a user