Skip to content

Commit 15269c9

Browse files
committed
Python: Add StreamResponse test
1 parent 0f9ab8f commit 15269c9

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

python/ql/test/library-tests/frameworks/aiohttp/response_test.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,18 @@ async def file_response(request): # $ requestHandler
7676
resp = web.FileResponse(path=filename) # $ HttpResponse mimetype=application/octet-stream MISSING: getAPathArgument=filename
7777
return resp
7878

79+
80+
@routes.get("/streaming_response") # $ routeSetup="/streaming_response"
81+
async def streaming_response(request): # $ requestHandler
82+
resp = web.StreamResponse() # $ HttpResponse mimetype=application/octet-stream
83+
await resp.prepare(request)
84+
85+
await resp.write(b"foo") # $ MISSING: responseBody=b"foo"
86+
await resp.write(data=b"bar") # $ MISSING: responseBody=b"bar"
87+
await resp.write_eof(b"baz") # $ MISSING: responseBody=b"bar"
88+
89+
return resp
90+
7991
################################################################################
8092
# Cookies
8193
################################################################################

0 commit comments

Comments
 (0)