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))
def test_export_calendar_txt(self):
Service.objects.create(
feed=self.feed, service_id='W', monday=True, tuesday=False,
wednesday=True, thursday=False, friday=True, saturday=False,
sunday=True, start_date=date(2012, 7, 17),
end_date=date(2013, 7, 17))
calendar_txt = Service.export_txt(self.feed)
self.assertEqual(calendar_txt, """\
service_id,monday,tuesday,wednesday,thursday,friday,saturday,sunday,\
start_date,end_date
W,1,0,1,0,1,0,1,20120717,20130717
""")
def test_export_calendar_txt(self):
Service.objects.create(feed=self.feed,
service_id='W',
monday=True,
tuesday=False,
wednesday=True,
thursday=False,
friday=True,
saturday=False,
sunday=True,
start_date=date(2012, 7, 17),
end_date=date(2013, 7, 17))
calendar_txt = Service.export_txt(self.feed)
self.assertEqual(
calendar_txt, """\
service_id,monday,tuesday,wednesday,thursday,friday,saturday,sunday,\
start_date,end_date
W,1,0,1,0,1,0,1,20120717,20130717
""")
def test_export_calendar_txt_none(self):
calendar_txt = Service.export_txt(self.feed)
self.assertFalse(calendar_txt)
def test_export_calendar_txt_none(self):
calendar_txt = Service.export_txt(self.feed)
self.assertFalse(calendar_txt)
