def test_mark_visited(self): url_db=url_saver.url_saver_mysql() url_db.add("http://www.sina.com/2008/","task1") url_db.add("http://www.sohu.com/3.html","task5") url_db.add("http://www.sohu.com/4.html","task5") url_db.mark_error("http://www.sohu.com/2.html","task5") url_db.mark_visited("http://www.sohu.com/3.html","task5") self.assertEqual(url_db.pop("task5"),"http://www.sohu.com/4.html") self.assertEqual("http://www.sina.com/2008/",url_db.pop("task1"))
def test_save(self): ''' test if the url saver works as expected''' url_db=url_saver.url_saver_mysql() url_db.add("http://www.sohu.com/","task3") self.assertEqual(url_db.pop("task3"),"http://www.sohu.com/")
def test_mark_error(self): url_db=url_saver.url_saver_mysql() url_db.add("http://www.sohu.com/1.html","task4") url_db.add("http://www.sohu.com/2.html","task4") url_db.mark_error("http://www.sohu.com/1.html","task4") self.assertEqual(url_db.pop("task4"),"http://www.sohu.com/2.html")