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