def setPersonsByOrgEventShift(org_id, event_id, shift_id):
try:
if is_request_json():
if not request.json.has_key('person_id') is None:
assignShirt = shiftperson_controller.insertShiftPerson(shift_id, request.json['person_id'], db)
return Response(response=assignShirt, mimetype='application/json')
except Exception, e:
return abort(404)
def test_insertShiftPerson_duplicate(self):
shiftPerson1 = models.ShiftPerson.query.first()
shiftPerson2 = models.ShiftPerson(shiftPerson1.personFK)
shiftPersonJSON = shiftperson_controller.shiftPersonToJSON(shiftPerson2)
shiftPersonDict = json.loads(shiftPersonJSON)
result = shiftperson_controller.insertShiftPerson(shiftPerson1.shiftFK, shiftPerson1.personFK, self.db)
resultDict = json.loads(result)
for key,value in resultDict.iteritems():
if key == 'success':
self.assertEqual(value, 'false')
if key == 'msg':
self.assertEqual(value, 'Duplicate')
elif key == models.SHIFTPERSON_PK_KEY:
self.assertEqual(value, 'None')
def test_insertShiftPerson_badperson(self):
shiftFK = 1
personFK = 999
shiftPerson1 = models.ShiftPerson()
shiftPerson1.personFK = 999
shiftPersonJSON = shiftperson_controller.shiftPersonToJSON(shiftPerson1)
shiftPersonDict = json.loads(shiftPersonJSON)
result = shiftperson_controller.insertShiftPerson(shiftFK, personFK, self.db)
resultDict = json.loads(result)
for key,value in resultDict.iteritems():
if key == 'success':
self.assertEqual(value, 'false')
if key == 'msg':
self.assertEqual(value, 'BadPerson')
elif key == models.SHIFTPERSON_PK_KEY:
self.assertEqual(value, 'None')
def test_insertShiftPerson_true(self):
shiftFK = 2
personFK = 3
shiftPersonJSON = '{'
shiftPersonJSON += '"{key}":"{val}"'.format(key=models.SHIFTPERSON_PERSON_KEY, val=personFK)
shiftPersonJSON += '}'
shiftPersonDict = json.loads(shiftPersonJSON)
result = shiftperson_controller.insertShiftPerson(shiftFK, personFK, self.db)
self.assertIsNotNone(result)
newKey = 0
resultDict = json.loads(result)
for key,value in resultDict.iteritems():
if key == 'success':
self.assertEqual(value, 'true')
if key == models.SHIFTPERSON_PK_KEY:
self.assertTrue(value > 0)
newKey = value
newShiftPerson = models.ShiftPerson.query.filter_by(pk=newKey).first()
self.assertEqual(newShiftPerson.pk, newKey)
self.assertEqual(newShiftPerson.shiftFK, shiftFK)
self.assertEqual(newShiftPerson.personFK, personFK)
