|
@@ -1,11 +1,7 @@
|
|
|
from django.core import exceptions as django_exceptions
|
|
|
-from django.core.exceptions import PermissionDenied
|
|
|
-from django.http import Http404
|
|
|
from django.test import TestCase
|
|
|
|
|
|
from misago.core import exceptionhandler
|
|
|
-from misago.core.exceptions import Banned
|
|
|
-from misago.users.models import Ban
|
|
|
|
|
|
|
|
|
INVALID_EXCEPTIONS = [
|
|
@@ -46,37 +42,3 @@ class GetExceptionHandlerTests(TestCase):
|
|
|
for exception in INVALID_EXCEPTIONS:
|
|
|
with self.assertRaises(ValueError):
|
|
|
exceptionhandler.get_exception_handler(exception())
|
|
|
-
|
|
|
-
|
|
|
-class HandleAPIExceptionTests(TestCase):
|
|
|
- def test_banned(self):
|
|
|
- """banned exception is correctly handled"""
|
|
|
- ban = Ban(user_message="This is test ban!")
|
|
|
-
|
|
|
- response = exceptionhandler.handle_api_exception(Banned(ban), None)
|
|
|
-
|
|
|
- self.assertEqual(response.status_code, 403)
|
|
|
- self.assertEqual(response.data['ban']['message']['html'], "<p>This is test ban!</p>")
|
|
|
-
|
|
|
- def test_permission_denied(self):
|
|
|
- """permission denied exception is correctly handled"""
|
|
|
- response = exceptionhandler.handle_api_exception(PermissionDenied(), None)
|
|
|
- self.assertEqual(response.status_code, 403)
|
|
|
- self.assertEqual(response.data['detail'], "Permission denied.")
|
|
|
-
|
|
|
- def test_permission_message_denied(self):
|
|
|
- """permission denied with message is correctly handled"""
|
|
|
- exception = PermissionDenied("You shall not pass!")
|
|
|
- response = exceptionhandler.handle_api_exception(exception, None)
|
|
|
- self.assertEqual(response.status_code, 403)
|
|
|
- self.assertEqual(response.data['detail'], "You shall not pass!")
|
|
|
-
|
|
|
- def test_unhandled_exception(self):
|
|
|
- """our exception handler is not interrupting other exceptions"""
|
|
|
- for exception in INVALID_EXCEPTIONS:
|
|
|
- response = exceptionhandler.handle_api_exception(exception(), None)
|
|
|
- self.assertIsNone(response)
|
|
|
-
|
|
|
- response = exceptionhandler.handle_api_exception(Http404(), None)
|
|
|
- self.assertEqual(response.status_code, 404)
|
|
|
- self.assertEqual(response.data['detail'], "Not found.")
|