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

Introduce resolve_permission() utility function

Browse Source
This commit is contained in:
Jeremy Stretch
2020-05-28 10:35:59 -04:00
parent ca199cdefe
commit dc56e49410
4 changed files with 30 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -7,6 +7,7 @@ from django.urls import reverse, NoReverseMatch
from rest_framework.test import APIClient
from users.models import ObjectPermission, Token
from utilities.permissions import resolve_permission
from .utils import disable_warnings, post_data
@ -32,11 +33,9 @@ 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_label, codename = name.split('.')
action, model_name = codename.split('_')
ct, action = resolve_permission(name)
self.user.object_permissions.create(**{
'model': ContentType.objects.get(app_label=app_label, model=model_name),
'model': ct,
f'can_{action}': True
})
@ -45,11 +44,9 @@ class TestCase(_TestCase):
Remove a set of permissions from the test user, if assigned.
"""
for name in names:
app_label, codename = name.split('.')
action, model_name = codename.split('_')
ct, action = resolve_permission(name)
self.user.object_permissions.filter(**{
'model': ContentType.objects.get(app_label=app_label, model=model_name),
'model': ct,
f'can_{action}': True
}).delete()