diff --git a/django_prbac/tests/test_decorators.py b/django_prbac/tests/test_decorators.py index 1dd9d8f..e1d522a 100644 --- 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):