def test_video_id(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') self.assertEqual(vt.video_id, '_ShmidkrcY0')
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_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 video, created = Video.get_or_create_for_url( 'http://www.youtube.com/watch?v=_ShmidkrcY0') vu = video.videourl_set.all()[:1].get() self.assertEqual(vu.videoid, '_ShmidkrcY0') 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) 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)
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() video, created = Video.get_or_create_for_url( 'http://www.youtube.com/watch?v=_ShmidkrcY0') vu = video.videourl_set.all()[:1].get() self.assertEqual(vu.videoid, '_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(vu.owner_username, None)