def test_getShiftsByPersonJSON_true(self):
key = 3
shiftPersonListJSON = shiftperson_controller.getShiftsByPersonJSON(key)
shiftPersonListDict = json.loads(shiftPersonListJSON)
for rootKey, rootVal in shiftPersonListDict.iteritems():
self.assertEqual(rootKey, 'Shifts')
counter = 0
for shiftJSON in rootVal:
shift = shifts_controller.extractShiftFromDict(shiftJSON)
if shift.pk == 1:
counter += 1
self.assertEqual(shift.startdatetime, datetime.datetime(2013, 7, 12, 12, 0))
self.assertEqual(shift.enddatetime, datetime.datetime(2013, 7, 12, 13, 0))
self.assertEqual(shift.location, 'Booth A')
self.assertEqual(shift.minWorkers, 2)
self.assertEqual(shift.maxWorkers, 4)
elif shift.pk == 3:
counter += 1
self.assertEqual(shift.startdatetime, datetime.datetime(2013, 7, 12, 14, 0))
self.assertEqual(shift.enddatetime, datetime.datetime(2013, 7, 12, 15, 0))
self.assertEqual(shift.location, 'Booth A')
self.assertEqual(shift.minWorkers, 3)
self.assertEqual(shift.maxWorkers, 4)
elif shift.pk == 4:
counter += 1
self.assertEqual(shift.startdatetime, datetime.datetime(2013, 7, 12, 15, 0))
self.assertEqual(shift.enddatetime, datetime.datetime(2013, 7, 12, 16, 0))
self.assertEqual(shift.location, 'Booth A')
self.assertEqual(shift.minWorkers, 3)
self.assertEqual(shift.maxWorkers, 4)
self.assertEqual(counter, 3)
def test_extractShiftFromDict(self):
control = models.Shift.query.first()
stringJSON = shifts_controller.shiftToJSON(control)
data = json.loads(stringJSON)
target = shifts_controller.extractShiftFromDict(data)
self.assertEqual(control.pk, target.pk)
self.assertEqual(control.eventFK, target.eventFK)
self.assertEqual(control.startdatetime, target.startdatetime)
self.assertEqual(control.enddatetime, target.enddatetime)
self.assertEqual(control.location, target.location)
self.assertEqual(control.minWorkers, target.minWorkers)
self.assertEqual(control.maxWorkers, target.maxWorkers)
