From 5efd6ff8f2bf3a17e218a6ea6b14ff88decfaa8a Mon Sep 17 00:00:00 2001 From: Simon Date: Sat, 27 Sep 2025 15:20:54 +0700 Subject: [PATCH] fix unknown type query building --- backend/channel/src/remote_query.py | 4 ++-- backend/channel/tests/test_src/test_remote_query.py | 11 +++++++++++ 2 files changed, 13 insertions(+), 2 deletions(-) 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)