def privilege():
try:
user_id = current_user.entityFK
if request.method == 'GET':
org_json = controller_privileges.getOrgsWithPrivilegesForPersonJSON(user_id)
privilege_json = controller_privileges.getAllPrivilegesJSON()
if is_request_json():
return jsonify(privilege_data=org_json + privilege_json)
else:
org_dict = json.loads(org_json)
privilege_dict = json.loads(privilege_json)
return render_template('privilege.html', orgs=org_dict, privileges=privilege_dict)
except Exception, e:
return abort(404)
def test_getAllPrivilegesJSON(self):
jsonString = controller_privileges.getAllPrivilegesJSON()
self.assertIsNotNone(jsonString)
self.assertTrue(len(jsonString) > 0)
dict = json.loads(jsonString)
privCount = 0
for key1,value1 in dict.iteritems():
if key1 == 'Privileges':
for dictPrivilege in value1:
privilege = controller_privileges.extractPrivilegeFromDict(dictPrivilege)
privCount += 1
self.assertTrue(privilege.pk >= 1 or privilege.pk <= 8)
self.assertTrue(privilege.privilege == 'ASSIGN_EMPS_TO_SHIFTS' or privilege.privilege == 'DELETE_ORGANIZATION' or
privilege.privilege == 'MODIFY_ORGANIZATION' or privilege.privilege == 'REGISTER_NEW_ORGANIZATION' or
privilege.privilege == 'SOME_OTHER_EMP_PRIVILEGE' or privilege.privilege == 'VIEW_ALL_EMPLOYEES_IN_ORG' or
privilege.privilege == 'VIEW_ALL_ORGANIZATIONS' or privilege.privilege == 'YET_ANOTHER_EMP_PRIVILEGE')
self.assertEqual(privCount, 8)
