encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-07-30 18:09:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

Be strict with permission get in test

Browse Source
This commit is contained in:
Markus Törnqvist 2013-05-19 15:09:57 +03:00
parent da0b7f3f4d
commit 187e26e6af
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
rest_framework/tests/test_permissions.py
View File

@ -39,14 +39,19 @@ class ModelPermissionsIntegrationTests(TestCase):
def setUp(self): def setUp(self):
User.objects.create_user('disallowed', 'disallowed@example.com', 'password') User.objects.create_user('disallowed', 'disallowed@example.com', 'password')
user = User.objects.create_user('permitted', 'permitted@example.com', 'password') user = User.objects.create_user('permitted', 'permitted@example.com', 'password')
# Keep get() lines clean and short
perm_kwargs = {
'content_type__app_label': 'tests',
'content_type__name': 'basic model',
}
user.user_permissions = [ user.user_permissions = [
Permission.objects.get(content_type__name='basic model', codename='add_basicmodel'), Permission.objects.get(codename='add_basicmodel', **perm_kwargs),
Permission.objects.get(content_type__name='basic model', codename='change_basicmodel'), Permission.objects.get(codename='change_basicmodel', **perm_kwargs),
Permission.objects.get(content_type__name='basic model', codename='delete_basicmodel') Permission.objects.get(codename='delete_basicmodel', **perm_kwargs)
] ]
user = User.objects.create_user('updateonly', 'updateonly@example.com', 'password') user = User.objects.create_user('updateonly', 'updateonly@example.com', 'password')
user.user_permissions = [ user.user_permissions = [
Permission.objects.get(content_type__name='basic model', codename='change_basicmodel'), Permission.objects.get(codename='change_basicmodel', **perm_kwargs),
] ]
self.permitted_credentials = basic_auth_header('permitted', 'password') self.permitted_credentials = basic_auth_header('permitted', 'password')