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)