add baize.asgi test

pyproject.toml

@@ -23,10 +23,11 @@ dev = [
 ]
 benchmark = ["uvicorn>=0.16.0", "asgiref>=3.4.1"]
 test = [
-  "pytest<8.0.0,>=7.0.1",
-  "pytest-cov<4.0.0,>=3.0.0",
-  "pytest-asyncio<1.0.0,>=0.11.0",
-  "starlette>=0.37.2",
+    "pytest<8.0.0,>=7.0.1",
+    "pytest-cov<4.0.0,>=3.0.0",
+    "pytest-asyncio<1.0.0,>=0.11.0",
+    "starlette>=0.37.2",
+    "baize>=0.20.8",
 ]
 
 [tool.pdm.build]

tests/test_asgi.py

@@ -224,3 +224,19 @@ async def dispatch(self, request, call_next):
         assert response.status_code == 200
         assert response.text == '{"hello":"world"}'
         assert response.headers["x-middleware"] == "true"
+
+
+def test_baize_stream_response():
+    from baize.asgi import StreamResponse
+
+    async def stream():
+        for i in range(10):
+            yield str(i).encode()
+
+    app = ASGIMiddleware(StreamResponse(stream()))
+    with httpx.Client(
+        transport=httpx.WSGITransport(app=app), base_url="http://testserver:80"
+    ) as client:
+        response = client.get("/")
+        assert response.status_code == 200
+        assert response.text == "0123456789"