From 40f6ee60d4f340f007038e9f9e3feec23200c81a Mon Sep 17 00:00:00 2001 From: Simon Date: Thu, 17 Jul 2025 20:51:59 +0700 Subject: [PATCH] fix unknown vid_type query building --- backend/channel/src/remote_query.py | 2 +- backend/channel/tests/test_src/test_remote_query.py | 11 +++++++++++ 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/backend/channel/src/remote_query.py b/backend/channel/src/remote_query.py index 90cbde22..52104fce 100644 --- a/backend/channel/src/remote_query.py +++ b/backend/channel/src/remote_query.py @@ -23,7 +23,7 @@ class VideoQueryBuilder: VideoTypeEnum.SHORTS: self.shorts_query, } - if video_type: + if video_type and video_type != VideoTypeEnum.UNKNOWN: # build query for specific type/s if not isinstance(video_type, list): video_type = [video_type] diff --git a/backend/channel/tests/test_src/test_remote_query.py b/backend/channel/tests/test_src/test_remote_query.py index a123d75c..470b50d9 100644 --- a/backend/channel/tests/test_src/test_remote_query.py +++ b/backend/channel/tests/test_src/test_remote_query.py @@ -77,6 +77,17 @@ def test_build_multiple_queries(default_config, empty_overwrites): assert result == [(VideoTypeEnum.VIDEOS, 5), (VideoTypeEnum.SHORTS, 2)] +def test_build_unknown_queries(default_config, empty_overwrites): + """vid_type unknown""" + builder = VideoQueryBuilder(default_config, empty_overwrites) + result = builder.build_queries(VideoTypeEnum.UNKNOWN) + assert result == [ + (VideoTypeEnum.VIDEOS, 5), + (VideoTypeEnum.STREAMS, 3), + (VideoTypeEnum.SHORTS, 2), + ] + + def test_overwrite_applied(default_config, overwrites): """with overwrite from channel config""" builder = VideoQueryBuilder(default_config, overwrites)