Ticket #17113: test_custom_handlers.diff

tests/regressiontests/test_utils/tests.py

@@ -1 +1 @@
 from __future__ import with_statement, absolute_import
 
+from django.core.urlresolvers import get_resolver
 from django.forms import EmailField
 from django.test import SimpleTestCase, TestCase, skipUnlessDBFeature
 from django.utils.unittest import skip
@@ -117 +118 @@
         self.restore_warnings_state()
 
 
+class CustomUrlHandlersTests(TestCase):
+    urls = 'regressiontests.test_utils.urls'
+
+    def test_custom_error_handlers(self):
+        """ Test that error handlers of a custom urls.py are used """
+        from .views import empty_view
+        handler = (empty_view, {})
+        resolver = get_resolver(None)
+        self.assertEqual(resolver.resolve404(), handler)
+
+
 class SkippingExtraTests(TestCase):
     fixtures = ['should_not_be_loaded.json']
 

tests/regressiontests/test_utils/urls.py

@@ -4 +4 @@
 
 from . import views
 
+handler404 = 'regressiontests.test_utils.views.empty_view'
 
 urlpatterns = patterns('',
     (r'^test_utils/get_person/(\d+)/$', views.get_person),

tests/regressiontests/test_utils/views.py

@@ -8 +8 @@
 
 def get_person(request, pk):
     person = get_object_or_404(Person, pk=pk)
-    return HttpResponse(person.name)
- No newline at end of file
+    return HttpResponse(person.name)
+
+def empty_view(request, *args, **kwargs):
+    return HttpResponse('')