def test_removeShiftPerson_true(self): # Define the values we're going to have to add for the test. shiftFK = 3 personFK = 4 newShiftPerson = models.ShiftPerson(shiftFK, personFK) # Ensure it's not already in db. result = shiftperson_controller._isDuplicateAssignment(newShiftPerson) self.assertFalse(result) # Add it to db newShiftPersonPK = shiftperson_controller._insertShiftPerson(newShiftPerson, self.db) self.assertTrue(newShiftPersonPK > 0) # Ensure it's in db. result = shiftperson_controller._isDuplicateAssignment(newShiftPerson) self.assertTrue(result) # Now that we added it, lets delete it. result = shiftperson_controller.removeShiftPerson(shiftFK, personFK, self.db) self.assertIsNotNone(result) resultDict = json.loads(result) for key,value in resultDict.iteritems(): if key == 'success': self.assertEqual(value, 'true') if key == models.SHIFTPERSON_PK_KEY: self.assertEqual(value, newShiftPersonPK) # Ensure it's not in db. result = shiftperson_controller._isDuplicateAssignment(newShiftPerson) self.assertFalse(result)
def test__removeShiftPerson_true(self): # Define the values we're going to have to add for the test. shiftFK = 3 personFK = 4 newShiftPerson = models.ShiftPerson(shiftFK, personFK) # Ensure it's not already in db. result = shiftperson_controller._isDuplicateAssignment(newShiftPerson) self.assertFalse(result) # Add it to db newShiftPersonPK = shiftperson_controller._insertShiftPerson(newShiftPerson, self.db) self.assertTrue(newShiftPersonPK > 0) # Ensure it's in db. result = shiftperson_controller._isDuplicateAssignment(newShiftPerson) self.assertTrue(result) # Now that we added it, lets delete it. result = shiftperson_controller._removeShiftPerson(shiftFK, personFK, self.db) self.assertTrue(result) # Ensure it's not in db. result = shiftperson_controller._isDuplicateAssignment(newShiftPerson) self.assertFalse(result)
def test__insertShiftPerson_true(self): shiftFK = 1 personFK = 5 shiftPerson = models.ShiftPerson(shiftFK, personFK) result = shiftperson_controller._insertShiftPerson(shiftPerson, self.db) self.assertTrue(result > 0) newShiftPerson = models.ShiftPerson.query.filter_by(pk=result).first() self.assertEqual(newShiftPerson.pk, result) self.assertEqual(newShiftPerson.shiftFK, shiftFK) self.assertEqual(newShiftPerson.personFK, personFK)
def test__insertShiftPerson_badperson(self): shiftFK = 1 personFK = 999 shiftPerson = models.ShiftPerson(shiftFK, personFK) result = shiftperson_controller._insertShiftPerson(shiftPerson, self.db) self.assertEqual(result, 'BadPerson')
def test__insertShiftPerson_duplicate(self): shiftFK = 1 personFK = 4 shiftPerson = models.ShiftPerson(shiftFK, personFK) result = shiftperson_controller._insertShiftPerson(shiftPerson, self.db) self.assertEqual(result, 'Duplicate')