mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-06-21 23:44:31 +08:00
266 lines
8.9 KiB
Python
266 lines
8.9 KiB
Python
from unittest.mock import patch
|
|
|
|
from app.modules.zspace.zspace import ZSpace
|
|
|
|
|
|
class _FakeResponse:
|
|
"""模拟极影视 HTTP 响应。"""
|
|
|
|
def __init__(self, payload: dict, status_code: int = 200):
|
|
self._payload = payload
|
|
self.status_code = status_code
|
|
|
|
def json(self) -> dict:
|
|
"""返回模拟的 JSON 数据。"""
|
|
return self._payload
|
|
|
|
|
|
def _build_client() -> ZSpace:
|
|
"""构造不触发登录流程的极影视客户端。"""
|
|
client = ZSpace.__new__(ZSpace)
|
|
client._host = "http://zspace.local/"
|
|
client._apikey = "zspace-token"
|
|
client.user = "user-id"
|
|
return client
|
|
|
|
|
|
def test_get_items_fetches_all_recursive_pages() -> None:
|
|
"""全量同步应递归查询并持续翻页，且忽略非媒体条目。"""
|
|
client = _build_client()
|
|
responses = [
|
|
_FakeResponse({
|
|
"Items": [
|
|
{
|
|
"Id": "movie-1",
|
|
"Type": "Movie",
|
|
"Name": "电影一",
|
|
"ProductionYear": 2025,
|
|
"ProviderIds": {"Tmdb": "101"},
|
|
"Path": "/media/movie-1.mkv",
|
|
},
|
|
{
|
|
"Id": "audio-1",
|
|
"Type": "Audio",
|
|
"Name": "音频一",
|
|
},
|
|
],
|
|
"TotalRecordCount": 3,
|
|
}),
|
|
_FakeResponse({
|
|
"Items": [
|
|
{
|
|
"Id": "series-1",
|
|
"Type": "Series",
|
|
"Name": "剧集一",
|
|
"ProductionYear": 2024,
|
|
"ProviderIds": {"Tmdb": "202"},
|
|
"Path": "/media/series-1",
|
|
},
|
|
],
|
|
"TotalRecordCount": 3,
|
|
}),
|
|
]
|
|
|
|
with patch("app.modules.zspace.zspace.DEFAULT_ITEMS_PAGE_SIZE", 2), patch(
|
|
"app.modules.zspace.zspace.RequestUtils"
|
|
) as request_utils_cls:
|
|
request_utils_cls.return_value.get_res.side_effect = responses
|
|
items = list(client.get_items(parent="library-id"))
|
|
|
|
assert [item.item_id for item in items] == ["movie-1", "series-1"]
|
|
calls = request_utils_cls.return_value.get_res.call_args_list
|
|
assert calls[0].kwargs["params"]["Recursive"] == "true"
|
|
assert calls[0].kwargs["params"]["StartIndex"] == 0
|
|
assert calls[0].kwargs["params"]["Limit"] == 2
|
|
assert calls[1].kwargs["params"]["StartIndex"] == 2
|
|
assert calls[1].kwargs["params"]["Limit"] == 2
|
|
|
|
|
|
def test_get_items_expands_boxset_movies() -> None:
|
|
"""电影合集应向下查询并返回全部子电影。"""
|
|
client = _build_client()
|
|
responses = [
|
|
_FakeResponse({
|
|
"Items": [{
|
|
"Id": "collection-1",
|
|
"Type": "BoxSet",
|
|
"Name": "疯狂动物城（系列）",
|
|
}],
|
|
"TotalRecordCount": 1,
|
|
}),
|
|
_FakeResponse({
|
|
"Items": [
|
|
{
|
|
"Id": "movie-1",
|
|
"Type": "Movie",
|
|
"Name": "疯狂动物城",
|
|
"ProductionYear": None,
|
|
"ProviderIds": {},
|
|
"Path": "",
|
|
},
|
|
{
|
|
"Id": "movie-2",
|
|
"Type": "Movie",
|
|
"Name": "疯狂动物城2",
|
|
"ProductionYear": 2025,
|
|
"ProviderIds": {"Tmdb": "1084242"},
|
|
"Path": "/media/疯狂动物城2.mkv",
|
|
},
|
|
],
|
|
"TotalRecordCount": 2,
|
|
}),
|
|
_FakeResponse({
|
|
"Id": "movie-1",
|
|
"ParentId": "collection-1",
|
|
"Type": "Movie",
|
|
"Name": "疯狂动物城",
|
|
"ProductionYear": 2016,
|
|
"ProviderIds": {"Tmdb": "269149"},
|
|
"Path": "/media/疯狂动物城.mkv",
|
|
}),
|
|
]
|
|
|
|
with patch("app.modules.zspace.zspace.RequestUtils") as request_utils_cls:
|
|
request_utils_cls.return_value.get_res.side_effect = responses
|
|
items = list(client.get_items(parent="library-id"))
|
|
|
|
assert [item.item_id for item in items] == ["movie-1", "movie-2"]
|
|
assert [item.tmdbid for item in items] == [269149, 1084242]
|
|
calls = request_utils_cls.return_value.get_res.call_args_list
|
|
assert calls[0].kwargs["params"]["Recursive"] == "true"
|
|
assert calls[1].kwargs["params"]["ParentId"] == "collection-1"
|
|
|
|
|
|
def test_get_items_expands_boxset_series() -> None:
|
|
"""电视剧合集应向下展开并返回全部子 Series。"""
|
|
client = _build_client()
|
|
responses = [
|
|
_FakeResponse({
|
|
"Items": [{
|
|
"Id": "collection-1",
|
|
"Type": "BoxSet",
|
|
"Name": "绝命毒师",
|
|
}],
|
|
"TotalRecordCount": 1,
|
|
}),
|
|
_FakeResponse({
|
|
"Items": [
|
|
{
|
|
"Id": "series-1",
|
|
"Type": "Series",
|
|
"Name": "绝命毒师 第 1 季",
|
|
"ProductionYear": 2008,
|
|
"ProviderIds": {"Tmdb": "1396"},
|
|
"Path": "/media/绝命毒师/Season 1",
|
|
},
|
|
{
|
|
"Id": "series-2",
|
|
"Type": "Series",
|
|
"Name": "绝命毒师 第 2 季",
|
|
"ProductionYear": 2009,
|
|
"ProviderIds": {"Tmdb": "1396"},
|
|
"Path": "/media/绝命毒师/Season 2",
|
|
},
|
|
],
|
|
"TotalRecordCount": 2,
|
|
}),
|
|
]
|
|
|
|
with patch("app.modules.zspace.zspace.RequestUtils") as request_utils_cls:
|
|
request_utils_cls.return_value.get_res.side_effect = responses
|
|
items = list(client.get_items(parent="library-id"))
|
|
|
|
assert [item.item_id for item in items] == ["series-1", "series-2"]
|
|
assert [item.tmdbid for item in items] == [1396, 1396]
|
|
|
|
|
|
def test_get_items_loads_detail_when_list_metadata_is_incomplete() -> None:
|
|
"""列表项缺少关键元数据时应使用详情接口补全。"""
|
|
client = _build_client()
|
|
responses = [
|
|
_FakeResponse({
|
|
"Items": [
|
|
{
|
|
"Id": "movie-1",
|
|
"Type": "Movie",
|
|
"Name": "疯狂动物城",
|
|
"ProductionYear": None,
|
|
"ProviderIds": {},
|
|
"Path": "",
|
|
}
|
|
],
|
|
"TotalRecordCount": 1,
|
|
}),
|
|
_FakeResponse({
|
|
"Id": "movie-1",
|
|
"ParentId": "library-id",
|
|
"Type": "Movie",
|
|
"Name": "疯狂动物城",
|
|
"OriginalTitle": "Zootopia",
|
|
"ProductionYear": 2016,
|
|
"ProviderIds": {
|
|
"Tmdb": "269149",
|
|
"Imdb": "tt2948356",
|
|
},
|
|
"Path": "/media/疯狂动物城 (2016)/疯狂动物城.mkv",
|
|
}),
|
|
]
|
|
|
|
with patch("app.modules.zspace.zspace.RequestUtils") as request_utils_cls:
|
|
request_utils_cls.return_value.get_res.side_effect = responses
|
|
items = list(client.get_items(parent="library-id"))
|
|
|
|
assert len(items) == 1
|
|
assert items[0].item_id == "movie-1"
|
|
assert items[0].tmdbid == 269149
|
|
assert items[0].imdbid == "tt2948356"
|
|
assert items[0].year == 2016
|
|
assert items[0].path.endswith("疯狂动物城.mkv")
|
|
assert request_utils_cls.return_value.get_res.call_args_list[1].args[0] == (
|
|
"http://zspace.local/emby/Users/user-id/Items/movie-1"
|
|
)
|
|
|
|
|
|
def test_get_items_uses_total_count_when_server_returns_short_pages() -> None:
|
|
"""服务端单页少于请求数量时应以总记录数决定是否继续翻页。"""
|
|
client = _build_client()
|
|
responses = [
|
|
_FakeResponse({
|
|
"Items": [
|
|
{
|
|
"Id": "movie-1",
|
|
"Type": "Movie",
|
|
"Name": "电影一",
|
|
"ProductionYear": 2025,
|
|
"ProviderIds": {"Tmdb": "101"},
|
|
"Path": "/media/movie-1.mkv",
|
|
}
|
|
],
|
|
"TotalRecordCount": 2,
|
|
}),
|
|
_FakeResponse({
|
|
"Items": [
|
|
{
|
|
"Id": "movie-2",
|
|
"Type": "Movie",
|
|
"Name": "电影二",
|
|
"ProductionYear": 2024,
|
|
"ProviderIds": {"Tmdb": "102"},
|
|
"Path": "/media/movie-2.mkv",
|
|
}
|
|
],
|
|
"TotalRecordCount": 2,
|
|
}),
|
|
]
|
|
|
|
with patch("app.modules.zspace.zspace.DEFAULT_ITEMS_PAGE_SIZE", 100), patch(
|
|
"app.modules.zspace.zspace.RequestUtils"
|
|
) as request_utils_cls:
|
|
request_utils_cls.return_value.get_res.side_effect = responses
|
|
items = list(client.get_items(parent="library-id"))
|
|
|
|
assert [item.item_id for item in items] == ["movie-1", "movie-2"]
|
|
calls = request_utils_cls.return_value.get_res.call_args_list
|
|
assert calls[0].kwargs["params"]["StartIndex"] == 0
|
|
assert calls[1].kwargs["params"]["StartIndex"] == 1
|