class TestCraigsList(unittest.TestCase): def setUp(self): self.craigs_list = CraigsList() def get_test_file_to_soup(self, file): cl_test_file = open(file) cl_test_soup = BeautifulSoup(cl_test_file.read()) cl_test_file.close() return cl_test_soup def test_get_links(self): cl_list_soup = self.get_test_file_to_soup('cl_list_test.html') test_date = datetime.date(2010, 9, 11) todays_links = self.craigs_list._get_days_links(cl_list_soup, test_date) self.assertEqual(todays_links[0].__str__(), '<a href="http://newyork.craigslist.org/mnh/sof/1950616953.html">Software Engineer (Hadoop)</a>') def test_parse_content(self): cl_post_soup = self.get_test_file_to_soup('cl_good_job.html') content = self.craigs_list.parse_content(cl_post_soup) self.assertEqual(content, good_post_content)
def setUp(self): self.craigs_list = CraigsList()