diff --git a/async_asgi_testclient/tests/test_testing.py b/async_asgi_testclient/tests/test_testing.py index 4e0da1c..94fd268 100644 --- a/async_asgi_testclient/tests/test_testing.py +++ b/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"]) diff --git a/test-requirements.txt b/test-requirements.txt index 8816c81..c4c1c06 100644 --- a/test-requirements.txt +++ b/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