mirror of
https://hubproxy.babadafafafafa.cn/https://github.com/jxxghp/MoviePilot.git
synced 2026-09-20 08:03:34 +08:00
317 lines
10 KiB
Python
317 lines
10 KiB
Python
from __future__ import annotations
|
||
|
||
from contextlib import contextmanager
|
||
from unittest.mock import AsyncMock, Mock, patch
|
||
|
||
import pytest
|
||
|
||
from app.adapters.external import server as server_module
|
||
from app.adapters.external.server import (
|
||
MoviePilotServerHelper,
|
||
configure_server_application_services,
|
||
)
|
||
from app.application.server.report import ServerReportService
|
||
from app.application.server.share import ServerSharingService
|
||
from app.schemas.types import MediaSource
|
||
|
||
|
||
@contextmanager
|
||
def _runtime_settings(**values):
|
||
"""按键覆盖中心服务测试配置，其余读取继续委托真实只读端口。"""
|
||
original = server_module.get_runtime_setting
|
||
with patch.object(
|
||
server_module,
|
||
"get_runtime_setting",
|
||
side_effect=lambda key: values[key] if key in values else original(key),
|
||
):
|
||
yield
|
||
|
||
|
||
@pytest.fixture(autouse=True)
|
||
def _configure_server_services() -> None:
|
||
"""
|
||
清理安装用户 ID 缓存，避免不同用例之间互相影响。
|
||
"""
|
||
MoviePilotServerHelper._user_uid = None
|
||
configure_server_application_services(
|
||
report_service=ServerReportService(
|
||
config_reader=Mock(return_value=None),
|
||
config_writer=Mock(),
|
||
installed_plugins_provider=Mock(return_value=[]),
|
||
subscribes_provider=Mock(return_value=[]),
|
||
async_subscribes_provider=AsyncMock(return_value=[]),
|
||
plugin_report_sender=Mock(),
|
||
async_plugin_report_sender=AsyncMock(),
|
||
subscribe_report_sender=Mock(),
|
||
async_subscribe_report_sender=AsyncMock(),
|
||
async_config_writer=AsyncMock(),
|
||
repo_url_sanitizer=MoviePilotServerHelper.sanitize_plugin_repo_url,
|
||
),
|
||
sharing_service=ServerSharingService(
|
||
subscribe_provider=Mock(return_value=None),
|
||
async_subscribe_provider=AsyncMock(return_value=None),
|
||
workflow_provider=Mock(return_value=None),
|
||
async_workflow_provider=AsyncMock(return_value=None),
|
||
user_uuid_provider=Mock(return_value="user-1"),
|
||
subscribe_sender=Mock(),
|
||
async_subscribe_sender=AsyncMock(),
|
||
workflow_sender=Mock(),
|
||
async_workflow_sender=AsyncMock(),
|
||
response_handler=Mock(return_value=(True, "")),
|
||
subscribe_cache_clearer=Mock(),
|
||
workflow_cache_clearer=Mock(),
|
||
),
|
||
)
|
||
|
||
|
||
def test_server_request_adds_user_uid_header():
|
||
"""
|
||
发往 MoviePilot 服务端的请求会自动携带安装用户 ID。
|
||
"""
|
||
with (
|
||
patch.object(MoviePilotServerHelper, "get_user_uid", return_value="uid-1"),
|
||
_runtime_settings(MP_SERVER_HOST="https://movie-pilot.org"),
|
||
):
|
||
headers = MoviePilotServerHelper.build_headers(
|
||
"https://movie-pilot.org/plugin/install",
|
||
headers={"Content-Type": "application/json"},
|
||
)
|
||
|
||
assert headers["X-MoviePilot-User-Uid"] == "uid-1"
|
||
assert headers["Content-Type"] == "application/json"
|
||
|
||
|
||
def test_non_server_request_does_not_add_user_uid_header():
|
||
"""
|
||
发往其他域名的请求不会携带安装用户 ID。
|
||
"""
|
||
with (
|
||
patch.object(MoviePilotServerHelper, "get_user_uid", return_value="uid-1"),
|
||
_runtime_settings(MP_SERVER_HOST="https://movie-pilot.org"),
|
||
):
|
||
headers = MoviePilotServerHelper.build_headers(
|
||
"https://example.com/plugin/install",
|
||
headers={"Content-Type": "application/json"},
|
||
)
|
||
|
||
assert "X-MoviePilot-User-Uid" not in headers
|
||
|
||
|
||
def test_existing_user_uid_header_is_preserved():
|
||
"""
|
||
调用方显式传入的安装用户 ID 请求头不被覆盖。
|
||
"""
|
||
with (
|
||
patch.object(MoviePilotServerHelper, "get_user_uid", return_value="uid-1"),
|
||
_runtime_settings(MP_SERVER_HOST="https://movie-pilot.org"),
|
||
):
|
||
headers = MoviePilotServerHelper.build_headers(
|
||
"https://movie-pilot.org/plugin/install",
|
||
headers={
|
||
"Content-Type": "application/json",
|
||
"X-MoviePilot-User-Uid": "custom-uid",
|
||
},
|
||
)
|
||
|
||
assert headers["X-MoviePilot-User-Uid"] == "custom-uid"
|
||
|
||
|
||
def test_existing_user_uid_header_is_detected_case_insensitively():
|
||
"""
|
||
调用方使用不同大小写的安装用户 ID 请求头时不会重复注入。
|
||
"""
|
||
with (
|
||
patch.object(MoviePilotServerHelper, "get_user_uid", return_value="uid-1"),
|
||
_runtime_settings(MP_SERVER_HOST="https://movie-pilot.org"),
|
||
):
|
||
headers = MoviePilotServerHelper.build_headers(
|
||
"https://movie-pilot.org/plugin/install",
|
||
headers={
|
||
"Content-Type": "application/json",
|
||
"x-moviepilot-user-uid": "custom-uid",
|
||
},
|
||
)
|
||
|
||
assert "X-MoviePilot-User-Uid" not in headers
|
||
assert headers["x-moviepilot-user-uid"] == "custom-uid"
|
||
|
||
|
||
def test_content_type_can_be_added():
|
||
"""
|
||
构建 JSON 请求头时会补充 Content-Type。
|
||
"""
|
||
with (
|
||
patch.object(MoviePilotServerHelper, "get_user_uid", return_value="uid-1"),
|
||
_runtime_settings(MP_SERVER_HOST="https://movie-pilot.org"),
|
||
):
|
||
headers = MoviePilotServerHelper.build_headers(
|
||
"https://movie-pilot.org/plugin/install",
|
||
content_type="application/json",
|
||
)
|
||
|
||
assert headers["Content-Type"] == "application/json"
|
||
|
||
|
||
def test_subscribe_fork_uses_fork_endpoint():
|
||
"""
|
||
订阅复用请求使用服务端 fork 接口。
|
||
"""
|
||
with (
|
||
_runtime_settings(MP_SERVER_HOST="https://movie-pilot.org"),
|
||
patch.object(MoviePilotServerHelper, "_get", return_value=None) as request,
|
||
):
|
||
MoviePilotServerHelper.subscribe_fork(9)
|
||
|
||
request.assert_called_once_with(
|
||
"https://movie-pilot.org/subscribe/fork/9",
|
||
timeout=5,
|
||
)
|
||
|
||
|
||
def test_workflow_fork_uses_fork_endpoint():
|
||
"""
|
||
工作流复用请求使用服务端 fork 接口。
|
||
"""
|
||
with (
|
||
_runtime_settings(MP_SERVER_HOST="https://movie-pilot.org"),
|
||
patch.object(MoviePilotServerHelper, "_get", return_value=None) as request,
|
||
):
|
||
MoviePilotServerHelper.workflow_fork(9)
|
||
|
||
request.assert_called_once_with(
|
||
"https://movie-pilot.org/workflow/fork/9",
|
||
timeout=5,
|
||
)
|
||
|
||
|
||
def test_user_permissions_uses_server_endpoint():
|
||
"""
|
||
用户权限请求使用服务端权限接口。
|
||
"""
|
||
with (
|
||
_runtime_settings(MP_SERVER_HOST="https://movie-pilot.org"),
|
||
patch.object(MoviePilotServerHelper, "_get", return_value=None) as request,
|
||
):
|
||
MoviePilotServerHelper.user_permissions("jxxghp")
|
||
|
||
request.assert_called_once_with(
|
||
"https://movie-pilot.org/user/permissions",
|
||
params={"github_user": "jxxghp"},
|
||
include_user_uid=False,
|
||
timeout=5,
|
||
)
|
||
|
||
|
||
def test_is_admin_user_uses_server_permissions():
|
||
"""
|
||
共享管理权限由服务端权限结果决定。
|
||
"""
|
||
response = Mock(status_code=200)
|
||
response.json.return_value = {"is_admin": True}
|
||
with (
|
||
patch.object(MoviePilotServerHelper, "get_github_user", return_value="jxxghp"),
|
||
patch.object(MoviePilotServerHelper, "user_permissions", return_value=response),
|
||
):
|
||
assert MoviePilotServerHelper.is_admin_user()
|
||
|
||
|
||
def test_is_admin_user_returns_false_without_server_permission():
|
||
"""
|
||
服务端未返回管理权限时不授予共享管理权限。
|
||
"""
|
||
response = Mock(status_code=200)
|
||
response.json.return_value = {"is_admin": False}
|
||
with (
|
||
patch.object(MoviePilotServerHelper, "get_github_user", return_value="user"),
|
||
patch.object(MoviePilotServerHelper, "user_permissions", return_value=response),
|
||
):
|
||
assert not MoviePilotServerHelper.is_admin_user()
|
||
|
||
|
||
def test_subscribe_statistic_payload_only_keeps_server_contract():
|
||
"""订阅统计载荷应删除本地运行列和所有旧专用媒体 ID。"""
|
||
payload = MoviePilotServerHelper._build_subscribe_statistic_payload(
|
||
{
|
||
"id": 1,
|
||
"name": "Test",
|
||
"type": "电影",
|
||
"media_source": MediaSource.Douban,
|
||
"media_id": 42,
|
||
"tmdbid": 99,
|
||
"state": "N",
|
||
"username": "tester",
|
||
}
|
||
)
|
||
|
||
assert payload == {
|
||
"name": "Test",
|
||
"type": "电影",
|
||
"media_source": MediaSource.Douban.value,
|
||
"media_id": "42",
|
||
}
|
||
|
||
|
||
def test_subscribe_share_payload_only_keeps_server_contract():
|
||
"""订阅分享载荷应保留分享配置并剔除本地下载状态。"""
|
||
payload = MoviePilotServerHelper._build_subscribe_share_payload(
|
||
{
|
||
"share_title": "Share",
|
||
"share_user": "tester",
|
||
"name": "Test",
|
||
"type": "电视剧",
|
||
"media_source": "bangumi",
|
||
"media_id": "7",
|
||
"include": "WEB-DL",
|
||
"audio_quality": "lossless",
|
||
"downloader": "default",
|
||
"bangumiid": 7,
|
||
}
|
||
)
|
||
|
||
assert payload == {
|
||
"share_title": "Share",
|
||
"share_user": "tester",
|
||
"name": "Test",
|
||
"type": "电视剧",
|
||
"media_source": MediaSource.Bangumi.value,
|
||
"media_id": "7",
|
||
"include": "WEB-DL",
|
||
}
|
||
|
||
|
||
def test_subscribe_payload_rejects_incomplete_unified_identity():
|
||
"""中心服务载荷不得再从旧专用 ID 推导主身份。"""
|
||
assert (
|
||
MoviePilotServerHelper._build_subscribe_statistic_payload(
|
||
{
|
||
"name": "Legacy",
|
||
"type": "电影",
|
||
"tmdbid": 99,
|
||
}
|
||
)
|
||
is None
|
||
)
|
||
|
||
|
||
def test_durable_subscribe_report_treats_disabled_sharing_as_success():
|
||
"""用户关闭统计分享时，durable intent 应视为无需远端投递。"""
|
||
with (
|
||
_runtime_settings(SUBSCRIBE_STATISTIC_SHARE=False),
|
||
patch.object(
|
||
MoviePilotServerHelper,
|
||
"sub_reg",
|
||
) as reporter,
|
||
):
|
||
assert MoviePilotServerHelper.sub_reg_durable({"media_id": "1"})
|
||
reporter.assert_not_called()
|
||
|
||
with (
|
||
_runtime_settings(SUBSCRIBE_STATISTIC_SHARE=False),
|
||
patch.object(
|
||
MoviePilotServerHelper,
|
||
"sub_done",
|
||
) as reporter,
|
||
):
|
||
assert MoviePilotServerHelper.sub_done_durable({"media_id": "1"})
|
||
reporter.assert_not_called()
|