[youtube:search] Add tests

This commit is contained in:
pukkandan 2022-02-03 07:44:37 +05:30 committed by dirkf
parent 7a497f1405
commit 0c0876f790

View file

@ -3206,7 +3206,14 @@ class YoutubeSearchIE(SearchInfoExtractor, YoutubeBaseInfoExtractor):
_SEARCH_KEY = 'ytsearch' _SEARCH_KEY = 'ytsearch'
_SEARCH_PARAMS = 'EgIQAQ%3D%3D' # Videos only _SEARCH_PARAMS = 'EgIQAQ%3D%3D' # Videos only
_MAX_RESULTS = float('inf') _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): def _get_n_results(self, query, n):
"""Get a specified number of results for a query""" """Get a specified number of results for a query"""
@ -3219,7 +3226,14 @@ class YoutubeSearchDateIE(YoutubeSearchIE):
_SEARCH_KEY = 'ytsearchdate' _SEARCH_KEY = 'ytsearchdate'
IE_DESC = 'YouTube.com searches, newest videos first' IE_DESC = 'YouTube.com searches, newest videos first'
_SEARCH_PARAMS = 'CAISAhAB' # Videos only, sorted by date _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): class YoutubeSearchURLIE(YoutubeBaseInfoExtractor):
@ -3232,7 +3246,8 @@ class YoutubeSearchURLIE(YoutubeBaseInfoExtractor):
'info_dict': { 'info_dict': {
'id': 'youtube-dl test video', 'id': 'youtube-dl test video',
'title': 'youtube-dl test video', 'title': 'youtube-dl test video',
} },
'params': {'playlistend': 5}
}, { }, {
'url': 'https://www.youtube.com/results?q=test&sp=EgQIBBgB', 'url': 'https://www.youtube.com/results?q=test&sp=EgQIBBgB',
'only_matching': True, 'only_matching': True,