summaryrefslogtreecommitdiff
path: root/tests/middleware_exceptions/middleware.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/middleware_exceptions/middleware.py')
-rw-r--r--tests/middleware_exceptions/middleware.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/tests/middleware_exceptions/middleware.py b/tests/middleware_exceptions/middleware.py
index 9fb9726b50..b87da84b9d 100644
--- a/tests/middleware_exceptions/middleware.py
+++ b/tests/middleware_exceptions/middleware.py
@@ -1,3 +1,5 @@
+import asyncio
+
from django.http import Http404, HttpResponse
from django.template import engines
from django.template.response import TemplateResponse
@@ -11,6 +13,9 @@ 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
def __call__(self, request):
return self.get_response(request)