def test_set_values(self, mock_get_video_info):
video_info = google.VideoInfo('test-channel-id', 'title',
'description', 100,
'http://example.com/thumb.png')
mock_get_video_info.return_value = video_info
vt = YoutubeVideoType('http://www.youtube.com/watch?v=_ShmidkrcY0')
video = Video()
vt.set_values(video)
self.assertEqual(video.title, video_info.title)
self.assertEqual(video.description, video_info.description)
self.assertEqual(video.duration, video_info.duration)
self.assertEqual(video.thumbnail, video_info.thumbnail_url)
def test_get_video_info_exception(self, mock_get_video_info):
video_info = google.VideoInfo('test-channel-id', 'title',
'description', 100,
'http://example.com/thumb.png')
mock_get_video_info.side_effect = google.APIError()
vt = YoutubeVideoType('http://www.youtube.com/watch?v=_ShmidkrcY0')
video = Video()
vt.set_values(video, None, None, None)
self.assertEqual(vt.video_id, '_ShmidkrcY0')
self.assertEqual(video.description, '')
self.assertEqual(video.duration, None)
self.assertEqual(video.thumbnail, '')
def test_get_video_info_exception(self, mock_get_video_info):
video_info = google.VideoInfo('test-channel-id', 'title',
'description', 100,
'http://example.com/thumb.png')
mock_get_video_info.side_effect = google.APIError()
vt = YoutubeVideoType('http://www.youtube.com/watch?v=_ShmidkrcY0')
video = Video()
vt.set_values(video)
self.assertEqual(vt.video_id, '_ShmidkrcY0')
self.assertEqual(video.description, '')
self.assertEqual(video.duration, None)
self.assertEqual(video.thumbnail, '')
# since get_video_info failed, we don't know the channel id of our
# video URL. We should use a dummy value to make it easier to fix the
# issue in the future
self.assertEqual(vt.owner_username(), None)
