def test_extract_single_base(self, img_mock): img_mock.get_image.return_value = MockClientResponse(link=self.url_extract_dict['SINGLE']) ie = ImgurExtractor(self.get_single_post(), MockObjects.get_blank_user()) ie.extract_single() content = ie.extracted_content[0] self.assertEqual(self.url_extract_dict['SINGLE'], content.url) self.assertEqual('Picture(s)', content.post_title) self.assertEqual('Pics', content.subreddit) self.assertEqual(1521473630, content.date_created) self.assertEqual('C:/Users/Gorgoth/Downloads/JohnEveryman/fb2yRj0.jpg', content.filename) self.assertTrue(len(ie.failed_extract_posts) == 0)
def test_extract_single(self, img_mock): post = MockObjects.get_generic_mock_post() post.url = 'https://imgur.com/fb2yRj0' img_mock.return_value = 'https://i.imgur.com/fb2yRj0.jpg' ie = ImgurExtractor(post, MockObjects.get_blank_user()) ie.extract_single() content = ie.extracted_content[0] img_mock.assert_called_with('fb2yRj0') self.assertEqual(self.url_extract_dict['SINGLE'], content.url) self.assertEqual('Picture(s)', content.post_title) self.assertEqual('Pics', content.subreddit) self.assertEqual(1521473630, content.date_created) self.assertEqual('C:/Users/Gorgoth/Downloads/JohnEveryman/fb2yRj0.jpg', content.make_filename()) self.assertTrue(len(ie.failed_extract_posts) == 0)