Fix upload / download stream tests

async_asgi_testclient/tests/test_testing.py

@@ -459,11 +459,13 @@ async def test_request_stream(starlette_app):
     from starlette.responses import StreamingResponse
 
     async def up_stream(request):
-        async def gen():
-            async for chunk in request.stream():
+        chunks = [chunk async for chunk in request.stream()]
+
+        async def gen(chunks):
+            for chunk in chunks:
                 yield chunk
 
-        return StreamingResponse(gen())
+        return StreamingResponse(gen(chunks))
 
     starlette_app.add_route("/upload_stream", up_stream, methods=["POST"])
 
@@ -492,11 +494,13 @@ def gen():
         return StreamingResponse(gen())
 
     async def up_stream(request):
-        async def gen():
-            async for chunk in request.stream():
+        chunks = [chunk async for chunk in request.stream()]
+
+        async def gen(chunks):
+            for chunk in chunks:
                 yield chunk
 
-        return StreamingResponse(gen())
+        return StreamingResponse(gen(chunks))
 
     starlette_app.add_route("/download_stream", down_stream, methods=["GET"])
     starlette_app.add_route("/upload_stream", up_stream, methods=["POST"])

test-requirements.txt

@@ -1,5 +1,5 @@
 quart==0.17.0; python_version >= '3.7'
-starlette==0.12.13
+starlette==0.21.0
 python-multipart==0.0.5
 pytest==6.2.5
 pytest-asyncio==0.15.0