don't allow us to create permissions decorator without permissions

ie: not for an organisation, and not for a service
This commit is contained in:
Leo Hemsted
2018-03-01 12:08:22 +00:00
parent d14f33ea70
commit 7f268c0ab3
3 changed files with 46 additions and 7 deletions

View File

@@ -121,20 +121,23 @@ class User(UserMixin):
if unknown_permissions:
raise TypeError('{} are not valid permissions'.format(list(unknown_permissions)))
# platform admins should be able to do most things (except eg send messages, or create api keys)
if self.platform_admin and not restrict_admin_usage:
return True
# Service id is always set on the request for service specific views.
service_id = _get_service_id_from_view_args()
org_id = _get_org_id_from_view_args()
if not service_id and not org_id:
# we shouldn't have any pages that require permissions, but don't specify a service or organisation.
# use @user_is_platform_admin for platform admin only pages
raise NotImplementedError
# platform admins should be able to do most things (except eg send messages, or create api keys)
if self.platform_admin and not restrict_admin_usage:
return True
if org_id:
return org_id in self.organisations
elif service_id:
return any(x in self._permissions.get(service_id, []) for x in permissions)
else:
return False
def has_permission_for_service(self, service_id, permission):
return permission in self._permissions.get(service_id, [])

View File

@@ -107,7 +107,7 @@
{% macro edit_field(text, link, permissions=[]) -%}
{% call field(align='right') %}
{% if current_user.has_permissions(*permissions) or not permissions %}
{% if not permissions or current_user.has_permissions(*permissions) %}
<a href="{{ link }}">{{ text }}</a>
{% endif %}
{% endcall %}

View File

@@ -156,6 +156,42 @@ def test_user_has_permissions_for_organisation(
index()
def test_platform_admin_can_see_orgs_they_dont_have(
client,
platform_admin_user,
mocker,
):
platform_admin_user.organisations = []
mocker.patch('app.user_api_client.get_user', return_value=platform_admin_user)
client.login(platform_admin_user)
request.view_args = {'org_id': 'org_2'}
@user_has_permissions()
def index():
pass
index()
def test_cant_use_decorator_without_view_args(
client,
platform_admin_user,
mocker,
):
mocker.patch('app.user_api_client.get_user', return_value=platform_admin_user)
client.login(platform_admin_user)
request.view_args = {}
@user_has_permissions()
def index():
pass
with pytest.raises(NotImplementedError):
index()
def test_user_doesnt_have_permissions_for_organisation(
client,
mocker,