diff options
Diffstat (limited to 'tests/middleware_exceptions')
| -rw-r--r-- | tests/middleware_exceptions/middleware.py | 7 |
1 files changed, 3 insertions, 4 deletions
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) |