[youtube:search] Add tests

Rework 2c2c2bd with an actual Mix page and realistic playlist size
From 2c2c2bd348 (commitcomment-65953545)
2024-12-19 23:22:04 +00:00 · 2022-02-04 11:09:18 +00:00 · 2022-02-04 04:09:23 +00:00
2 changed files with 21 additions and 6 deletions
--- a/test/test_youtube_lists.py
+++ b/test/test_youtube_lists.py
@ -36,12 +36,12 @@ class TestYoutubeLists(unittest.TestCase):
        dl = FakeYDL()
        dl.params['format'] = 'best'
        ie = YoutubeTabIE(dl)
-        result = dl.extract_info('https://www.youtube.com/watch?v=uVJ0Il5WvbE&list=PLhQjrBD2T381k8ul4WQ8SQ165XqY149WW',
+        result = dl.extract_info('https://www.youtube.com/watch?v=tyITL_exICo&list=RDCLAK5uy_kLWIr9gv1XLlPbaDS965-Db4TrBoUTxQ8',
                                 download=False, ie_key=ie.ie_key(), process=True)
        entries = (result or {}).get('entries', [{'id': 'not_found', }])
-        self.assertTrue(len(entries) >= 50)
+        self.assertTrue(len(entries) >= 25)
        original_video = entries[0]
-        self.assertEqual(original_video['id'], 'uVJ0Il5WvbE')
+        self.assertEqual(original_video['id'], 'tyITL_exICo')

    def test_youtube_flat_playlist_extraction(self):
        dl = FakeYDL()
--- a/youtube_dl/extractor/youtube.py
+++ b/youtube_dl/extractor/youtube.py
@ -3206,7 +3206,14 @@ class YoutubeSearchIE(SearchInfoExtractor, YoutubeBaseInfoExtractor):
    _SEARCH_KEY = 'ytsearch'
    _SEARCH_PARAMS = 'EgIQAQ%3D%3D'  # Videos only
    _MAX_RESULTS = float('inf')
-    _TESTS = []
+    _TESTS = [{
+        'url': 'ytsearch10:youtube-dl test video',
+        'playlist_count': 10,
+        'info_dict': {
+            'id': 'youtube-dl test video',
+            'title': 'youtube-dl test video',
+        }
+    }]

    def _get_n_results(self, query, n):
        """Get a specified number of results for a query"""
@ -3219,7 +3226,14 @@ class YoutubeSearchDateIE(YoutubeSearchIE):
    _SEARCH_KEY = 'ytsearchdate'
    IE_DESC = 'YouTube.com searches, newest videos first'
    _SEARCH_PARAMS = 'CAISAhAB'  # Videos only, sorted by date
-    _TESTS = []
+    _TESTS = [{
+        'url': 'ytsearchdate10:youtube-dl test video',
+        'playlist_count': 10,
+        'info_dict': {
+            'id': 'youtube-dl test video',
+            'title': 'youtube-dl test video',
+        }
+    }]


 class YoutubeSearchURLIE(YoutubeBaseInfoExtractor):
@ -3232,7 +3246,8 @@ class YoutubeSearchURLIE(YoutubeBaseInfoExtractor):
        'info_dict': {
            'id': 'youtube-dl test video',
            'title': 'youtube-dl test video',
-        }
+        },
+        'params': {'playlistend': 5}
    }, {
        'url': 'https://www.youtube.com/results?q=test&sp=EgQIBBgB',
        'only_matching': True,
Author	SHA1	Message	Date
pukkandan	0c0876f790	[youtube:search] Add tests	2022-02-04 11:09:18 +00:00
dirkf	7a497f1405	Rework `2c2c2bd` with an actual Mix page and realistic playlist size From `2c2c2bd348 (commitcomment-65953545)`	2022-02-04 04:09:23 +00:00