def test_getPrivilegesForPersonJSON(self):
# Define prerequisite data.
personKey = 3
organizationKey = 1
# Get the result of the tested method.
jsonString = controller_privileges.getPrivilegesForPersonJSON(personKey, organizationKey)
# Validate the result.
self.assertIsNotNone(jsonString)
self.assertTrue(len(jsonString) > 0)
dict = json.loads(jsonString)
personKeyInOutput = False
orgKeyInOutput = False
privCount = 0
for key1,value1 in dict.iteritems():
if key1 == models.EMPLOYEE_ENTITYFK_KEY:
self.assertEqual(value1, personKey)
personKeyInOutput = True
elif key1 == models.ORGANIZATION_ENTITYFK_KEY:
self.assertEqual(value1, organizationKey)
orgKeyInOutput = True
elif key1 == 'PersonPrivileges':
for dictPrivilege in value1:
privilege = controller_privileges.extractPrivilegeFromDict(dictPrivilege)
privCount += 1
self.assertTrue(privilege.pk is 5 or privilege.pk is 6)
self.assertTrue(privilege.privilege == 'VIEW_ALL_EMPLOYEES_IN_ORG' or privilege.privilege == 'ASSIGN_EMPS_TO_SHIFTS')
self.assertEqual(privCount, 2)
self.assertTrue(personKeyInOutput)
self.assertTrue(orgKeyInOutput)
def privilege_org_member(org_id, person_id):
try:
user_id = current_user.entityFK;
if request.method == 'GET':
privileges = controller_privileges.getPrivilegesForPersonJSON(person_id, org_id)
if is_request_json():
return Response(response=privileges, mimetype='application/json')
elif request.method == 'POST':
if is_request_json():
result = controller_privileges.grantPrivilegeToPersonJSON(db, request.json['privilege_id'], person_id, org_id)
return Response(response=result, mimetype='application/json')
except Exception, e:
return abort(404)
def test_getPrivilegesForPersonJSON_noprivilege(self):
# Define prerequisite data.
personKey = 5
organizationKey = 1
# Get the result of the tested method.
jsonString = controller_privileges.getPrivilegesForPersonJSON(personKey, organizationKey)
# Validate the result.
self.assertIsNotNone(jsonString)
self.assertTrue(len(jsonString) > 0)
dict = json.loads(jsonString)
personKeyInOutput = False
orgKeyInOutput = False
for key1,value1 in dict.iteritems():
if key1 == models.EMPLOYEE_ENTITYFK_KEY:
self.assertEqual(value1, personKey)
personKeyInOutput = True
elif key1 == models.ORGANIZATION_ENTITYFK_KEY:
self.assertEqual(value1, organizationKey)
orgKeyInOutput = True
elif key1 == 'PersonPrivileges':
self.assertEqual(value1, 'None')
self.assertTrue(personKeyInOutput)
self.assertTrue(orgKeyInOutput)
