Add Request.stream

This commit is contained in:
38elements
2017-05-05 20:09:32 +09:00
parent 7cf3d49f00
commit 9f2ba26e9d
7 changed files with 116 additions and 8 deletions

View File

@@ -0,0 +1,26 @@
from sanic import Sanic
from sanic.response import stream
app = Sanic('test_request_stream', is_request_stream=True)
@app.post('/stream')
async def handler(request):
async def sample_streaming_fn(response):
while True:
body = await request.stream.get()
if body is None:
break
body = body.decode('utf-8').replace('1', 'A')
response.write(body)
return stream(sample_streaming_fn)
def test_request_stream():
data = ""
for i in range(1, 250000):
data += str(i)
request, response = app.test_client.post('/stream', data=data)
text = data.replace('1', 'A')
assert response.status == 200
assert response.text == text