def test_getPeopleByShiftJSON_false(self):
key = 999
shiftPersonListJSON = shiftperson_controller.getPeopleByShiftJSON(key)
shiftPersonListDict = json.loads(shiftPersonListJSON)
for rootKey, rootVal in shiftPersonListDict.iteritems():
self.assertEqual(rootKey, 'Workers')
self.assertEqual(rootVal, 'None')
def getPersonsByOrgEventShift(org_id, event_id, shift_id):
try:
shiftPersonJSON = shiftperson_controller.getPeopleByShiftJSON(shift_id)
shiftPersonDict = json.loads(shiftPersonJSON)
if is_request_json():
return jsonify(shiftPersonDict, event_id=event_id, shift_id=shift_id)
except Exception, e:
return abort(404)
def test_getPeopleByShiftJSON_true(self):
key = 1
shiftPersonListJSON = shiftperson_controller.getPeopleByShiftJSON(key)
shiftPersonListDict = json.loads(shiftPersonListJSON)
for rootKey, rootVal in shiftPersonListDict.iteritems():
self.assertEqual(rootKey, 'Workers')
counter = 0
for personJSON in rootVal:
person = controllers.extractEmployeeFromDict(personJSON)
if person.entityFK == 3:
counter += 1
self.assertEqual(person.firstname, 'Chris')
self.assertEqual(person.lastname, 'Workman')
elif person.entityFK == 4:
counter += 1
self.assertEqual(person.firstname, 'Ryoji')
self.assertEqual(person.lastname, 'Betchaku')
self.assertEqual(counter, 2)
