mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Fix permission assignment in tests

Browse Source
This commit is contained in:
Jeremy Stretch
2020-05-27 16:53:30 -04:00
parent fb7446487e
commit ce46512c74
4 changed files with 38 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
netbox/utilities/testing/testcases.py
View File

@@ -33,18 +33,31 @@ class TestCase(_TestCase):
Assign a set of permissions to the test user. Accepts permission names in the form <app>.<action>_<model>.
"""
for name in names:
app, codename = name.split('.')
perm = Permission.objects.get(content_type__app_label=app, codename=codename)
self.user.user_permissions.add(perm)
app_label, codename = name.split('.')
action, model_name = codename.split('_')
kwargs = {
'model': ContentType.objects.get(app_label=app_label, model=model_name),
f'can_{action}': True
}
obj_perm = ObjectPermission(**kwargs)
obj_perm.save()
obj_perm.users.add(self.user)
def remove_permissions(self, *names):
"""
Remove a set of permissions from the test user, if assigned.
"""
for name in names:
app, codename = name.split('.')
perm = Permission.objects.get(content_type__app_label=app, codename=codename)
self.user.user_permissions.remove(perm)
app_label, codename = name.split('.')
action, model_name = codename.split('_')
kwargs = {
'user': self.user,
'model': ContentType.objects.get(app_label=app_label, model=model_name),
f'can_{action}': True
}
ObjectPermission.objects.filter(**kwargs).delete()
#
# Convenience methods
@@ -493,10 +506,14 @@ class ViewTestCases:
with disable_warnings('django.request'):
self.assertHttpStatus(self.client.post(**request), 403)
# Assign the required permission and submit again
self.add_permissions(
'{}.add_{}'.format(self.model._meta.app_label, self.model._meta.model_name)
# Assign object-level permission
obj_perm = ObjectPermission(
model=ContentType.objects.get_for_model(self.model),
can_add=True
)
obj_perm.save()
obj_perm.users.add(self.user)
response = self.client.post(**request)
self.assertHttpStatus(response, 302)