Browse Source

fix misago.core tests suite

Rafał Pitoń 10 years ago
parent
commit
4e501c46af

+ 5 - 0
misago/core/tests/test_errorpages.py

@@ -2,6 +2,8 @@ from django.core.urlresolvers import reverse
 from django.test import Client, TestCase
 from django.test.client import RequestFactory
 
+from misago.users.models import AnonymousUser
+
 from misago.core.testproject.views import (mock_custom_403_error_page,
                                            mock_custom_404_error_page)
 
@@ -45,6 +47,9 @@ class CustomErrorPagesTests(TestCase):
         self.misago_request = RequestFactory().get(reverse('misago:index'))
         self.site_request = RequestFactory().get(reverse('raise_403'))
 
+        self.misago_request.user = AnonymousUser()
+        self.site_request.user = AnonymousUser()
+
         self.misago_request.preloaded_ember_data = {}
         self.site_request.preloaded_ember_data = {}
 

+ 3 - 0
misago/core/tests/test_exceptionhandler_middleware.py

@@ -3,12 +3,15 @@ from django.http import Http404
 from django.test import TestCase
 from django.test.client import RequestFactory
 
+from misago.users.models import AnonymousUser
+
 from misago.core.middleware.exceptionhandler import ExceptionHandlerMiddleware
 
 
 class ExceptionHandlerMiddlewareTests(TestCase):
     def setUp(self):
         self.request = RequestFactory().get(reverse('misago:index'))
+        self.request.user = AnonymousUser()
         self.request.preloaded_ember_data = {}
 
     def test_middleware_returns_response_for_supported_exception(self):

+ 7 - 6
misago/core/tests/test_exceptionhandlers.py

@@ -53,23 +53,24 @@ class GetExceptionHandlerTests(TestCase):
 class HandleAPIExceptionTests(TestCase):
     def test_permission_denied(self):
         """permission denied exception is correctly handled"""
-        response = exceptionhandler.handle_api_exception(PermissionDenied())
+        response = exceptionhandler.handle_api_exception(
+            PermissionDenied(), None)
         self.assertEqual(response.status_code, 403)
-        self.assertEqual(response.data['detail'], "Permission denied")
+        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)
+        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())
+            response = exceptionhandler.handle_api_exception(exception(), None)
             self.assertIsNone(response)
 
-        response = exceptionhandler.handle_api_exception(Http404())
+        response = exceptionhandler.handle_api_exception(Http404(), None)
         self.assertEqual(response.status_code, 404)
-        self.assertEqual(response.data['detail'], "Not found")
+        self.assertEqual(response.data['detail'], "Not found.")