class TestURLFrontier(TestCase):
def setUp(self):
links = Links("data/test_seeds")
self.url_frontier = URLFrontier(links)
def test_pop_url(self):
self.assertEqual(self.url_frontier.pop_url(), "http://www.lonelyplanet.com/vietnam/transport/getting-around/local-transport")
self.assertEqual(self.url_frontier.pop_url(), "http://www.vietnamtravel.org/rail-travel-train-timetables")
def setUp(self):
links = Links("data/test_seeds")
self.url_frontier = URLFrontier(links)