Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 8 additions & 3 deletions tests/test_middleware.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
import asyncio
import sys
from unittest.mock import patch

from django.contrib.auth.models import User
Expand All @@ -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
Expand Down Expand Up @@ -35,7 +40,7 @@ def test_sync_mode(self):
lambda x: HttpResponse("<html><body>Test app</body></html>")
)

self.assertFalse(asyncio.iscoroutinefunction(middleware))
self.assertFalse(iscoroutinefunction(middleware))

response = middleware(request)
self.assertEqual(response.status_code, 200)
Expand All @@ -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)
Expand Down
Loading