update tests to reflect the change back from Http404 to PermissionDenied

dimagi · Feb 14, 2014 · a0044e0 · a0044e0
1 parent 9d61f50
commit a0044e0
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/django_prbac/tests/test_decorators.py b/django_prbac/tests/test_decorators.py
@@ -3,7 +3,7 @@
 
 # Django imports
 from django.test import TestCase
-from django.http import HttpRequest, Http404
+from django.http import HttpRequest
 
 # Local imports
 from django_prbac.decorators import requires_privilege
@@ -25,7 +25,7 @@ def view(request, *args, **kwargs):
             pass
 
         request = HttpRequest()
-        with self.assertRaises(Http404):
+        with self.assertRaises(PermissionDenied):
             view(request)
 
     def test_requires_privilege_no_such(self):
@@ -41,7 +41,7 @@ def view(request, *args, **kwargs):
         requestor_role = arbitrary.role()
         request = HttpRequest()
         request.role = requestor_role
-        with self.assertRaises(Http404):
+        with self.assertRaises(PermissionDenied):
             view(request)
 
     def test_requires_privilege_denied(self):
@@ -59,7 +59,7 @@ def view(request, *args, **kwargs):
 
         request = HttpRequest()
         request.role = requestor_role.instantiate({})
-        with self.assertRaises(Http404):
+        with self.assertRaises(PermissionDenied):
             view(request)
 
 
@@ -74,7 +74,7 @@ def view(request, *args, **kwargs):
 
        request = HttpRequest()
        request.role = requestor_role.instantiate({})
-       with self.assertRaises(Http404):
+       with self.assertRaises(PermissionDenied):
            view(request)
 
     def test_requires_privilege_ok(self):