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()
