mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-21 00:07:32 +08:00
120 lines
3.9 KiB
Python
120 lines
3.9 KiB
Python
from app.modules.indexer.spider import (
|
|
SiteSpider,
|
|
resolve_category_media_type,
|
|
select_media_categories,
|
|
)
|
|
from app.schemas.types import MediaType
|
|
|
|
|
|
def _category_config() -> dict:
|
|
"""构造包含电影、电视剧和音乐分类的站点配置。"""
|
|
return {
|
|
"movie": [{"id": "1", "name": "电影"}],
|
|
"tv": [{"id": "2", "name": "电视剧"}],
|
|
"music": [{"id": "3", "name": "音乐"}],
|
|
}
|
|
|
|
|
|
def test_select_media_categories_supports_music():
|
|
"""站点搜索限定音乐类型时应只提交音乐分类。"""
|
|
assert select_media_categories(_category_config(), MediaType.MUSIC) == [
|
|
{"id": "3", "name": "音乐"}
|
|
]
|
|
|
|
|
|
def test_select_media_categories_includes_music_when_type_unspecified():
|
|
"""未限定媒体类型的浏览应继续包含全部已配置分类。"""
|
|
assert [item["id"] for item in select_media_categories(_category_config(), None)] == [
|
|
"1",
|
|
"2",
|
|
"3",
|
|
]
|
|
|
|
|
|
def test_resolve_category_media_type_supports_music():
|
|
"""音乐分类 ID 应映射为统一音乐媒体类型。"""
|
|
assert resolve_category_media_type("3", _category_config()) == MediaType.MUSIC
|
|
|
|
|
|
def test_resolve_category_media_type_rejects_ambiguous_category():
|
|
"""一个分类同时归属多个媒体类型时应保持未知，避免误整理。"""
|
|
category = _category_config()
|
|
category["movie"].append({"id": "3", "name": "综合"})
|
|
|
|
assert resolve_category_media_type("3", category) == MediaType.UNKNOWN
|
|
|
|
|
|
def test_site_level_music_type_fills_missing_torrent_category():
|
|
"""音乐专属站点缺少逐条分类字段时应使用站点级音乐类型。"""
|
|
spider = SiteSpider(
|
|
indexer={
|
|
"id": "music",
|
|
"name": "Music",
|
|
"domain": "https://music.example/",
|
|
"media_type": "music",
|
|
"search": {},
|
|
"torrents": {"fields": {}},
|
|
},
|
|
mtype=MediaType.MUSIC,
|
|
)
|
|
|
|
spider._SiteSpider__get_category(None)
|
|
|
|
assert spider.torrents_info["category"] == MediaType.MUSIC.value
|
|
|
|
|
|
def test_requested_result_media_type_overrides_unrepresentable_site_category():
|
|
"""音乐专属查询使用非主分类筛选时，应按显式契约把结果标记为音乐。"""
|
|
spider = SiteSpider(
|
|
indexer={
|
|
"id": "typed-music",
|
|
"name": "Typed Music",
|
|
"domain": "https://music.example/",
|
|
"search": {
|
|
"paths": [
|
|
{"path": "torrents?q={keyword}", "type": "all"},
|
|
{"path": "music?q={keyword}", "type": "music"},
|
|
],
|
|
"result_media_type": "requested",
|
|
},
|
|
"category": {
|
|
"movie": [{"id": "1", "name": "电影"}],
|
|
"music": [{"id": "20", "name": "音乐规格"}],
|
|
},
|
|
"torrents": {
|
|
"fields": {
|
|
"category": {"selector": "a.category"},
|
|
}
|
|
},
|
|
},
|
|
mtype=MediaType.MUSIC,
|
|
)
|
|
|
|
search_url = spider._SiteSpider__get_search_url()
|
|
spider._SiteSpider__get_category(None)
|
|
|
|
assert search_url == "https://music.example/music?q="
|
|
assert spider.torrents_info["category"] == MediaType.MUSIC.value
|
|
|
|
|
|
def test_requested_result_media_type_requires_an_active_type_filter():
|
|
"""未命中专属路径或分类参数时，不得把混合搜索结果强制标记为请求类型。"""
|
|
spider = SiteSpider(
|
|
indexer={
|
|
"id": "mixed",
|
|
"name": "Mixed",
|
|
"domain": "https://mixed.example/",
|
|
"search": {
|
|
"paths": [{"path": "torrents?q={keyword}", "type": "all"}],
|
|
"result_media_type": "requested",
|
|
},
|
|
"torrents": {"fields": {}},
|
|
},
|
|
mtype=MediaType.MOVIE,
|
|
)
|
|
|
|
spider._SiteSpider__get_search_url()
|
|
spider._SiteSpider__get_category(None)
|
|
|
|
assert "category" not in spider.torrents_info
|