Ticket #32815: django-issue-32815.diff

tests/middleware_exceptions/middleware.py

@@ -1 +1 @@
 import asyncio
+import contextvars
 
 from django.http import Http404, HttpResponse
 from django.template import engines
@@ -7 +8 @@
     async_only_middleware, sync_and_async_middleware, sync_only_middleware,
 )
 
+current_context = contextvars.ContextVar('current_context')
 log = []
 
 
@@ -123 +125 @@
     pass
 
 
+@sync_and_async_middleware
+class SyncAndAsyncTemplateResponseMiddleware(BaseMiddleware):
+    def process_view(self, request, view_func, view_args, view_kwargs):
+        request.META['_CONTEXT_RESET_TOKEN'] = current_context.set(id(request))
+        return None
+
+    def process_template_response(self, request, response):
+        current_context.reset(request.META['_CONTEXT_RESET_TOKEN'])
+        response.context_data['mw'].append(self.__class__.__name__)
+        return response
+
+
 @sync_only_middleware
 class DecoratedPaymentMiddleware(PaymentMiddleware):
     pass

tests/middleware_exceptions/tests.py

@@ -300 +300 @@
         self.assertEqual(response.content, b'OK')
         self.assertEqual(response.status_code, 200)
 
+    @override_settings(MIDDLEWARE=[
+        'middleware_exceptions.middleware.SyncAndAsyncTemplateResponseMiddleware',
+    ])
+    def test_sync_process_template_response(self):
+        response = self.client.get(
+            '/middleware_exceptions/template_response/'
+        )
+        self.assertEqual(
+            response.content,
+            b'template_response OK\nSyncAndAsyncTemplateResponseMiddleware',
+        )
+
+    @override_settings(MIDDLEWARE=[
+        'middleware_exceptions.middleware.SyncAndAsyncTemplateResponseMiddleware',
+    ])
+    async def test_async_process_template_response(self):
+        response = await self.async_client.get(
+            '/middleware_exceptions/template_response/'
+        )
+        self.assertEqual(
+            response.content,
+            b'template_response OK\nSyncAndAsyncTemplateResponseMiddleware',
+        )
+
 
 @override_settings(ROOT_URLCONF='middleware_exceptions.urls')
 class AsyncMiddlewareTests(SimpleTestCase):