mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-30 04:27:40 +08:00
97 lines
3.2 KiB
Python
97 lines
3.2 KiB
Python
from types import SimpleNamespace
|
|
from unittest.mock import Mock, patch
|
|
|
|
from app.application.subscription.query import SubscriptionQueryService
|
|
from app.chain.subscribe import SubscribeChain
|
|
from app.domain.context import MediaInfo
|
|
from app.schemas.types import MediaSource, MediaType
|
|
|
|
|
|
def test_subscription_query_service_builds_complete_exists_identity() -> None:
|
|
"""存在性查询必须保留媒体、音乐实体、季和剧集组全部身份维度。"""
|
|
repository = Mock()
|
|
repository.exists.return_value = True
|
|
service = SubscriptionQueryService(repository)
|
|
media = MediaInfo(
|
|
type=MediaType.TV,
|
|
title="Demo",
|
|
media_source=MediaSource.TMDB,
|
|
media_id="123",
|
|
episode_group="group-1",
|
|
)
|
|
|
|
assert service.exists(media, SimpleNamespace(begin_season=2)) is True
|
|
repository.exists.assert_called_once_with(
|
|
media_source=MediaSource.TMDB,
|
|
media_id="123",
|
|
music_type=None,
|
|
season=2,
|
|
episode_group="group-1",
|
|
)
|
|
|
|
|
|
def test_subscription_query_service_filters_source_and_music_state() -> None:
|
|
"""来源查询不透传展示字段，音乐状态查询保持 R/P 联合列表语义。"""
|
|
repository = Mock()
|
|
expected = SimpleNamespace(id=1)
|
|
repository.get_by.return_value = expected
|
|
repository.list.return_value = [
|
|
SimpleNamespace(type=MediaType.MOVIE.value),
|
|
SimpleNamespace(type=MediaType.MUSIC.value),
|
|
]
|
|
service = SubscriptionQueryService(repository)
|
|
|
|
result = service.get_by_source({
|
|
"id": 1,
|
|
"name": "Demo",
|
|
"type": MediaType.TV.value,
|
|
"season": 1,
|
|
"media_source": MediaSource.TMDB,
|
|
"media_id": "123",
|
|
"music_type": None,
|
|
})
|
|
|
|
assert result is expected
|
|
repository.get_by.assert_called_once_with(
|
|
type=MediaType.TV.value,
|
|
season=1,
|
|
media_source=MediaSource.TMDB,
|
|
media_id="123",
|
|
music_type=None,
|
|
)
|
|
assert service.has_music("R,P") is True
|
|
repository.list.assert_called_once_with("R,P")
|
|
|
|
|
|
def test_subscribe_chain_facade_delegates_three_query_slices() -> None:
|
|
"""SubscribeChain 保持三个公开方法签名并仅负责来源解析和结果转发。"""
|
|
service = Mock()
|
|
service.exists.return_value = True
|
|
service.get_by_source.return_value = SimpleNamespace(id=7)
|
|
service.has_music.return_value = True
|
|
media = MediaInfo(
|
|
type=MediaType.MOVIE,
|
|
title="Demo",
|
|
media_source=MediaSource.TMDB,
|
|
media_id="123",
|
|
)
|
|
source = (
|
|
'Subscribe|{"type":"电影","season":null,'
|
|
'"media_source":"themoviedb","media_id":"123"}'
|
|
)
|
|
|
|
with patch.object(SubscribeChain, "_subscription_query", return_value=service):
|
|
chain = object.__new__(SubscribeChain)
|
|
assert chain.exists(media) is True
|
|
assert chain.get_subscribe_by_source(source).id == 7
|
|
assert chain.has_music_subscribe() is True
|
|
|
|
service.exists.assert_called_once_with(media, None)
|
|
service.get_by_source.assert_called_once_with({
|
|
"type": "电影",
|
|
"season": None,
|
|
"media_source": "themoviedb",
|
|
"media_id": "123",
|
|
})
|
|
service.has_music.assert_called_once_with("R,P")
|