File size: 2,134 Bytes
f293f76 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
from unittest import mock
from pytube import Channel
@mock.patch('pytube.request.get')
def test_init_with_url(request_get, channel_videos_html):
request_get.return_value = channel_videos_html
c = Channel('https://www.youtube.com/c/ProgrammingKnowledge/videos')
assert c.channel_url == 'https://www.youtube.com/c/ProgrammingKnowledge'
assert c.videos_url == f'{c.channel_url}/videos'
assert c.playlists_url == f'{c.channel_url}/playlists'
assert c.community_url == f'{c.channel_url}/community'
assert c.featured_channels_url == f'{c.channel_url}/channels'
assert c.about_url == f'{c.channel_url}/about'
@mock.patch('pytube.request.get')
def test_channel_name(request_get, channel_videos_html):
request_get.return_value = channel_videos_html
c = Channel('https://www.youtube.com/c/ProgrammingKnowledge/videos')
assert c.channel_name == 'ProgrammingKnowledge'
@mock.patch('pytube.request.get')
def test_channel_video_list(request_get, channel_videos_html):
request_get.return_value = channel_videos_html
c = Channel('https://www.youtube.com/c/ProgrammingKnowledge/videos')
first_ten = [
'https://www.youtube.com/watch?v=t_xLpJo_35k',
'https://www.youtube.com/watch?v=ccbh5YhxouQ',
'https://www.youtube.com/watch?v=wDnFjDjxW_0',
'https://www.youtube.com/watch?v=F3W_p_4XftA',
'https://www.youtube.com/watch?v=_fxm0xGGEi4',
'https://www.youtube.com/watch?v=cRbKZzcuIsg',
'https://www.youtube.com/watch?v=sdDu3dfIuow',
'https://www.youtube.com/watch?v=10KIbp-gJCE',
'https://www.youtube.com/watch?v=wZIT-cRtd6s',
'https://www.youtube.com/watch?v=KucCvEbTj0w',
]
assert c.video_urls[:10] == first_ten
@mock.patch('pytube.request.get')
def test_videos_html(request_get, channel_videos_html):
request_get.return_value = channel_videos_html
c = Channel('https://www.youtube.com/c/ProgrammingKnowledge')
assert c.html == channel_videos_html
# Because the Channel object subclasses the Playlist object, most of the tests
# are already taken care of by the Playlist test suite.
|