def test_import_calendar_duplicate(self):
calendar_txt = StringIO("""\
service_id,monday,tuesday,wednesday,thursday,friday,saturday,sunday,\
start_date,end_date
W,1,0,1,0,1,0,1,20120414,20121231
W,0,1,0,1,0,1,0,20120414,20121231
""")
Service.import_txt(calendar_txt, self.feed)
service = Service.objects.get() # Only one
self.assertEqual(service.feed, self.feed)
self.assertEqual(service.service_id, 'W')
def test_import_calendar_duplicate(self):
calendar_txt = StringIO("""\
service_id,monday,tuesday,wednesday,thursday,friday,saturday,sunday,\
start_date,end_date
W,1,0,1,0,1,0,1,20120414,20121231
W,0,1,0,1,0,1,0,20120414,20121231
""")
Service.import_txt(calendar_txt, self.feed)
service = Service.objects.get() # Only one
self.assertEqual(service.feed, self.feed)
self.assertEqual(service.service_id, 'W')
def test_import_calendar_txt(self):
calendar_txt = StringIO("""\
service_id,monday,tuesday,wednesday,thursday,friday,saturday,sunday,\
start_date,end_date
W,1,0,1,0,1,0,1,20120414,20121231
""")
Service.import_txt(calendar_txt, self.feed)
service = Service.objects.get()
self.assertEqual(service.feed, self.feed)
self.assertEqual(service.service_id, 'W')
self.assertTrue(service.monday)
self.assertFalse(service.tuesday)
self.assertTrue(service.wednesday)
self.assertFalse(service.thursday)
self.assertTrue(service.friday)
self.assertFalse(service.saturday)
self.assertTrue(service.sunday)
self.assertEqual(service.start_date, date(2012, 4, 14))
self.assertEqual(service.end_date, date(2012, 12, 31))
def test_import_calendar_txt(self):
calendar_txt = StringIO.StringIO("""\
service_id,monday,tuesday,wednesday,thursday,friday,saturday,sunday,\
start_date,end_date
W,1,0,1,0,1,0,1,20120414,20121231
""")
Service.import_txt(calendar_txt, self.feed)
service = Service.objects.get()
self.assertEqual(service.feed, self.feed)
self.assertEqual(service.service_id, 'W')
self.assertTrue(service.monday)
self.assertFalse(service.tuesday)
self.assertTrue(service.wednesday)
self.assertFalse(service.thursday)
self.assertTrue(service.friday)
self.assertFalse(service.saturday)
self.assertTrue(service.sunday)
self.assertEqual(service.start_date, date(2012, 4, 14))
self.assertEqual(service.end_date, date(2012, 12, 31))
