add tests for async locale middleware

Author: amirreza8002

tests/test_middlewares/test_locale.py

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

tests/test_middlewares/test_middleware_mixin.py

@@ -6,6 +6,7 @@
 from django.http.response import HttpResponse
 
 from django_async_extensions.middleware.base import AsyncMiddlewareMixin
+from django_async_extensions.middleware.locale import AsyncLocaleMiddleware
 from django_async_extensions.middleware.security import AsyncSecurityMiddleware
 
 req = HttpResponse()
@@ -33,6 +34,7 @@ async def process_request(self, request):
 class TestMiddlewareMixin:
     middlewares = [
         AsyncSecurityMiddleware,
+        AsyncLocaleMiddleware,
     ]
 
     def test_repr(self):

tests/test_middlewares/urls.py

@@ -1,7 +1,18 @@
+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 _
+
 
 from . import views
 
+
+async def stream_http_generator():
+    yield _("Yes")
+    yield "/"
+    yield _("No")
+
+
 urlpatterns = [
     path("middleware_exceptions/view/", views.normal_view),
     path("middleware_exceptions/error/", views.server_error),
@@ -14,3 +25,8 @@
         views.async_exception_in_render,
     ),
 ]
+
+urlpatterns += i18n_patterns(
+    path("simple/", lambda r: HttpResponse()),
+    path("streaming/", lambda r: StreamingHttpResponse(stream_http_generator())),
+)