diff options
| -rw-r--r-- | archaeological_finds/tests.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py index 976669f8a..a55e075b7 100644 --- a/archaeological_finds/tests.py +++ b/archaeological_finds/tests.py @@ -20,7 +20,7 @@ import json from django.conf import settings -from django.contrib.auth.models import User, Permission +from django.contrib.auth.models import User, Permission, ContentType from django.core.files import File from django.core.files.uploadedfile import SimpleUploadedFile from django.core.urlresolvers import reverse @@ -683,10 +683,12 @@ class FindPermissionTest(FindInit, TestCase): def setUp(self): self.username, self.password, self.user = create_superuser() self.alt_username, self.alt_password, self.alt_user = create_user() + ct_find = ContentType.objects.get(app_label='archaeological_finds', + model='find') self.alt_user.user_permissions.add(Permission.objects.get( - codename='view_own_find')) + codename='view_own_find', content_type=ct_find)) self.alt_user.user_permissions.add(Permission.objects.get( - codename='change_own_find')) + codename='change_own_find', content_type=ct_find)) self.alt_username2, self.alt_password2, self.alt_user2 = create_user( username='luke', password='iamyourfather' ) |