def test_assign_extractor_vidble(self):
ex = Extractor.assign_extractor(MockObjects.get_mock_post_vidble())
self.assertEqual('VidbleExtractor', ex.__name__)
def test_assign_extractor_imgur(self):
ex = Extractor.assign_extractor(MockObjects.get_mock_post_imgur())
self.assertEqual('ImgurExtractor', ex.__name__)
def test_assign_extractor_gfycat(self):
ex = Extractor.assign_extractor(MockObjects.get_mock_post_gfycat())
self.assertEqual('GfycatExtractor', ex.__name__)
def test_assign_extractor_direct(self):
ex = Extractor.assign_extractor(
MockObjects.get_unsupported_direct_post())
self.assertEqual('DirectExtractor', ex.__name__)