git.maemo.org Git - pywienerlinien/blob - tests.py

   1 import unittest
   2 from wlSearch import Search
   3 from datetime import datetime
   4 from BeautifulSoup import BeautifulSoup
   5
   6
   7 class FetchTests(unittest.TestCase):
   8
   9     def setUp(self):
  10         origin = 'Karlsplatz'
  11         destination = 'Handelskai'
  12         dtime = datetime.now()
  13         dtime = dtime.replace(hour=15, minute=0)
  14         search = Search(origin, destination)
  15         self.bs = BeautifulSoup(search.get_html(dtime))
  16
  17     def test_overview(self):
  18         self.assertEquals(1, len(self.bs.findAll('table', {'id': 'tbl_fahrten'})))
  19
  20     def test_details(self):
  21         self.assertTrue(len(self.bs.findAll('div', {'class': 'data_table tourdetail'})) > 0)
  22
  23
  24 if __name__ == '__main__':
  25     unittest.main()