Fix warning on calling DB functions in appconfig.ready

Author: Archmonger

tests/test_app/apps.py

@@ -1,4 +1,3 @@
-import contextlib
 import sys
 
 from django.apps import AppConfig
@@ -11,8 +10,6 @@ class TestAppConfig(AppConfig):
     name = "test_app"
 
     def ready(self):
-        from django.contrib.auth.models import User
-
         register_iframe("test_app.views.view_to_iframe_sync_func")
         register_iframe(views.view_to_iframe_async_func)
         register_iframe(views.ViewToIframeSyncClass)
@@ -22,8 +19,3 @@ def ready(self):
 
         if "test" in sys.argv:
             return
-
-        with contextlib.suppress(Exception):
-            User.objects.create_superuser(
-                username="admin", email="admin@example.com", password="password"
-            )

tests/test_app/middleware.py

@@ -0,0 +1,31 @@
+import contextlib
+
+from asgiref.sync import iscoroutinefunction, markcoroutinefunction
+
+
+class AutoCreateAdminMiddleware:
+    async_capable = True
+    sync_capable = True
+
+    def __init__(self, get_response):
+        from django.contrib.auth.models import User
+
+        # One-time configuration and initialization.
+        self.get_response = get_response
+        with contextlib.suppress(Exception):
+            User.objects.create_superuser(
+                username="admin", email="admin@example.com", password="password"
+            )
+
+        if iscoroutinefunction(self.get_response):
+            markcoroutinefunction(self)
+
+    def __call__(self, request):
+        if iscoroutinefunction(self.get_response):
+
+            async def async_call():
+                return await self.get_response(request)
+
+            return async_call()
+
+        return self.get_response(request)

tests/test_app/settings_multi_db.py

@@ -33,6 +33,7 @@
 MIDDLEWARE = [
     "django.middleware.security.SecurityMiddleware",
     "servestatic.middleware.ServeStaticMiddleware",
+    "test_app.middleware.AutoCreateAdminMiddleware",
     "django.contrib.sessions.middleware.SessionMiddleware",
     "django.middleware.common.CommonMiddleware",
     "django.middleware.csrf.CsrfViewMiddleware",

tests/test_app/settings_single_db.py

@@ -33,6 +33,7 @@
 MIDDLEWARE = [
     "django.middleware.security.SecurityMiddleware",
     "servestatic.middleware.ServeStaticMiddleware",
+    "test_app.middleware.AutoCreateAdminMiddleware",
     "django.contrib.sessions.middleware.SessionMiddleware",
     "django.middleware.common.CommonMiddleware",
     "django.middleware.csrf.CsrfViewMiddleware",