cryptorugmunch
4d2603cdd5
- api.py, cli.py, mcp_sse.py now import from pry_mcp - tests import from pry_mcp (snapshot tests still verify mcp_production shim)
97 lines
2.9 KiB
Python
97 lines
2.9 KiB
Python
# SPDX-License-Identifier: MIT
|
|
# Copyright (c) 2026 Rug Munch Media LLC
|
|
#
|
|
# Part of Pry — https://git.rugmunch.io/RugMunchMedia/pryscraper
|
|
# Licensed under MIT. See LICENSE.
|
|
"""Tests for production MCP server."""
|
|
|
|
import asyncio
|
|
|
|
from pry_mcp import MCP_PROTOCOL_VERSION, PRY_TOOLS, make_fallback_server, register_all
|
|
|
|
|
|
def test_mcp_protocol_version() -> None:
|
|
assert MCP_PROTOCOL_VERSION is not None
|
|
assert isinstance(MCP_PROTOCOL_VERSION, str)
|
|
assert MCP_PROTOCOL_VERSION == "2024-11-05"
|
|
|
|
|
|
def test_server_creation() -> None:
|
|
server = make_fallback_server()
|
|
register_all(server)
|
|
assert server is not None
|
|
assert len(server.tools) >= 10
|
|
assert len(server.resources) >= 3
|
|
|
|
|
|
def test_required_tools() -> None:
|
|
server = make_fallback_server()
|
|
register_all(server)
|
|
required = [
|
|
"pry_scrape",
|
|
"pry_crawl",
|
|
"pry_extract",
|
|
"pry_template",
|
|
"pry_monitor",
|
|
"pry_compliance",
|
|
"pry_enrich",
|
|
"pry_screenshot",
|
|
]
|
|
for tool_name in required:
|
|
assert tool_name in server.tools, f"Missing tool: {tool_name}"
|
|
|
|
|
|
def test_tool_schemas() -> None:
|
|
for tool in PRY_TOOLS:
|
|
assert "name" in tool
|
|
assert "description" in tool
|
|
assert "inputSchema" in tool
|
|
schema = tool["inputSchema"]
|
|
assert schema.get("type") == "object"
|
|
assert "properties" in schema
|
|
assert "required" in schema
|
|
|
|
|
|
def test_scrape_tool_schema() -> None:
|
|
scrape = next(t for t in PRY_TOOLS if t["name"] == "pry_scrape")
|
|
assert "url" in scrape["inputSchema"]["required"]
|
|
assert "url" in scrape["inputSchema"]["properties"]
|
|
|
|
|
|
def test_resources() -> None:
|
|
server = make_fallback_server()
|
|
register_all(server)
|
|
assert "pry://catalog" in server.resources
|
|
assert "pry://stats" in server.resources
|
|
assert "pry://x402/pricing" in server.resources
|
|
assert "pry://referrals" in server.resources
|
|
|
|
|
|
def test_fallback_initialize() -> None:
|
|
server = make_fallback_server()
|
|
register_all(server)
|
|
result = asyncio.run(server.handle_request({"method": "initialize", "id": 1, "params": {}}))
|
|
assert "protocolVersion" in result["result"]
|
|
assert result["result"]["serverInfo"]["name"] == "pry"
|
|
assert result["result"]["serverInfo"]["version"] == "3.0.0"
|
|
|
|
|
|
def test_fallback_tools_list() -> None:
|
|
server = make_fallback_server()
|
|
register_all(server)
|
|
result = asyncio.run(server.handle_request({"method": "tools/list", "id": 1}))
|
|
assert "tools" in result["result"]
|
|
assert len(result["result"]["tools"]) >= 10
|
|
|
|
|
|
def test_fallback_ping() -> None:
|
|
server = make_fallback_server()
|
|
result = asyncio.run(server.handle_request({"method": "ping", "id": 1}))
|
|
assert result["result"] == {}
|
|
|
|
|
|
def test_fallback_unknown_method() -> None:
|
|
server = make_fallback_server()
|
|
result = asyncio.run(server.handle_request({"method": "unknown", "id": 1}))
|
|
assert "error" in result
|
|
assert result["error"]["code"] == -32601
|