added tests for async locale middleware

Author: amirreza8002

tests/test_middlewares/test_locale.py

@@ -0,0 +1,42 @@
+import pytest
+
+from django.test import AsyncClient
+
+
+client = AsyncClient()
+
+
+@pytest.mark.django_db
+class TestLocaleMiddleware:
+    @pytest.fixture(autouse=True)
+    def set_settings(self, settings):
+        old_use_i18n = settings.USE_I18N
+        old_language = settings.LANGUAGES
+        old_middleware = settings.MIDDLEWARE
+        old_urlconf = settings.ROOT_URLCONF
+
+        settings.USE_I18N = True
+        settings.LANGUAGES = [("en", "English"), ("fr", "French")]
+        settings.MIDDLEWARE = [
+            "django_async_extensions.amiddleware.locale.AsyncLocaleMiddleware",
+            "django.middleware.common.CommonMiddleware",
+        ]
+        settings.ROOT_URLCONF = "test_middlewares.urls"
+
+        yield settings
+
+        settings.USE_I18N = old_use_i18n
+        settings.LANGUAGES = old_language
+        settings.MIDDLEWARE = old_middleware
+        settings.ROOT_URLCONF = old_urlconf
+
+    async def test_streaming_response(self):
+        # Regression test for #5241
+        response = await client.get("/fr/streaming/")
+        assert b"Oui/Non" in b"".join(
+            [content async for content in response.streaming_content]
+        )
+        response = await client.get("/en/streaming/")
+        assert b"Yes/No" in b"".join(
+            [content async for content in response.streaming_content]
+        )

tests/test_middlewares/urls.py

@@ -0,0 +1,16 @@
+from django.conf.urls.i18n import i18n_patterns
+from django.http import HttpResponse, StreamingHttpResponse
+from django.urls import path
+from django.utils.translation import gettext_lazy as _
+
+
+async def stream_http_generator():
+    yield _("Yes")
+    yield "/"
+    yield _("No")
+
+
+urlpatterns = i18n_patterns(
+    path("simple/", lambda r: HttpResponse()),
+    path("streaming/", lambda r: StreamingHttpResponse(stream_http_generator())),
+)