Просмотр исходного кода

Covered middleware with tests.

Rafał Pitoń 11 лет назад
Родитель
Сommit
b465749df8
1 измененных файлов с 22 добавлено и 0 удалено
  1. 22 0
      misago/views/tests/test_middleware.py

+ 22 - 0
misago/views/tests/test_middleware.py

@@ -0,0 +1,22 @@
+from django.core.urlresolvers import reverse
+from django.http import Http404
+from django.test import TestCase
+from django.test.client import RequestFactory
+from misago.views.middleware import MisagoExceptionHandlerMiddleware
+
+
+class MisagoExceptionHandlerMiddlewareTests(TestCase):
+    def setUp(self):
+        self.request = RequestFactory().get(reverse('forum_index'))
+
+    def test_middleware_returns_none_for_non_supported_exception(self):
+        exception = TypeError()
+        middleware = MisagoExceptionHandlerMiddleware()
+
+        self.assertFalse(middleware.process_exception(self.request, exception))
+
+    def test_middleware_returns_response_for_supported_exception(self):
+        exception = Http404()
+        middleware = MisagoExceptionHandlerMiddleware()
+
+        self.assertTrue(middleware.process_exception(self.request, exception))