def test__removeShift_true(self):
# Define the values we're going to have to add for the test.
newShift = models.Shift()
newShift.eventFK = 1
newShift.startdatetime = datetime.datetime(2013,7,28,16,0)
newShift.enddatetime = datetime.datetime(2013,7,28,17,0)
newShift.location = 'Booth C.5'
newShift.minWorkers = 24
newShift.maxWorkers = 42
# Ensure it's not already in db.
result = shifts_controller._isDuplicateShift(newShift)
self.assertFalse(result)
# Add it to db
newShiftPK = shifts_controller._insertShift(newShift, self.db)
self.assertTrue(newShiftPK > 0)
# Ensure it's in db.
result = shifts_controller._isDuplicateShift(newShift)
self.assertTrue(result)
# Now that we added it, lets delete it.
result = shifts_controller._removeShift(newShiftPK, self.db)
self.assertTrue(result)
# Ensure it's not in db.
result = shifts_controller._isDuplicateShift(newShift)
self.assertFalse(result)
def test__removeShift_invalid(self):
# Define the values we're going to have to add for the test.
newShift = models.Shift()
newShift.eventFK = 1
newShift.startdatetime = datetime.datetime(2013,7,28,16,0)
newShift.enddatetime = datetime.datetime(2013,7,28,17,0)
newShift.location = 'Booth C.5'
newShift.minWorkers = 24
newShift.maxWorkers = 42
# Ensure it's not already in db.
result = shifts_controller._isDuplicateShift(newShift)
self.assertFalse(result)
# Try to delete it even though we know it's not there
result = shifts_controller._removeShift(newShift.pk, self.db)
self.assertFalse(result)
def test__removeShift_false(self):
# Try to delete a record that's not there
result = shifts_controller._removeShift(9999999, self.db)
self.assertFalse(result)
