diff options
Diffstat (limited to 'tests/middleware_exceptions/middleware.py')
-rw-r--r-- | tests/middleware_exceptions/middleware.py | 35 |
1 files changed, 22 insertions, 13 deletions
diff --git a/tests/middleware_exceptions/middleware.py b/tests/middleware_exceptions/middleware.py index b87da84b9d..5a0c82afb1 100644 --- a/tests/middleware_exceptions/middleware.py +++ b/tests/middleware_exceptions/middleware.py @@ -4,7 +4,9 @@ from django.http import Http404, HttpResponse from django.template import engines from django.template.response import TemplateResponse from django.utils.decorators import ( - async_only_middleware, sync_and_async_middleware, sync_only_middleware, + async_only_middleware, + sync_and_async_middleware, + sync_only_middleware, ) log = [] @@ -23,58 +25,64 @@ class BaseMiddleware: class ProcessExceptionMiddleware(BaseMiddleware): def process_exception(self, request, exception): - return HttpResponse('Exception caught') + return HttpResponse("Exception caught") @async_only_middleware class AsyncProcessExceptionMiddleware(BaseMiddleware): async def process_exception(self, request, exception): - return HttpResponse('Exception caught') + return HttpResponse("Exception caught") class ProcessExceptionLogMiddleware(BaseMiddleware): def process_exception(self, request, exception): - log.append('process-exception') + log.append("process-exception") class ProcessExceptionExcMiddleware(BaseMiddleware): def process_exception(self, request, exception): - raise Exception('from process-exception') + raise Exception("from process-exception") class ProcessViewMiddleware(BaseMiddleware): def process_view(self, request, view_func, view_args, view_kwargs): - return HttpResponse('Processed view %s' % view_func.__name__) + return HttpResponse("Processed view %s" % view_func.__name__) @async_only_middleware class AsyncProcessViewMiddleware(BaseMiddleware): async def process_view(self, request, view_func, view_args, view_kwargs): - return HttpResponse('Processed view %s' % view_func.__name__) + return HttpResponse("Processed view %s" % view_func.__name__) class ProcessViewNoneMiddleware(BaseMiddleware): def process_view(self, request, view_func, view_args, view_kwargs): - log.append('processed view %s' % view_func.__name__) + log.append("processed view %s" % view_func.__name__) return None class ProcessViewTemplateResponseMiddleware(BaseMiddleware): def process_view(self, request, view_func, view_args, view_kwargs): - template = engines['django'].from_string('Processed view {{ view }}{% for m in mw %}\n{{ m }}{% endfor %}') - return TemplateResponse(request, template, {'mw': [self.__class__.__name__], 'view': view_func.__name__}) + template = engines["django"].from_string( + "Processed view {{ view }}{% for m in mw %}\n{{ m }}{% endfor %}" + ) + return TemplateResponse( + request, + template, + {"mw": [self.__class__.__name__], "view": view_func.__name__}, + ) class TemplateResponseMiddleware(BaseMiddleware): def process_template_response(self, request, response): - response.context_data['mw'].append(self.__class__.__name__) + response.context_data["mw"].append(self.__class__.__name__) return response @async_only_middleware class AsyncTemplateResponseMiddleware(BaseMiddleware): async def process_template_response(self, request, response): - response.context_data['mw'].append(self.__class__.__name__) + response.context_data["mw"].append(self.__class__.__name__) return response @@ -98,7 +106,7 @@ class AsyncNoTemplateResponseMiddleware(BaseMiddleware): class NotFoundMiddleware(BaseMiddleware): def __call__(self, request): - raise Http404('not found') + raise Http404("not found") class PaymentMiddleware(BaseMiddleware): @@ -130,6 +138,7 @@ class DecoratedPaymentMiddleware(PaymentMiddleware): class NotSyncOrAsyncMiddleware(BaseMiddleware): """Middleware that is deliberately neither sync or async.""" + sync_capable = False async_capable = False |