diff --git a/tests/test_middleware.py b/tests/test_middleware.py index 56081ce56..72d4e3db6 100644 --- a/tests/test_middleware.py +++ b/tests/test_middleware.py @@ -1,4 +1,4 @@ -import asyncio +import sys from unittest.mock import patch from django.contrib.auth.models import User @@ -7,6 +7,11 @@ from debug_toolbar.middleware import DebugToolbarMiddleware +if sys.version_info >= (3, 12): + from inspect import iscoroutinefunction +else: + from asyncio import iscoroutinefunction + def show_toolbar_if_staff(request): # Hit the database, but always return True @@ -35,7 +40,7 @@ def test_sync_mode(self): lambda x: HttpResponse("Test app") ) - self.assertFalse(asyncio.iscoroutinefunction(middleware)) + self.assertFalse(iscoroutinefunction(middleware)) response = middleware(request) self.assertEqual(response.status_code, 200) @@ -54,7 +59,7 @@ async def get_response(request): middleware = DebugToolbarMiddleware(get_response) request = self.async_factory.get("/") - self.assertTrue(asyncio.iscoroutinefunction(middleware)) + self.assertTrue(iscoroutinefunction(middleware)) response = await middleware(request) self.assertEqual(response.status_code, 200)