diff --git a/backend/channel/src/remote_query.py b/backend/channel/src/remote_query.py index 2a49a0b7..71049757 100644 --- a/backend/channel/src/remote_query.py +++ b/backend/channel/src/remote_query.py @@ -102,7 +102,7 @@ def get_last_channel_videos( builder = VideoQueryBuilder(config, limit=limit) queries = [] - if query_filter is None: + if query_filter is None or query_filter == VideoTypeEnum.UNKNOWN: queries = builder.build_queries() elif isinstance(query_filter, list): queries = builder.build_queries(vid_types=query_filter) @@ -113,7 +113,7 @@ def get_last_channel_videos( last_videos: list[dict] = [] - if not query: + if not queries: return last_videos for vid_type_enum, limit_amount in queries: diff --git a/backend/channel/tests/test_src/test_remote_query.py b/backend/channel/tests/test_src/test_remote_query.py index a680b2fb..ff3e32cf 100644 --- a/backend/channel/tests/test_src/test_remote_query.py +++ b/backend/channel/tests/test_src/test_remote_query.py @@ -68,6 +68,17 @@ def test_build_specific_query(default_config, empty_overwrites): assert result == (VideoTypeEnum.VIDEOS, 5) +def test_build_unknown_type(default_config, empty_overwrites): + """unknown vid_type build list""" + builder = VideoQueryBuilder(default_config, empty_overwrites, limit=None) + result = builder.build_queries() + assert result == [ + (VideoTypeEnum.VIDEOS, None), + (VideoTypeEnum.STREAMS, None), + (VideoTypeEnum.SHORTS, None), + ] + + def test_build_multiple_queries(default_config, empty_overwrites): """vid_type list""" builder = VideoQueryBuilder(default_config, empty_overwrites)