def test_get_links_no_hub(self):
with open(TEST_FILES_DIR + 'rss.xml') as f:
data = f.read()
p = feedparser.parse(data)
links = get_links(p)
self.assertIsNone(links[0])
self.assertEqual(links[1], 'https://davidbeath.com/rss.xml')
def test_get_links(self):
with open(TEST_FILES_DIR + 'superfeedrblog.xml') as f:
data = f.read()
p = feedparser.parse(data)
links = get_links(p)
self.assertEqual(links[0], 'http://pubsubhubbub.superfeedr.com/')
self.assertEqual(links[1],
'https://superfeedr-blog-feed.herokuapp.com/')
def test_get_links_no_feed(self):
links = get_links('Invalid data')
self.assertIsNone(links)
