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.assertIsNotNone(result)
resultDict = json.loads(result)
for key,value in resultDict.iteritems():
if key == 'success':
self.assertEqual(value, 'true')
if key == models.SHIFT_PK_KEY:
self.assertEqual(value, newShiftPK)
# Ensure it's not in db.
result = shifts_controller._isDuplicateShift(newShift)
self.assertFalse(result)
def removeShiftFromEvent(org_id, event_id, shift_id):
try:
result = shifts_controller.removeShift(event_id, db)
return Response(response=result, mimetype='application/json')
except Exception, e:
return abort(404)
