def test_find_feeds__non_recursive(self): forager = Forager() with self.mocked_requests(forager): response = forager.find_feeds('http://site-a.com/') self.assertEqual(response, { 'http://site-a.com/feed/', 'http://site-a.com/comments/feed/' })
def test_find_feeds__depth_3(self): forager = Forager(max_depth=3) with self.mocked_requests(forager): response = forager.find_feeds('http://site-b.com/') self.assertEqual(response, { 'http://site-b.com/hidden_feed/', 'http://site-b.com/feed_like.xml', 'http://site-c.com/a_feed.xml' })
def test_find_feeds__fixed_urls(self): forager = Forager(max_depth=3) with self.mocked_requests(forager): response = forager.find_feeds('site-b.com') self.assertTrue(len(response) > 0)
def test_find_feeds__depth_1(self): forager = Forager(max_depth=1) with self.mocked_requests(forager): response = forager.find_feeds('http://site-b.com/') self.assertEqual(response, set())