def test_extract_video_with_audio(self): post = MockObjects.get_mock_post_reddit_video() post.is_video = True fallback_url = post.url + '/DASH_2_4_M?source=fallback' post.media = {'reddit_video': {'fallback_url': fallback_url}} re = RedditVideoExtractor(post, MockObjects.get_blank_user()) re.extract_content() self.assertEqual(2, len(re.extracted_content)) vid_content = re.extracted_content[0] self.assertEqual(fallback_url, vid_content.url) self.assertEqual('PublicFreakout', vid_content.subreddit) self.assertEqual( 'C:/Users/Gorgoth/Downloads/JohnEveryman/abcde(video).mp4', vid_content.filename) audio_content = re.extracted_content[1] self.assertEqual(post.url + '/audio', audio_content.url) self.assertEqual('PublicFreakout', audio_content.subreddit) self.assertEqual( 'C:/Users/Gorgoth/Downloads/JohnEveryman/abcde(audio).mp3', audio_content.filename) self.assertEqual(0, len(re.failed_extract_posts)) self.assertEqual(1, len(VideoMerger.videos_to_merge))
def test_extract_video_failed_to_find_url(self, rv_mock): rv_mock.return_value = None post = MockObjects.get_mock_post_reddit_video() post.is_video = True re = RedditVideoExtractor(post, MockObjects.get_blank_user()) re.extract_content() self.assertEqual(1, len(re.failed_extract_posts)) self.assertEqual(0, len(re.extracted_content))
def test_extract_video_with_audio_crossposted_post(self, rv_mock): parent_post = MockObjects.get_mock_post_reddit_video() parent_id = 'pppppp' parent_post.id = parent_id parent_post.is_video = True parent_post.title = 'Whoa, Parent Vid' parent_post.subreddit = 'DefinitelyNotPublicFreakout' fallback_url = parent_post.url + '/DASH_2_4_M?source=fallback' parent_post.media = {'reddit_video': {'fallback_url': fallback_url}} rv_mock.return_value = parent_post post = MockObjects.get_mock_post_reddit_video() post.crosspost_parent = 'notsurewhatgoeshere_' + parent_id post.url = 'https://v.redd.it/nottherealurl' re = RedditVideoExtractor(post, MockObjects.get_blank_user()) re.extract_content() rv_mock.assert_called() self.assertEqual(2, (len(re.extracted_content))) vid_content = re.extracted_content[0] self.assertEqual(fallback_url, vid_content.url) self.assertEqual('PublicFreakout', vid_content.subreddit) self.assertEqual( 'C:/Users/Gorgoth/Downloads/JohnEveryman/abcde(video).mp4', vid_content.filename) audio_content = re.extracted_content[1] self.assertEqual(parent_post.url + '/audio', audio_content.url) self.assertEqual('PublicFreakout', audio_content.subreddit) self.assertEqual( 'C:/Users/Gorgoth/Downloads/JohnEveryman/abcde(audio).mp3', audio_content.filename) self.assertEqual(0, len(re.failed_extract_posts)) self.assertEqual(1, len(VideoMerger.videos_to_merge))
def test_extract_gif(self): post = MockObjects.get_mock_post_reddit_video() fallback_url = post.url + '/DASH_2_4_M?source=fallback' post.media = {'reddit_video': {'fallback_url': fallback_url}} re = RedditVideoExtractor(post, MockObjects.get_blank_user()) re.extract_content() self.assertEqual(1, len(re.extracted_content)) content = re.extracted_content[0] self.assertEqual(fallback_url, content.url) self.assertEqual('PublicFreakout', content.subreddit) self.assertEqual('Reddit Video Broh', content.post_title) self.assertEqual('C:/Users/Gorgoth/Downloads/JohnEveryman/abcde.mp4', content.filename) self.assertEqual(0, len(re.failed_extract_posts)) self.assertEqual(0, len(VideoMerger.videos_to_merge))