Fix Live Server Test Case on Django 5.2+

.github/workflows/tests.yml

@@ -28,6 +28,12 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
 
+      - name: Set up Chrome
+        uses: browser-actions/setup-chrome@v1
+
+      - name: Set up ChromeDriver
+        uses: nanasess/setup-chromedriver@v2
+
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip wheel setuptools

channels/testing/live.py

@@ -18,6 +18,13 @@ def make_application(*, static_wrapper):
     return application
 
 
+def set_database_connection():
+    from django.conf import settings
+
+    test_db_name = settings.DATABASES["default"]["TEST"]["NAME"]
+    settings.DATABASES["default"]["NAME"] = test_db_name
+
+
 class ChannelsLiveServerTestCase(TransactionTestCase):
     """
     Does basically the same as TransactionTestCase but also launches a
@@ -39,36 +46,48 @@ def live_server_url(self):
     def live_server_ws_url(self):
         return "ws://%s:%s" % (self.host, self._port)
 
-    def _pre_setup(self):
+    @classmethod
+    def setUpClass(cls):
         for connection in connections.all():
-            if self._is_in_memory_db(connection):
+            if cls._is_in_memory_db(connection):
                 raise ImproperlyConfigured(
                     "ChannelLiveServerTestCase can not be used with in memory databases"
                 )
 
-        super(ChannelsLiveServerTestCase, self)._pre_setup()
+        super().setUpClass()
 
-        self._live_server_modified_settings = modify_settings(
-            ALLOWED_HOSTS={"append": self.host}
+        cls._live_server_modified_settings = modify_settings(
+            ALLOWED_HOSTS={"append": cls.host}
         )
-        self._live_server_modified_settings.enable()
+        cls._live_server_modified_settings.enable()
 
         get_application = partial(
             make_application,
-            static_wrapper=self.static_wrapper if self.serve_static else None,
+            static_wrapper=cls.static_wrapper if cls.serve_static else None,
         )
-        self._server_process = self.ProtocolServerProcess(self.host, get_application)
-        self._server_process.start()
-        self._server_process.ready.wait()
-        self._port = self._server_process.port.value
-
-    def _post_teardown(self):
-        self._server_process.terminate()
-        self._server_process.join()
-        self._live_server_modified_settings.disable()
-        super(ChannelsLiveServerTestCase, self)._post_teardown()
-
-    def _is_in_memory_db(self, connection):
+        cls._server_process = cls.ProtocolServerProcess(
+            cls.host,
+            get_application,
+            setup=set_database_connection,
+        )
+        cls._server_process.start()
+        while True:
+            if not cls._server_process.ready.wait(timeout=1):
+                if cls._server_process.is_alive():
+                    continue
+                raise RuntimeError("Server stopped") from None
+            break
+        cls._port = cls._server_process.port.value
+
+    @classmethod
+    def tearDownClass(cls):
+        cls._server_process.terminate()
+        cls._server_process.join()
+        cls._live_server_modified_settings.disable()
+        super().tearDownClass()
+
+    @classmethod
+    def _is_in_memory_db(cls, connection):
         """
         Check if DatabaseWrapper holds in memory database.
         """

setup.cfg

@@ -42,6 +42,7 @@ tests =
     pytest
     pytest-django
     pytest-asyncio
+    selenium
 daphne =
     daphne>=4.0.0
 
@@ -53,6 +54,8 @@ exclude =
 exclude = venv/*,tox/*,docs/*,testproject/*,build/*
 max-line-length = 88
 extend-ignore = E203, W503
+per-file-ignores =
+    tests/sample_project/config/asgi.py:E402
 
 [isort]
 profile = black

tests/conftest.py

@@ -1,26 +1,12 @@
+import os
+
 import pytest
 from django.conf import settings
 
 
 def pytest_configure():
-    settings.configure(
-        DATABASES={
-            "default": {
-                "ENGINE": "django.db.backends.sqlite3",
-                # Override Django’s default behaviour of using an in-memory database
-                # in tests for SQLite, since that avoids connection.close() working.
-                "TEST": {"NAME": "test_db.sqlite3"},
-            }
-        },
-        INSTALLED_APPS=[
-            "django.contrib.auth",
-            "django.contrib.contenttypes",
-            "django.contrib.sessions",
-            "django.contrib.admin",
-            "channels",
-        ],
-        SECRET_KEY="Not_a_secret_key",
-    )
+    os.environ["DJANGO_SETTINGS_MODULE"] = "tests.sample_project.config.settings"
+    settings._setup()
 
 
 def pytest_generate_tests(metafunc):

tests/sample_project/config/asgi.py

@@ -0,0 +1,37 @@
+"""
+ASGI config for sample_project project.
+
+It exposes the ASGI callable as a module-level variable named ``application``.
+
+For more information on this file, see
+https://docs.djangoproject.com/en/5.2/howto/deployment/asgi/
+"""
+
+from django.core.asgi import get_asgi_application
+from django.urls import path
+
+application = get_asgi_application()
+
+from channels.auth import AuthMiddlewareStack
+from channels.routing import ProtocolTypeRouter, URLRouter
+from channels.security.websocket import AllowedHostsOriginValidator
+from tests.sample_project.sampleapp.consumers import LiveMessageConsumer
+
+application = ProtocolTypeRouter(
+    {
+        "websocket": AllowedHostsOriginValidator(
+            AuthMiddlewareStack(
+                URLRouter(
+                    [
+                        path(
+                            "ws/message/",
+                            LiveMessageConsumer.as_asgi(),
+                            name="live_message_counter",
+                        ),
+                    ]
+                )
+            )
+        ),
+        "http": application,
+    }
+)

tests/sample_project/config/settings.py

@@ -0,0 +1,99 @@
+from pathlib import Path
+
+BASE_DIR = Path(__file__).resolve().parent.parent
+
+SECRET_KEY = "Not_a_secret_key"
+
+DEBUG = True
+
+ALLOWED_HOSTS = []
+
+INSTALLED_APPS = [
+    "daphne",
+    "django.contrib.admin",
+    "django.contrib.auth",
+    "django.contrib.contenttypes",
+    "django.contrib.sessions",
+    "django.contrib.messages",
+    "django.contrib.staticfiles",
+    "tests.sample_project.sampleapp",
+    "channels",
+]
+
+MIDDLEWARE = [
+    "django.middleware.security.SecurityMiddleware",
+    "django.contrib.sessions.middleware.SessionMiddleware",
+    "django.middleware.common.CommonMiddleware",
+    "django.middleware.csrf.CsrfViewMiddleware",
+    "django.contrib.auth.middleware.AuthenticationMiddleware",
+    "django.contrib.messages.middleware.MessageMiddleware",
+    "django.middleware.clickjacking.XFrameOptionsMiddleware",
+]
+
+ROOT_URLCONF = "tests.sample_project.config.urls"
+
+TEMPLATES = [
+    {
+        "BACKEND": "django.template.backends.django.DjangoTemplates",
+        "DIRS": [],
+        "APP_DIRS": True,
+        "OPTIONS": {
+            "context_processors": [
+                "django.template.context_processors.csrf",
+                "django.template.context_processors.request",
+                "django.contrib.auth.context_processors.auth",
+                "django.contrib.messages.context_processors.messages",
+            ],
+        },
+    },
+]
+
+WSGI_APPLICATION = "tests.sample_project.config.wsgi.application"
+ASGI_APPLICATION = "tests.sample_project.config.asgi.application"
+
+CHANNEL_LAYERS = {
+    "default": {
+        "BACKEND": "channels.layers.InMemoryChannelLayer",
+    },
+}
+
+DATABASES = {
+    "default": {
+        "ENGINE": "django.db.backends.sqlite3",
+        "NAME": BASE_DIR / "sampleapp/sampleapp.sqlite3",
+        # Override Django’s default behaviour of using an in-memory database
+        # in tests for SQLite, since that avoids connection.close() working.
+        "TEST": {"NAME": "test_db.sqlite3"},
+    }
+}
+
+
+AUTH_PASSWORD_VALIDATORS = [
+    {
+        "NAME": (
+            "django.contrib.auth.password_validation."
+            "UserAttributeSimilarityValidator"
+        ),
+    },
+    {
+        "NAME": "django.contrib.auth.password_validation.MinimumLengthValidator",
+    },
+    {
+        "NAME": "django.contrib.auth.password_validation.CommonPasswordValidator",
+    },
+    {
+        "NAME": "django.contrib.auth.password_validation.NumericPasswordValidator",
+    },
+]
+
+LANGUAGE_CODE = "en-us"
+
+TIME_ZONE = "UTC"
+
+USE_I18N = True
+
+USE_TZ = True
+
+STATIC_URL = "static/"
+
+DEFAULT_AUTO_FIELD = "django.db.models.BigAutoField"

tests/sample_project/config/urls.py

@@ -0,0 +1,31 @@
+"""
+URL configuration for sample_project project.
+
+The `urlpatterns` list routes URLs to views. For more information please see:
+    https://docs.djangoproject.com/en/5.2/topics/http/urls/
+Examples:
+Function views
+    1. Add an import:  from my_app import views
+    2. Add a URL to urlpatterns:  path('', views.home, name='home')
+Class-based views
+    1. Add an import:  from other_app.views import Home
+    2. Add a URL to urlpatterns:  path('', Home.as_view(), name='home')
+Including another URLconf
+    1. Import the include() function: from django.urls import include, path
+    2. Add a URL to urlpatterns:  path('blog/', include('blog.urls'))
+"""
+
+from django.conf import settings
+from django.contrib import admin
+from django.urls import path
+from django.views.generic import RedirectView
+
+urlpatterns = [
+    path("admin/", admin.site.urls),
+    path(
+        "favicon.ico",
+        RedirectView.as_view(
+            url=settings.STATIC_URL + "sampleapp/images/django.svg", permanent=True
+        ),
+    ),
+]

tests/sample_project/config/wsgi.py

@@ -0,0 +1,16 @@
+"""
+WSGI config for sample_project project.
+
+It exposes the WSGI callable as a module-level variable named ``application``.
+
+For more information on this file, see
+https://docs.djangoproject.com/en/5.2/howto/deployment/wsgi/
+"""
+
+import os
+
+from django.core.wsgi import get_wsgi_application
+
+os.environ.setdefault("DJANGO_SETTINGS_MODULE", "config.settings")
+
+application = get_wsgi_application()

tests/sample_project/manage.py

@@ -0,0 +1,22 @@
+#!/usr/bin/env python
+"""Django's command-line utility for administrative tasks."""
+import os
+import sys
+
+
+def main():
+    """Run administrative tasks."""
+    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "config.settings")
+    try:
+        from django.core.management import execute_from_command_line
+    except ImportError as exc:
+        raise ImportError(
+            "Couldn't import Django. Are you sure it's installed and "
+            "available on your PYTHONPATH environment variable? Did you "
+            "forget to activate a virtual environment?"
+        ) from exc
+    execute_from_command_line(sys.argv)
+
+
+if __name__ == "__main__":
+    main()

tests/sample_project/sampleapp/admin.py

@@ -0,0 +1,9 @@
+from django.contrib import admin
+
+from .models import Message
+
+
+@admin.register(Message)
+class MessageAdmin(admin.ModelAdmin):
+    list_display = ("title", "created")
+    change_list_template = "admin/sampleapp/message/change_list.html"

tests/sample_project/sampleapp/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class SampleappConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "tests.sample_project.sampleapp"