예제 #1
0
    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)