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 setUp(self):
self.rss_url = "http://www.wdcdn.net/rss/presentation/library/client/iowa/id/128b053b916ea1f7f20233e8a26bc45d"
self.consumer = FeedConsumer(self.rss_url)
def test_bad_url(self):
bad_consumer = FeedConsumer('http://www.bonk.net')
videos = bad_consumer.parse()
self.assertEqual(len(videos), 0)
