Fixed #25163 -- Fixed exception handling in nested test client requests.

This commit is contained in:
pscottdevos 2015-08-07 14:28:54 -05:00 committed by Tim Graham
parent 56ed80ac2a
commit 7fa1dd8a80
4 changed files with 23 additions and 3 deletions

View File

@ -442,9 +442,9 @@ class Client(RequestFactory):
signal_uid = "template-render-%s" % id(request) signal_uid = "template-render-%s" % id(request)
signals.template_rendered.connect(on_template_render, dispatch_uid=signal_uid) signals.template_rendered.connect(on_template_render, dispatch_uid=signal_uid)
# Capture exceptions created by the handler. # Capture exceptions created by the handler.
got_request_exception.connect(self.store_exc_info, dispatch_uid="request-exception") exception_uid = "request-exception-%s" % id(request)
got_request_exception.connect(self.store_exc_info, dispatch_uid=exception_uid)
try: try:
try: try:
response = self.handler(environ) response = self.handler(environ)
except TemplateDoesNotExist as e: except TemplateDoesNotExist as e:
@ -493,7 +493,7 @@ class Client(RequestFactory):
return response return response
finally: finally:
signals.template_rendered.disconnect(dispatch_uid=signal_uid) signals.template_rendered.disconnect(dispatch_uid=signal_uid)
got_request_exception.disconnect(dispatch_uid="request-exception") got_request_exception.disconnect(dispatch_uid=exception_uid)
def get(self, path, data=None, follow=False, secure=False, **extra): def get(self, path, data=None, follow=False, secure=False, **extra):
""" """

View File

@ -640,6 +640,14 @@ class ClientTest(TestCase):
self.assertEqual(mail.outbox[1].to[0], 'second@example.com') self.assertEqual(mail.outbox[1].to[0], 'second@example.com')
self.assertEqual(mail.outbox[1].to[1], 'third@example.com') self.assertEqual(mail.outbox[1].to[1], 'third@example.com')
def test_exception_following_nested_client_request(self):
"""
A nested test client request shouldn't clobber exception signals from
the outer client request.
"""
with self.assertRaisesMessage(Exception, 'exception message'):
self.client.get('/nesting_exception_view/')
@override_settings( @override_settings(
MIDDLEWARE_CLASSES=['django.middleware.csrf.CsrfViewMiddleware'], MIDDLEWARE_CLASSES=['django.middleware.csrf.CsrfViewMiddleware'],

View File

@ -31,6 +31,7 @@ urlpatterns = [
url(r'^broken_view/$', views.broken_view), url(r'^broken_view/$', views.broken_view),
url(r'^mail_sending_view/$', views.mail_sending_view), url(r'^mail_sending_view/$', views.mail_sending_view),
url(r'^mass_mail_sending_view/$', views.mass_mail_sending_view), url(r'^mass_mail_sending_view/$', views.mass_mail_sending_view),
url(r'^nesting_exception_view/$', views.nesting_exception_view),
url(r'^django_project_redirect/$', views.django_project_redirect), url(r'^django_project_redirect/$', views.django_project_redirect),
url(r'^accounts/login/$', auth_views.login, {'template_name': 'login.html'}), url(r'^accounts/login/$', auth_views.login, {'template_name': 'login.html'}),

View File

@ -11,6 +11,7 @@ from django.http import (
) )
from django.shortcuts import render_to_response from django.shortcuts import render_to_response
from django.template import Context, Template from django.template import Context, Template
from django.test import Client
from django.utils.decorators import method_decorator from django.utils.decorators import method_decorator
from django.utils.six.moves.urllib.parse import urlencode from django.utils.six.moves.urllib.parse import urlencode
@ -309,5 +310,15 @@ def mass_mail_sending_view(request):
return HttpResponse("Mail sent") return HttpResponse("Mail sent")
def nesting_exception_view(request):
"""
A view that uses a nested client to call another view and then raises an
exception.
"""
client = Client()
client.get('/get_view/')
raise Exception('exception message')
def django_project_redirect(request): def django_project_redirect(request):
return HttpResponseRedirect('https://www.djangoproject.com/') return HttpResponseRedirect('https://www.djangoproject.com/')