def test_extractEventFromDict(self): control = models.Event.query.first() stringJSON = '{"event_pk":1, "event_name":"My Event", "event_desc":"This is my event", "event_start":"2013-07-12 12:00:00", "event_end":"2013-07-14 16:00:00", "event_orgfk":1}' data = json.loads(stringJSON) target = events.extractEventFromDict(data) self.assertEqual(control.pk, target.pk) self.assertEqual(control.name, target.name) self.assertEqual(control.description, target.description) self.assertEqual(control.startdate, target.startdate) self.assertEqual(control.enddate, target.enddate) self.assertEqual(control.organizationFK, target.organizationFK)
def test_getEventsByOrg_true(self): key = 1 eventListJSON = events.getEventsByOrgJSON(key) eventListDict = json.loads(eventListJSON) for rootKey, rootVal in eventListDict.iteritems(): self.assertEqual(rootKey, 'Events') counter = 0 for eventJSON in rootVal: event = events.extractEventFromDict(eventJSON) if event.pk == 1: counter += 1 self.assertEqual(event.pk, 1) self.assertEqual(event.name, 'My Event') self.assertEqual(event.description, 'This is my event') self.assertEqual(event.organizationFK, key) self.assertEqual(counter, 1)
def test_extractEventFromDict_incomplete(self): control = models.Event() control.name = "D Event 3" control.startdate = datetime.datetime(2013, 3, 22) control.enddate = datetime.datetime(2013, 3, 22) control.organizationFK = 4 #stringJSON = '{"event_name":"test_extractEventFromDict_incomplete event", "event_desc":"test_extractEventFromDict_incomplete description is indescribable", "event_start":"2013-03-28 17:30:00", "event_end":"2013-03-28 20:00:00"}' stringJSON = '{"event_name":"D Event 3","event_start":"03/22/2013","event_end":"03/22/2013","event_orgfk":4}' data = json.loads(stringJSON) target = events.extractEventFromDict(data) self.assertEqual(control.pk, target.pk) self.assertEqual(control.name, target.name) self.assertEqual(control.description, target.description) self.assertEqual(control.startdate, target.startdate) self.assertEqual(control.enddate, target.enddate) self.assertEqual(control.organizationFK, target.organizationFK)