def test_timeout_dict(self, sleep_mock, ex_mock, extractor_mock):
extractor_mock.assign_extractor.return_value = ex_mock
ExtractorUtils.time_limit_dict[
'ImgurExtractor'] = Const.TIMEOUT_INCREMENT
ExtractorUtils.timeout_dict['ImgurExtractor'] = time.time()
extractor = Extractor(MockObjects.get_user_with_single_content())
extractor.extract(MockObjects.get_mock_post_imgur())
sleep_mock.assert_called()
def test_assign_extractor_imgur(self):
ex = Extractor.assign_extractor(MockObjects.get_mock_post_imgur())
self.assertEqual('ImgurExtractor', ex.__name__)
