def test_autoGenerateShifts(self):
shiftCount = models.Shift.query.count()
shiftStartDateTime = datetime.datetime(2013, 3, 22, 3, 0)
shiftEndDateTime = datetime.datetime(2013, 3, 23, 7, 30)
shiftLength = 60
eventFK = 1
shifts_controller.autoGenerateShifts(eventFK, shiftStartDateTime, shiftEndDateTime, shiftLength, self.db)
shiftCountAfter = models.Shift.query.count()
self.assertEqual(shiftCountAfter, shiftCount + 8)
def _insertEvent(event, db):
result = 'Unknown'
isDup = _isDuplicateEvent(event)
org = controllers._getOrganizationByID(event.organizationFK)
if org is None:
result = 'BadOrg'
if(isDup == False and result == 'Unknown'):
db.session.add(event)
db.session.commit()
if(event.pk > 0):
result = event.pk
shifts_controller.autoGenerateShifts(event.pk, event.startdate, event.enddate, 60, db)
if(result == 'Unknown'):
result = 'Duplicate'
return result
