87 lines
2.3 KiB
Python
87 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
测试 Funstat MCP Server
|
||
|
||
这个脚本会测试 MCP 服务器的所有功能
|
||
"""
|
||
|
||
import asyncio
|
||
import sys
|
||
from server import FunstatMCPServer
|
||
|
||
async def test_server():
|
||
"""测试服务器功能"""
|
||
|
||
print("=" * 60)
|
||
print("🧪 Funstat MCP Server 测试")
|
||
print("=" * 60)
|
||
print()
|
||
|
||
server = FunstatMCPServer()
|
||
|
||
try:
|
||
# 初始化
|
||
print("1️⃣ 初始化服务器...")
|
||
await server.initialize()
|
||
print("✅ 初始化成功")
|
||
print()
|
||
|
||
# 测试工具列表
|
||
print("2️⃣ 获取工具列表...")
|
||
tools = await server.list_tools()
|
||
print(f"✅ 找到 {len(tools)} 个工具:")
|
||
for tool in tools:
|
||
print(f" - {tool.name}: {tool.description}")
|
||
print()
|
||
|
||
# 测试 /start 命令
|
||
print("3️⃣ 测试 /start 命令...")
|
||
result = await server.call_tool("funstat_start", {})
|
||
response = result[0].text
|
||
print("✅ 响应:")
|
||
print(response[:500])
|
||
if len(response) > 500:
|
||
print(f" ... (还有 {len(response) - 500} 个字符)")
|
||
print()
|
||
|
||
# 测试 /balance 命令
|
||
print("4️⃣ 测试 /balance 命令...")
|
||
result = await server.call_tool("funstat_balance", {})
|
||
print("✅ 响应:")
|
||
print(result[0].text)
|
||
print()
|
||
|
||
# 测试搜索功能
|
||
print("5️⃣ 测试搜索功能 (关键词: python)...")
|
||
result = await server.call_tool("funstat_search", {"query": "python"})
|
||
response = result[0].text
|
||
print("✅ 响应:")
|
||
print(response[:500])
|
||
if len(response) > 500:
|
||
print(f" ... (还有 {len(response) - 500} 个字符)")
|
||
print()
|
||
|
||
# 测试缓存
|
||
print("6️⃣ 测试缓存 (再次搜索 python)...")
|
||
result = await server.call_tool("funstat_search", {"query": "python"})
|
||
print("✅ 响应: (应该来自缓存)")
|
||
print(result[0].text[:200])
|
||
print()
|
||
|
||
print("=" * 60)
|
||
print("✅ 所有测试通过!")
|
||
print("=" * 60)
|
||
|
||
except Exception as e:
|
||
print(f"❌ 测试失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
sys.exit(1)
|
||
|
||
finally:
|
||
if server.client:
|
||
await server.client.disconnect()
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(test_server())
|