mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-10-28 19:06:36 +08:00
48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
"""
|
|
MCP JSON-RPC 请求格式错误的协议回归。
|
|
"""
|
|
|
|
import json
|
|
|
|
import pytest
|
|
from fastapi import FastAPI
|
|
from fastapi.testclient import TestClient
|
|
|
|
from app.api.endpoints import mcp
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("body", "request_id"),
|
|
[
|
|
(None, None),
|
|
([], None),
|
|
([{"jsonrpc": "2.0", "method": "ping", "id": 1}], None),
|
|
("invalid", None),
|
|
(1, None),
|
|
(True, None),
|
|
({}, None),
|
|
({"jsonrpc": "1.0", "id": 7}, 7),
|
|
],
|
|
)
|
|
def test_invalid_request_returns_jsonrpc_error(body: object, request_id: object) -> None:
|
|
"""
|
|
非对象及错误版本的请求须返回 400 协议错误，不能在取 id 时崩溃。
|
|
"""
|
|
app = FastAPI()
|
|
app.include_router(mcp.router, prefix="/mcp")
|
|
app.dependency_overrides[mcp.verify_apikey] = lambda: "test-api-key"
|
|
|
|
with TestClient(app) as client:
|
|
response = client.post(
|
|
"/mcp",
|
|
content=json.dumps(body),
|
|
headers={"Content-Type": "application/json"},
|
|
)
|
|
|
|
assert response.status_code == 400
|
|
assert response.json() == {
|
|
"jsonrpc": "2.0",
|
|
"id": request_id,
|
|
"error": {"code": -32600, "message": "Invalid Request"},
|
|
}
|