class FeedConsumerTestCase(unittest.TestCase): def setUp(self): self.rss_url = "http://www.wdcdn.net/rss/presentation/library/client/iowa/id/128b053b916ea1f7f20233e8a26bc45d" self.consumer = FeedConsumer(self.rss_url) @mock.patch.object(Image, 'get_smallest_image') @mock.patch.object(Video, 'get_codec') @mock.patch.object(Video, 'get_bitrate') @mock.patch.object(Video, 'get_duration') def test_parse(self, m_gd, m_gb, m_gc, m_gsi): m_gd.return_value = 230 m_gb.return_value = 1231231 m_gc.return_value = 'quicktime/mov' m_gsi.return_value = '/image1.jpg' videos = self.consumer.parse() self.assertEqual(len(videos), 15) self.assertEqual(videos[0].get('thumbnail'), '/image1.jpg') self.assertEqual(videos[0].get('duration'), 230) self.assertEqual(videos[0].get('bitrate'), 1231231) self.assertEqual(videos[0].get('video_codec'), 'quicktime/mov') def test_bad_url(self): bad_consumer = FeedConsumer('http://www.bonk.net') videos = bad_consumer.parse() self.assertEqual(len(videos), 0)
def lambda_handler(event, context): consumer = FeedConsumer( "http://www.wdcdn.net/rss/presentation/library/client/iowa" "/id/128b053b916ea1f7f20233e8a26bc45d" ) videos = consumer.parse() return {"payload": {"videos": videos}}
def test_bad_url(self): bad_consumer = FeedConsumer('http://www.bonk.net') videos = bad_consumer.parse() self.assertEqual(len(videos), 0)