def test_getShiftsByPersonJSON_false(self):
key = 999
shiftPersonListJSON = shiftperson_controller.getShiftsByPersonJSON(key)
shiftPersonListDict = json.loads(shiftPersonListJSON)
for rootKey, rootVal in shiftPersonListDict.iteritems():
self.assertEqual(rootKey, 'Shifts')
self.assertEqual(rootVal, 'None')
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)
