mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-10-30 03:48:03 +08:00
275 lines
9.6 KiB
Python
275 lines
9.6 KiB
Python
import asyncio
|
||
import json
|
||
from types import SimpleNamespace
|
||
from typing import Optional
|
||
from unittest.mock import AsyncMock, MagicMock, patch
|
||
|
||
import pytest
|
||
|
||
from app.agent.api.executor import ApiExecutionContext, MoviePilotApiExecutor
|
||
from app.agent.tools.base import format_tool_result_for_agent
|
||
from app.agent.tools.manager import MoviePilotToolsManager
|
||
from app.schemas.types import NotificationChannel
|
||
|
||
|
||
@pytest.mark.parametrize("app_domain", ["", "https://custom.domain.example:5443/moviepilot/"])
|
||
@pytest.mark.parametrize(
|
||
("host", "port", "base_url"),
|
||
[
|
||
(None, 3001, "http://127.0.0.1:3001"),
|
||
("", 3001, "http://127.0.0.1:3001"),
|
||
("0.0.0.0", 13001, "http://127.0.0.1:13001"),
|
||
("127.0.0.1", 3001, "http://127.0.0.1:3001"),
|
||
("192.0.2.10", 13001, "http://192.0.2.10:13001"),
|
||
("localhost", 13001, "http://localhost:13001"),
|
||
("::", 13001, "http://[::1]:13001"),
|
||
("[::]", 13001, "http://[::1]:13001"),
|
||
("::1", 3001, "http://[::1]:3001"),
|
||
("[::1]", 3001, "http://[::1]:3001"),
|
||
("2001:db8::10", 13001, "http://[2001:db8::10]:13001"),
|
||
("[2001:db8::10]", 13001, "http://[2001:db8::10]:13001"),
|
||
],
|
||
)
|
||
def test_executor_requests_local_listener_independently_of_public_domain(
|
||
app_domain: str,
|
||
host: Optional[str],
|
||
port: int,
|
||
base_url: str,
|
||
) -> None:
|
||
"""配置对外反代域名后，API 仍须按本机监听地址和端口发送且绕过环境代理。"""
|
||
runtime_settings = {"APP_DOMAIN": app_domain, "HOST": host, "PORT": port}
|
||
response = SimpleNamespace(
|
||
status_code=200,
|
||
headers={},
|
||
json=lambda: {"success": True, "data": []},
|
||
aclose=AsyncMock(),
|
||
)
|
||
request = AsyncMock(return_value=response)
|
||
request_factory = MagicMock(return_value=SimpleNamespace(request=request))
|
||
executor = MoviePilotApiExecutor(
|
||
context=ApiExecutionContext(user_id="1", username="admin", is_admin=True),
|
||
request_factory=request_factory,
|
||
)
|
||
|
||
with (
|
||
patch("app.agent.api.executor.get_runtime_setting", side_effect=runtime_settings.get),
|
||
patch("app.agent.api.executor.create_access_token", return_value="token"),
|
||
):
|
||
result = asyncio.run(executor.execute("subscription.list", query={"page": 1}))
|
||
|
||
assert json.loads(result) == {"success": True, "data": []}
|
||
request.assert_awaited_once_with(
|
||
method="GET",
|
||
url=f"{base_url}/api/v1/subscribe/",
|
||
params={"page": 1},
|
||
json=None,
|
||
raise_exception=True,
|
||
)
|
||
assert request_factory.call_args.kwargs["headers"]["Authorization"] == "Bearer token"
|
||
assert request_factory.call_args.kwargs["trust_env"] is False
|
||
response.aclose.assert_awaited_once()
|
||
|
||
|
||
def _execute_with_headers(headers: dict[str, str]) -> tuple[dict, AsyncMock]:
|
||
"""用内存 HTTP 响应执行一次固定 API operation。"""
|
||
response = SimpleNamespace(
|
||
status_code=200,
|
||
headers=headers,
|
||
json=lambda: {"success": True, "message": "", "data": [{"id": 1}]},
|
||
aclose=AsyncMock(),
|
||
)
|
||
request = AsyncMock(return_value=response)
|
||
request_factory = MagicMock(return_value=SimpleNamespace(request=request))
|
||
executor = MoviePilotApiExecutor(
|
||
context=ApiExecutionContext(
|
||
user_id="1",
|
||
username="admin",
|
||
is_admin=True,
|
||
),
|
||
request_factory=request_factory,
|
||
)
|
||
|
||
with patch("app.agent.api.executor.create_access_token", return_value="token"):
|
||
result = asyncio.run(executor.execute("subscription.list"))
|
||
|
||
return json.loads(result), response.aclose
|
||
|
||
|
||
def test_executor_exposes_exact_collection_total_when_api_reports_it() -> None:
|
||
"""Agent 结果应把精确数量响应头变成可直接读取的集合元数据。"""
|
||
result, close = _execute_with_headers(
|
||
{
|
||
"X-Result-Count": "20",
|
||
"X-Total-Count": "57",
|
||
"X-Page": "2",
|
||
"X-Page-Size": "20",
|
||
}
|
||
)
|
||
|
||
assert result["data"] == [{"id": 1}]
|
||
assert result["collection"] == {
|
||
"result_count": 20,
|
||
"total_count": 57,
|
||
"page": 2,
|
||
"count": 20,
|
||
}
|
||
close.assert_awaited_once()
|
||
|
||
|
||
def test_executor_does_not_invent_total_for_upstream_window() -> None:
|
||
"""外部接口未报告总数时，Agent 元数据只能包含当前窗口数量。"""
|
||
result, close = _execute_with_headers(
|
||
{
|
||
"X-Result-Count": "20",
|
||
"X-Page": "3",
|
||
"X-Page-Size": "20",
|
||
}
|
||
)
|
||
|
||
assert result["collection"] == {
|
||
"result_count": 20,
|
||
"page": 3,
|
||
"count": 20,
|
||
}
|
||
assert "total_count" not in result["collection"]
|
||
close.assert_awaited_once()
|
||
|
||
|
||
def test_executor_keeps_non_collection_payload_unchanged_without_headers() -> None:
|
||
"""非列表 API 未返回数量头时必须保持既有输出形状。"""
|
||
result, close = _execute_with_headers({})
|
||
|
||
assert result == {"success": True, "message": "", "data": [{"id": 1}]}
|
||
close.assert_awaited_once()
|
||
|
||
|
||
def _execute_and_get_request_timeout(
|
||
operation_id: str,
|
||
*,
|
||
path_params: Optional[dict] = None,
|
||
query: Optional[dict] = None,
|
||
) -> float:
|
||
"""执行一个内存 API 请求并返回构造 HTTP 客户端时采用的超时。"""
|
||
response = SimpleNamespace(
|
||
status_code=200,
|
||
headers={},
|
||
json=lambda: {"success": True},
|
||
aclose=AsyncMock(),
|
||
)
|
||
request_factory = MagicMock(
|
||
return_value=SimpleNamespace(request=AsyncMock(return_value=response))
|
||
)
|
||
executor = MoviePilotApiExecutor(
|
||
context=ApiExecutionContext(user_id="1", username="admin", is_admin=True),
|
||
request_factory=request_factory,
|
||
)
|
||
|
||
with patch("app.agent.api.executor.create_access_token", return_value="token"):
|
||
asyncio.run(
|
||
executor.execute(
|
||
operation_id,
|
||
path_params=path_params,
|
||
query=query,
|
||
)
|
||
)
|
||
|
||
return request_factory.call_args.kwargs["timeout"]
|
||
|
||
|
||
def test_executor_extends_request_timeout_only_for_search_torrents() -> None:
|
||
"""慢速种子搜索允许等待 290 秒，其它 operation 保持原有 30 秒。"""
|
||
assert _execute_and_get_request_timeout(
|
||
"search.torrents",
|
||
path_params={"media_id": "1423191"},
|
||
query={"media_source": "themoviedb"},
|
||
) == 290
|
||
assert _execute_and_get_request_timeout(
|
||
"search.title",
|
||
query={"keyword": "生化危机：爆发夜"},
|
||
) == 30
|
||
assert _execute_and_get_request_timeout("subscription.list") == 30
|
||
|
||
|
||
def test_manager_default_session_id_is_http_header_compatible() -> None:
|
||
"""HTTP API 工具管理器生成的默认会话标识必须是可发送的字符串。"""
|
||
manager = MoviePilotToolsManager(user_id="1")
|
||
executor = MoviePilotApiExecutor(
|
||
context=ApiExecutionContext(
|
||
user_id="1",
|
||
username="admin",
|
||
is_admin=True,
|
||
session_id=manager.session_id,
|
||
)
|
||
)
|
||
|
||
with patch("app.agent.api.executor.create_access_token", return_value="token"):
|
||
headers = executor._build_headers()
|
||
|
||
session_header = headers["X-MoviePilot-Agent-Session"]
|
||
assert isinstance(manager.session_id, str)
|
||
assert isinstance(session_header, str)
|
||
assert session_header == manager.session_id
|
||
|
||
|
||
def test_executor_keeps_collection_total_visible_in_truncated_tool_preview() -> None:
|
||
"""列表内容过大时，精确总数必须位于截断预览开头供 Agent 直接使用。"""
|
||
response = SimpleNamespace(
|
||
status_code=200,
|
||
headers={"X-Result-Count": "200", "X-Total-Count": "357"},
|
||
json=lambda: {
|
||
"success": True,
|
||
"message": "",
|
||
"data": [{"title": "x" * 1024} for _ in range(100)],
|
||
},
|
||
aclose=AsyncMock(),
|
||
)
|
||
request = AsyncMock(return_value=response)
|
||
request_factory = MagicMock(return_value=SimpleNamespace(request=request))
|
||
executor = MoviePilotApiExecutor(
|
||
context=ApiExecutionContext(
|
||
user_id="1",
|
||
username="admin",
|
||
is_admin=True,
|
||
),
|
||
request_factory=request_factory,
|
||
)
|
||
|
||
with patch("app.agent.api.executor.create_access_token", return_value="token"):
|
||
raw_result = asyncio.run(executor.execute("subscription.list"))
|
||
|
||
result = json.loads(
|
||
format_tool_result_for_agent(raw_result, tool_name="moviepilot_api")
|
||
)
|
||
|
||
assert result["tool_result_truncated"] is True
|
||
assert result["content_preview"].startswith(
|
||
'{"collection": {"result_count": 200, "total_count": 357}'
|
||
)
|
||
assert result["content_preview"].index('"total_count": 357') < result[
|
||
"content_preview"
|
||
].index('"data"')
|
||
response.aclose.assert_awaited_once()
|
||
|
||
|
||
def test_executor_serializes_channel_and_source_headers_as_ascii() -> None:
|
||
"""中文通知渠道和来源必须以 ASCII 形式发送到 Agent API。"""
|
||
request_factory = MagicMock(return_value=SimpleNamespace(request=AsyncMock()))
|
||
executor = MoviePilotApiExecutor(
|
||
context=ApiExecutionContext(
|
||
user_id="1",
|
||
username="admin",
|
||
is_admin=True,
|
||
channel=NotificationChannel.Wechat.value,
|
||
source="企业微信",
|
||
),
|
||
request_factory=request_factory,
|
||
)
|
||
|
||
with patch("app.agent.api.executor.create_access_token", return_value="token"):
|
||
headers = executor._build_headers()
|
||
|
||
assert headers["X-MoviePilot-Agent-Channel"] == "%E5%BE%AE%E4%BF%A1"
|
||
assert headers["X-MoviePilot-Agent-Channel"].isascii()
|
||
assert headers["X-MoviePilot-Agent-Source"] == "%E4%BC%81%E4%B8%9A%E5%BE%AE%E4%BF%A1"
|
||
assert headers["X-MoviePilot-Agent-Source"].isascii()
|