diff --git a/app/chain/subscribe.py b/app/chain/subscribe.py index 031879211..0b9307e44 100644 --- a/app/chain/subscribe.py +++ b/app/chain/subscribe.py @@ -3157,6 +3157,7 @@ class SubscribeChain(ChainBase): 'year': subscribe.year, 'type': subscribe.type, 'season': subscribe.season, + 'episode_group': subscribe.episode_group, 'tmdbid': subscribe.tmdbid, 'imdbid': subscribe.imdbid, 'tvdbid': subscribe.tvdbid, diff --git a/tests/test_subscribe_source_keyword.py b/tests/test_subscribe_source_keyword.py new file mode 100644 index 000000000..812376ffc --- /dev/null +++ b/tests/test_subscribe_source_keyword.py @@ -0,0 +1,27 @@ +import json +from types import SimpleNamespace + +from app.chain.subscribe import SubscribeChain +from app.schemas.types import MediaType + + +def test_subscribe_source_keyword_includes_episode_group(): + subscribe = SimpleNamespace( + id=1, + name="Test Show", + year="2026", + type=MediaType.TV.value, + season=1, + episode_group="group-season-03", + tmdbid=12345, + imdbid=None, + tvdbid=None, + doubanid=None, + bangumiid=None, + ) + + source = SubscribeChain.get_subscribe_source_keyword(subscribe) + payload = json.loads(source.split("|", 1)[1]) + + assert payload["episode_group"] == "group-season-03" + assert SubscribeChain.parse_subscribe_source_keyword(source)["episode_group"] == "group-season-03"