def test_extractShiftPersonFromDict(self): control = models.ShiftPerson.query.first() stringJSON = shiftperson_controller.shiftPersonToJSON(control) data = json.loads(stringJSON) target = shiftperson_controller.extractShiftPersonFromDict(data) self.assertEqual(control.pk, target.pk) self.assertEqual(control.shiftFK, target.shiftFK) self.assertEqual(control.personFK, target.personFK)
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_shiftPersonToJSON(self): control = models.ShiftPerson.query.first() controlJSON = shiftperson_controller.shiftPersonToJSON(control) expectedJSON = stringJSON = '{"shiftperson_pk":1,"shiftperson_shiftfk":1,"shiftperson_personfk":3}' self.assertEqual(controlJSON, expectedJSON)