"""Tests for prompt templates."""
from fastmcp.client import Client
async def test_echo_prompt(client: Client):
"""Test echo prompt."""
result = await client.get_prompt("echo_prompt", {"message": "Hello AI"})
assert len(result.messages) > 0
assert "Hello AI" in str(result.messages[0].content)
async def test_system_prompt_default(client: Client):
"""Test system prompt with default content."""
result = await client.get_prompt("system_prompt", {})
assert len(result.messages) > 0
async def test_system_prompt_custom(client: Client):
"""Test system prompt with custom content."""
result = await client.get_prompt("system_prompt", {"content": "You are a helpful assistant"})
assert len(result.messages) > 0