27 lines
726 B
Python
27 lines
726 B
Python
|
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
|