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__isDuplicateAssignment_false(self): shiftPerson1 = models.ShiftPerson.query.first() shiftPerson2 = models.ShiftPerson(shiftPerson1.shiftFK, shiftPerson1.personFK) shiftPerson2.pk = shiftPerson1.pk shiftPerson2.shiftFK = 2 result = shiftperson_controller._isDuplicateAssignment(shiftPerson2) self.assertFalse(result)
def test__removeShiftPerson_invalid(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) # Try to delete it even though we know it's not there result = shiftperson_controller._removeShiftPerson(shiftFK, personFK, self.db) self.assertFalse(result)