Tests: fix Permission get

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'
         )