def test_extractPersonFromDict(self): control = models.Person.query.first() stringJSON = '{"emp_entityfk":3,"firstname":"Chris","lastname":"Workman","username":"******","password":"******"}' data = json.loads(stringJSON) target = controllers.extractPersonFromDict(data) self.assertEqual(control.entityFK, target.entityFK) self.assertEqual(control.firstname, target.firstname) self.assertEqual(control.lastname, target.lastname) self.assertEqual(control.username, target.username) self.assertEqual(control.password, target.password)
def test_getPeopleInOrganizationJSON(self): # Define prerequisite data. organizationKey = 1 # Get the restult of the tested method. jsonString = controllers.getPeopleInOrganizationJSON(organizationKey) # Validate the result. self.assertIsNotNone(jsonString) self.assertTrue(len(jsonString) > 0) dict = json.loads(jsonString) for key,value in dict.iteritems(): self.assertEqual(key, 'People') count = 0 for personDict in value: count += 1 person = controllers.extractPersonFromDict(personDict) self.assertTrue(person.entityFK == 3 or person.entityFK == 4 or person.entityFK == 6) self.assertTrue(person.firstname == 'Chris' or person.firstname == 'Ryoji' or person.firstname == 'Cookie') self.assertTrue(person.lastname == 'Workman' or person.lastname == 'Betchaku' or person.lastname == 'Monster') self.assertEqual(count, 3)