mirror of
https://hubproxy.babadafafafafa.cn/https://github.com/jxxghp/MoviePilot.git
synced 2026-09-20 08:03:34 +08:00
120 lines
4.1 KiB
Python
120 lines
4.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
from types import SimpleNamespace
|
|
from unittest.mock import Mock
|
|
|
|
import app.modules.indexer as indexer_module
|
|
from app.modules.indexer import IndexerModule
|
|
from app.modules.indexer.parser import SiteSchema
|
|
|
|
|
|
def _build_parser_constructor(schema_value: str, calls: list[str]) -> Mock:
|
|
"""构造记录解析调用的站点解析器桩。"""
|
|
constructor = Mock(name=f"{schema_value}Parser")
|
|
constructor.schema = SimpleNamespace(value=schema_value)
|
|
|
|
def build_parser(**_kwargs):
|
|
"""构造一个解析失败且不发起真实请求的解析器实例。"""
|
|
parser = Mock()
|
|
parser.userid = None
|
|
parser.username = None
|
|
parser.user_level = None
|
|
parser.join_at = None
|
|
parser.upload = 0
|
|
parser.download = 0
|
|
parser.ratio = 0
|
|
parser.bonus = 0
|
|
parser.seeding = 0
|
|
parser.seeding_size = 0
|
|
parser.seeding_info = []
|
|
parser.leeching = 0
|
|
parser.leeching_size = 0
|
|
parser.message_unread = 0
|
|
parser.message_unread_contents = []
|
|
parser.err_msg = None
|
|
parser.parse.side_effect = lambda: calls.append(schema_value)
|
|
return parser
|
|
|
|
constructor.side_effect = build_parser
|
|
return constructor
|
|
|
|
|
|
def test_refresh_userdata_fallback_only_uses_common_schemas(monkeypatch):
|
|
"""声明模型失败时只尝试常用模型，不触发专用模型兜底请求。"""
|
|
calls = []
|
|
parser_constructors = [
|
|
_build_parser_constructor(SiteSchema.NexusPhp.value, calls),
|
|
_build_parser_constructor(SiteSchema.Gazelle.value, calls),
|
|
_build_parser_constructor(SiteSchema.Unit3d.value, calls),
|
|
_build_parser_constructor(SiteSchema.TNode.value, calls),
|
|
_build_parser_constructor(SiteSchema.HDDolby.value, calls),
|
|
]
|
|
monkeypatch.setattr(IndexerModule, "_site_schemas", parser_constructors)
|
|
|
|
module = object.__new__(IndexerModule)
|
|
result = module.refresh_userdata({
|
|
"name": "音乐乌托邦",
|
|
"url": "https://www.musopia.vip/",
|
|
"schema": SiteSchema.NexusPhp.value,
|
|
"public": False,
|
|
})
|
|
|
|
assert result.userid is None
|
|
assert result.err_msg == "未获取到站点用户信息，请检查 Cookie 是否有效"
|
|
assert calls == [
|
|
SiteSchema.NexusPhp.value,
|
|
SiteSchema.Gazelle.value,
|
|
SiteSchema.Unit3d.value,
|
|
]
|
|
assert not parser_constructors[3].called
|
|
assert not parser_constructors[4].called
|
|
|
|
|
|
def test_refresh_userdata_uses_declared_page_and_runtime_user_agent(monkeypatch):
|
|
"""用户数据刷新应使用站点声明页面，并为未配置站点 UA 补充运行时 UA。"""
|
|
captured = {}
|
|
constructor = Mock(name="NexusPhpParser")
|
|
constructor.schema = SimpleNamespace(value=SiteSchema.NexusPhp.value)
|
|
|
|
def build_parser(**kwargs):
|
|
"""构造一个成功返回用户 ID 的解析器桩。"""
|
|
captured.update(kwargs)
|
|
parser = Mock()
|
|
parser.userid = "123"
|
|
parser.username = "tester"
|
|
parser.user_level = None
|
|
parser.join_at = None
|
|
parser.upload = 0
|
|
parser.download = 0
|
|
parser.ratio = 0
|
|
parser.bonus = 0
|
|
parser.seeding = 0
|
|
parser.seeding_size = 0
|
|
parser.seeding_info = []
|
|
parser.leeching = 0
|
|
parser.leeching_size = 0
|
|
parser.message_unread = 0
|
|
parser.message_unread_contents = []
|
|
parser.err_msg = None
|
|
return parser
|
|
|
|
constructor.side_effect = build_parser
|
|
monkeypatch.setattr(IndexerModule, "_site_schemas", [constructor])
|
|
monkeypatch.setattr(
|
|
indexer_module,
|
|
"get_runtime_setting",
|
|
lambda key: "MoviePilot-Test-UA" if key == "USER_AGENT" else None,
|
|
)
|
|
|
|
module = object.__new__(IndexerModule)
|
|
result = module.refresh_userdata({
|
|
"name": "音乐乌托邦",
|
|
"url": "https://www.musopia.vip/",
|
|
"schema": SiteSchema.NexusPhp.value,
|
|
"userdata_path": "index.php",
|
|
"public": False,
|
|
})
|
|
|
|
assert result.userid == "123"
|
|
assert captured["url"] == "https://www.musopia.vip/index.php"
|
|
assert captured["ua"] == "MoviePilot-Test-UA"
|