mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-10-29 03:18:14 +08:00
88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
"""
|
|
搜索接口响应模型测试。
|
|
"""
|
|
|
|
from app.schemas.context import Context, MetaInfo, SubtitleInfo, TorrentInfo
|
|
from app.schemas.response import Response
|
|
from app.schemas.search import SearchLastContextData
|
|
|
|
|
|
def test_last_search_context_preserves_nested_torrent_context() -> None:
|
|
"""
|
|
最近搜索响应必须保留资源上下文的嵌套元数据和种子信息。
|
|
"""
|
|
context = Context(
|
|
meta_info=MetaInfo(title="Example", name="Example"),
|
|
torrent_info=TorrentInfo(title="Example torrent", site_name="Site A"),
|
|
)
|
|
|
|
response = Response[SearchLastContextData](
|
|
success=True,
|
|
data={
|
|
"params": {"keyword": "Example", "result_type": "torrent"},
|
|
"results": [context.model_dump(mode="json")],
|
|
},
|
|
).model_dump(mode="json")
|
|
|
|
result = response["data"]["results"][0]
|
|
assert result["meta_info"]["name"] == "Example"
|
|
assert result["torrent_info"]["title"] == "Example torrent"
|
|
assert result["torrent_info"]["site_name"] == "Site A"
|
|
|
|
|
|
def test_last_search_context_keeps_subtitle_result_shape() -> None:
|
|
"""
|
|
修正资源上下文类型后仍必须完整保留字幕搜索结果。
|
|
"""
|
|
subtitle = SubtitleInfo(
|
|
title="Example subtitle",
|
|
site_name="Subtitle Site",
|
|
enclosure="https://example.test/subtitle.srt",
|
|
)
|
|
|
|
response = Response[SearchLastContextData](
|
|
success=True,
|
|
data={
|
|
"params": {"keyword": "Example", "result_type": "subtitle"},
|
|
"results": [subtitle.model_dump(mode="json")],
|
|
},
|
|
).model_dump(mode="json")
|
|
|
|
result = response["data"]["results"][0]
|
|
assert result["title"] == "Example subtitle"
|
|
assert result["site_name"] == "Subtitle Site"
|
|
assert result["enclosure"] == "https://example.test/subtitle.srt"
|
|
|
|
|
|
def test_schema_metainfo_season_and_episode_properties() -> None:
|
|
"""
|
|
Schema MetaInfo 应具备与领域 MetaBase 一致的季、集属性和序列化兼容性。
|
|
"""
|
|
from app.schemas.types import MediaType
|
|
|
|
# 单季剧集（未显式指定季）
|
|
tv_no_season = MetaInfo(type="电视剧", begin_episode=1, end_episode=2)
|
|
assert tv_no_season.season_list == [1]
|
|
assert tv_no_season.season == "S01"
|
|
assert tv_no_season.season_seq == "1"
|
|
assert tv_no_season.episode == "E01-E02"
|
|
|
|
# 指定单季剧集
|
|
tv_single_season = MetaInfo(type=MediaType.TV, begin_season=2, begin_episode=5)
|
|
assert tv_single_season.season_list == [2]
|
|
assert tv_single_season.season == "S02"
|
|
assert tv_single_season.season_seq == "2"
|
|
assert tv_single_season.episode == "E05"
|
|
|
|
# 多季剧集
|
|
tv_multi_season = MetaInfo(type="电视剧", begin_season=1, end_season=3)
|
|
assert tv_multi_season.season_list == [1, 2, 3]
|
|
assert tv_multi_season.season == "S01-S03"
|
|
|
|
# 电影
|
|
movie = MetaInfo(type="电影")
|
|
assert movie.season_list == []
|
|
assert movie.season == ""
|
|
assert movie.season_seq == ""
|
|
assert movie.episode == ""
|