def test__getEventByID(self):
event = events._getEventByID(1)
self.assertIsNotNone(event)
self.assertEqual(event.name, 'My Event')
self.assertEqual(event.description, 'This is my event')
self.assertEqual(event.organizationFK, 1)
event = events._getEventByID(2)
self.assertIsNotNone(event)
self.assertEqual(event.name, 'Your Event')
self.assertEqual(event.description, 'This is your event')
self.assertEqual(event.organizationFK, 2)
event = events._getEventByID(999)
self.assertIsNone(event)
def _insertShift(shift, db):
result = "Unknown"
isDup = _isDuplicateShift(shift)
event = events._getEventByID(shift.eventFK)
if event is None:
result = "BadEvent"
if isDup == False and result == "Unknown":
db.session.add(shift)
db.session.commit()
if shift.pk > 0:
result = shift.pk
if result == "Unknown":
result = "Duplicate"
return result
