From 32d70b2f55b1f74736fd11bc8efce890ad5fa2f0 Mon Sep 17 00:00:00 2001 From: Carlton Gibson Date: Tue, 20 Dec 2022 11:10:48 +0100 Subject: Refs #34118 -- Adopted asgiref coroutine detection shims. Thanks to Mariusz Felisiak for review. --- tests/middleware_exceptions/middleware.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) (limited to 'tests/middleware_exceptions') diff --git a/tests/middleware_exceptions/middleware.py b/tests/middleware_exceptions/middleware.py index 5a0c82afb1..f50aa61327 100644 --- a/tests/middleware_exceptions/middleware.py +++ b/tests/middleware_exceptions/middleware.py @@ -1,4 +1,4 @@ -import asyncio +from asgiref.sync import iscoroutinefunction, markcoroutinefunction from django.http import Http404, HttpResponse from django.template import engines @@ -15,9 +15,8 @@ log = [] class BaseMiddleware: def __init__(self, get_response): self.get_response = get_response - if asyncio.iscoroutinefunction(self.get_response): - # Mark the class as async-capable. - self._is_coroutine = asyncio.coroutines._is_coroutine + if iscoroutinefunction(self.get_response): + markcoroutinefunction(self) def __call__(self, request): return self.get_response(request) -- cgit v1.3