tests/test_itinerary_parser.py

   1 #
   2 # vi: set softtabstop=4 shiftwidth=4 tabstop=8 expandtab:
   3
   4 import unittest
   5 import Itinerary
   6
   7 class TestItinGrab(unittest.TestCase):
   8
   9     def setUp(self):
  10         fp = open(self.filename, "r")
  11         html = fp.read()
  12         fp.close()
  13
  14         itinp = Itinerary.ItineraryParser()
  15         itinp.feed(html)
  16         itinp.close()
  17
  18         self.entries = itinp.entries
  19
  20
  21 class TestChampagneItin(TestItinGrab):
  22
  23     filename = "grabs/champagne-itin.html"
  24
  25     def testEntries(self):
  26         print self.entries
  27
  28 class TestLongItin(TestItinGrab):
  29
  30     filename = "grabs/long-itin.html"
  31
  32     def testEntries(self):
  33         print self.entries
  34
  35 if __name__ == '__main__':
  36     unittest.main()