Run linter
This commit is contained in:
Adam Hopkins
@@ -11,67 +11,66 @@ data = "abc" * 10000000
|
||||
|
||||
|
||||
def test_request_stream_method_view(app):
|
||||
'''for self.is_request_stream = True'''
|
||||
"""for self.is_request_stream = True"""
|
||||
|
||||
class SimpleView(HTTPMethodView):
|
||||
|
||||
def get(self, request):
|
||||
assert request.stream is None
|
||||
return text('OK')
|
||||
return text("OK")
|
||||
|
||||
@stream_decorator
|
||||
async def post(self, request):
|
||||
assert isinstance(request.stream, StreamBuffer)
|
||||
result = ''
|
||||
result = ""
|
||||
while True:
|
||||
body = await request.stream.read()
|
||||
if body is None:
|
||||
break
|
||||
result += body.decode('utf-8')
|
||||
result += body.decode("utf-8")
|
||||
return text(result)
|
||||
|
||||
app.add_route(SimpleView.as_view(), '/method_view')
|
||||
app.add_route(SimpleView.as_view(), "/method_view")
|
||||
|
||||
assert app.is_request_stream is True
|
||||
|
||||
request, response = app.test_client.get('/method_view')
|
||||
request, response = app.test_client.get("/method_view")
|
||||
assert response.status == 200
|
||||
assert response.text == 'OK'
|
||||
assert response.text == "OK"
|
||||
|
||||
request, response = app.test_client.post('/method_view', data=data)
|
||||
request, response = app.test_client.post("/method_view", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == data
|
||||
|
||||
|
||||
def test_request_stream_app(app):
|
||||
'''for self.is_request_stream = True and decorators'''
|
||||
"""for self.is_request_stream = True and decorators"""
|
||||
|
||||
@app.get('/get')
|
||||
@app.get("/get")
|
||||
async def get(request):
|
||||
assert request.stream is None
|
||||
return text('GET')
|
||||
return text("GET")
|
||||
|
||||
@app.head('/head')
|
||||
@app.head("/head")
|
||||
async def head(request):
|
||||
assert request.stream is None
|
||||
return text('HEAD')
|
||||
return text("HEAD")
|
||||
|
||||
@app.delete('/delete')
|
||||
@app.delete("/delete")
|
||||
async def delete(request):
|
||||
assert request.stream is None
|
||||
return text('DELETE')
|
||||
return text("DELETE")
|
||||
|
||||
@app.options('/options')
|
||||
@app.options("/options")
|
||||
async def options(request):
|
||||
assert request.stream is None
|
||||
return text('OPTIONS')
|
||||
return text("OPTIONS")
|
||||
|
||||
@app.post('/_post/<id>')
|
||||
@app.post("/_post/<id>")
|
||||
async def _post(request, id):
|
||||
assert request.stream is None
|
||||
return text('_POST')
|
||||
return text("_POST")
|
||||
|
||||
@app.post('/post/<id>', stream=True)
|
||||
@app.post("/post/<id>", stream=True)
|
||||
async def post(request, id):
|
||||
assert isinstance(request.stream, StreamBuffer)
|
||||
|
||||
@@ -80,15 +79,16 @@ def test_request_stream_app(app):
|
||||
body = await request.stream.read()
|
||||
if body is None:
|
||||
break
|
||||
await response.write(body.decode('utf-8'))
|
||||
await response.write(body.decode("utf-8"))
|
||||
|
||||
return stream(streaming)
|
||||
|
||||
@app.put('/_put')
|
||||
@app.put("/_put")
|
||||
async def _put(request):
|
||||
assert request.stream is None
|
||||
return text('_PUT')
|
||||
return text("_PUT")
|
||||
|
||||
@app.put('/put', stream=True)
|
||||
@app.put("/put", stream=True)
|
||||
async def put(request):
|
||||
assert isinstance(request.stream, StreamBuffer)
|
||||
|
||||
@@ -97,15 +97,16 @@ def test_request_stream_app(app):
|
||||
body = await request.stream.read()
|
||||
if body is None:
|
||||
break
|
||||
await response.write(body.decode('utf-8'))
|
||||
await response.write(body.decode("utf-8"))
|
||||
|
||||
return stream(streaming)
|
||||
|
||||
@app.patch('/_patch')
|
||||
@app.patch("/_patch")
|
||||
async def _patch(request):
|
||||
assert request.stream is None
|
||||
return text('_PATCH')
|
||||
return text("_PATCH")
|
||||
|
||||
@app.patch('/patch', stream=True)
|
||||
@app.patch("/patch", stream=True)
|
||||
async def patch(request):
|
||||
assert isinstance(request.stream, StreamBuffer)
|
||||
|
||||
@@ -114,56 +115,57 @@ def test_request_stream_app(app):
|
||||
body = await request.stream.read()
|
||||
if body is None:
|
||||
break
|
||||
await response.write(body.decode('utf-8'))
|
||||
await response.write(body.decode("utf-8"))
|
||||
|
||||
return stream(streaming)
|
||||
|
||||
assert app.is_request_stream is True
|
||||
|
||||
request, response = app.test_client.get('/get')
|
||||
request, response = app.test_client.get("/get")
|
||||
assert response.status == 200
|
||||
assert response.text == 'GET'
|
||||
assert response.text == "GET"
|
||||
|
||||
request, response = app.test_client.head('/head')
|
||||
request, response = app.test_client.head("/head")
|
||||
assert response.status == 200
|
||||
assert response.text == ''
|
||||
assert response.text == ""
|
||||
|
||||
request, response = app.test_client.delete('/delete')
|
||||
request, response = app.test_client.delete("/delete")
|
||||
assert response.status == 200
|
||||
assert response.text == 'DELETE'
|
||||
assert response.text == "DELETE"
|
||||
|
||||
request, response = app.test_client.options('/options')
|
||||
request, response = app.test_client.options("/options")
|
||||
assert response.status == 200
|
||||
assert response.text == 'OPTIONS'
|
||||
assert response.text == "OPTIONS"
|
||||
|
||||
request, response = app.test_client.post('/_post/1', data=data)
|
||||
request, response = app.test_client.post("/_post/1", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == '_POST'
|
||||
assert response.text == "_POST"
|
||||
|
||||
request, response = app.test_client.post('/post/1', data=data)
|
||||
request, response = app.test_client.post("/post/1", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == data
|
||||
|
||||
request, response = app.test_client.put('/_put', data=data)
|
||||
request, response = app.test_client.put("/_put", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == '_PUT'
|
||||
assert response.text == "_PUT"
|
||||
|
||||
request, response = app.test_client.put('/put', data=data)
|
||||
request, response = app.test_client.put("/put", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == data
|
||||
|
||||
request, response = app.test_client.patch('/_patch', data=data)
|
||||
request, response = app.test_client.patch("/_patch", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == '_PATCH'
|
||||
assert response.text == "_PATCH"
|
||||
|
||||
request, response = app.test_client.patch('/patch', data=data)
|
||||
request, response = app.test_client.patch("/patch", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == data
|
||||
|
||||
|
||||
def test_request_stream_handle_exception(app):
|
||||
'''for handling exceptions properly'''
|
||||
"""for handling exceptions properly"""
|
||||
|
||||
@app.post('/post/<id>', stream=True)
|
||||
@app.post("/post/<id>", stream=True)
|
||||
async def post(request, id):
|
||||
assert isinstance(request.stream, StreamBuffer)
|
||||
|
||||
@@ -172,51 +174,54 @@ def test_request_stream_handle_exception(app):
|
||||
body = await request.stream.read()
|
||||
if body is None:
|
||||
break
|
||||
await response.write(body.decode('utf-8'))
|
||||
await response.write(body.decode("utf-8"))
|
||||
|
||||
return stream(streaming)
|
||||
|
||||
# 404
|
||||
request, response = app.test_client.post('/in_valid_post', data=data)
|
||||
request, response = app.test_client.post("/in_valid_post", data=data)
|
||||
assert response.status == 404
|
||||
assert response.text == 'Error: Requested URL /in_valid_post not found'
|
||||
assert response.text == "Error: Requested URL /in_valid_post not found"
|
||||
|
||||
# 405
|
||||
request, response = app.test_client.get('/post/random_id', data=data)
|
||||
request, response = app.test_client.get("/post/random_id", data=data)
|
||||
assert response.status == 405
|
||||
assert response.text == 'Error: Method GET not allowed for URL' \
|
||||
' /post/random_id'
|
||||
assert (
|
||||
response.text == "Error: Method GET not allowed for URL"
|
||||
" /post/random_id"
|
||||
)
|
||||
|
||||
|
||||
def test_request_stream_blueprint(app):
|
||||
'''for self.is_request_stream = True'''
|
||||
bp = Blueprint('test_blueprint_request_stream_blueprint')
|
||||
"""for self.is_request_stream = True"""
|
||||
bp = Blueprint("test_blueprint_request_stream_blueprint")
|
||||
|
||||
@app.get('/get')
|
||||
@app.get("/get")
|
||||
async def get(request):
|
||||
assert request.stream is None
|
||||
return text('GET')
|
||||
return text("GET")
|
||||
|
||||
@bp.head('/head')
|
||||
@bp.head("/head")
|
||||
async def head(request):
|
||||
assert request.stream is None
|
||||
return text('HEAD')
|
||||
return text("HEAD")
|
||||
|
||||
@bp.delete('/delete')
|
||||
@bp.delete("/delete")
|
||||
async def delete(request):
|
||||
assert request.stream is None
|
||||
return text('DELETE')
|
||||
return text("DELETE")
|
||||
|
||||
@bp.options('/options')
|
||||
@bp.options("/options")
|
||||
async def options(request):
|
||||
assert request.stream is None
|
||||
return text('OPTIONS')
|
||||
return text("OPTIONS")
|
||||
|
||||
@bp.post('/_post/<id>')
|
||||
@bp.post("/_post/<id>")
|
||||
async def _post(request, id):
|
||||
assert request.stream is None
|
||||
return text('_POST')
|
||||
return text("_POST")
|
||||
|
||||
@bp.post('/post/<id>', stream=True)
|
||||
@bp.post("/post/<id>", stream=True)
|
||||
async def post(request, id):
|
||||
assert isinstance(request.stream, StreamBuffer)
|
||||
|
||||
@@ -225,15 +230,16 @@ def test_request_stream_blueprint(app):
|
||||
body = await request.stream.read()
|
||||
if body is None:
|
||||
break
|
||||
await response.write(body.decode('utf-8'))
|
||||
await response.write(body.decode("utf-8"))
|
||||
|
||||
return stream(streaming)
|
||||
|
||||
@bp.put('/_put')
|
||||
@bp.put("/_put")
|
||||
async def _put(request):
|
||||
assert request.stream is None
|
||||
return text('_PUT')
|
||||
return text("_PUT")
|
||||
|
||||
@bp.put('/put', stream=True)
|
||||
@bp.put("/put", stream=True)
|
||||
async def put(request):
|
||||
assert isinstance(request.stream, StreamBuffer)
|
||||
|
||||
@@ -242,15 +248,16 @@ def test_request_stream_blueprint(app):
|
||||
body = await request.stream.read()
|
||||
if body is None:
|
||||
break
|
||||
await response.write(body.decode('utf-8'))
|
||||
await response.write(body.decode("utf-8"))
|
||||
|
||||
return stream(streaming)
|
||||
|
||||
@bp.patch('/_patch')
|
||||
@bp.patch("/_patch")
|
||||
async def _patch(request):
|
||||
assert request.stream is None
|
||||
return text('_PATCH')
|
||||
return text("_PATCH")
|
||||
|
||||
@bp.patch('/patch', stream=True)
|
||||
@bp.patch("/patch", stream=True)
|
||||
async def patch(request):
|
||||
assert isinstance(request.stream, StreamBuffer)
|
||||
|
||||
@@ -259,109 +266,109 @@ def test_request_stream_blueprint(app):
|
||||
body = await request.stream.read()
|
||||
if body is None:
|
||||
break
|
||||
await response.write(body.decode('utf-8'))
|
||||
await response.write(body.decode("utf-8"))
|
||||
|
||||
return stream(streaming)
|
||||
|
||||
app.blueprint(bp)
|
||||
|
||||
assert app.is_request_stream is True
|
||||
|
||||
request, response = app.test_client.get('/get')
|
||||
request, response = app.test_client.get("/get")
|
||||
assert response.status == 200
|
||||
assert response.text == 'GET'
|
||||
assert response.text == "GET"
|
||||
|
||||
request, response = app.test_client.head('/head')
|
||||
request, response = app.test_client.head("/head")
|
||||
assert response.status == 200
|
||||
assert response.text == ''
|
||||
assert response.text == ""
|
||||
|
||||
request, response = app.test_client.delete('/delete')
|
||||
request, response = app.test_client.delete("/delete")
|
||||
assert response.status == 200
|
||||
assert response.text == 'DELETE'
|
||||
assert response.text == "DELETE"
|
||||
|
||||
request, response = app.test_client.options('/options')
|
||||
request, response = app.test_client.options("/options")
|
||||
assert response.status == 200
|
||||
assert response.text == 'OPTIONS'
|
||||
assert response.text == "OPTIONS"
|
||||
|
||||
request, response = app.test_client.post('/_post/1', data=data)
|
||||
request, response = app.test_client.post("/_post/1", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == '_POST'
|
||||
assert response.text == "_POST"
|
||||
|
||||
request, response = app.test_client.post('/post/1', data=data)
|
||||
request, response = app.test_client.post("/post/1", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == data
|
||||
|
||||
request, response = app.test_client.put('/_put', data=data)
|
||||
request, response = app.test_client.put("/_put", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == '_PUT'
|
||||
assert response.text == "_PUT"
|
||||
|
||||
request, response = app.test_client.put('/put', data=data)
|
||||
request, response = app.test_client.put("/put", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == data
|
||||
|
||||
request, response = app.test_client.patch('/_patch', data=data)
|
||||
request, response = app.test_client.patch("/_patch", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == '_PATCH'
|
||||
assert response.text == "_PATCH"
|
||||
|
||||
request, response = app.test_client.patch('/patch', data=data)
|
||||
request, response = app.test_client.patch("/patch", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == data
|
||||
|
||||
|
||||
def test_request_stream_composition_view(app):
|
||||
'''for self.is_request_stream = True'''
|
||||
"""for self.is_request_stream = True"""
|
||||
|
||||
def get_handler(request):
|
||||
assert request.stream is None
|
||||
return text('OK')
|
||||
return text("OK")
|
||||
|
||||
async def post_handler(request):
|
||||
assert isinstance(request.stream, StreamBuffer)
|
||||
result = ''
|
||||
result = ""
|
||||
while True:
|
||||
body = await request.stream.read()
|
||||
if body is None:
|
||||
break
|
||||
result += body.decode('utf-8')
|
||||
result += body.decode("utf-8")
|
||||
return text(result)
|
||||
|
||||
view = CompositionView()
|
||||
view.add(['GET'], get_handler)
|
||||
view.add(['POST'], post_handler, stream=True)
|
||||
app.add_route(view, '/composition_view')
|
||||
view.add(["GET"], get_handler)
|
||||
view.add(["POST"], post_handler, stream=True)
|
||||
app.add_route(view, "/composition_view")
|
||||
|
||||
assert app.is_request_stream is True
|
||||
|
||||
request, response = app.test_client.get('/composition_view')
|
||||
request, response = app.test_client.get("/composition_view")
|
||||
assert response.status == 200
|
||||
assert response.text == 'OK'
|
||||
assert response.text == "OK"
|
||||
|
||||
request, response = app.test_client.post('/composition_view', data=data)
|
||||
request, response = app.test_client.post("/composition_view", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == data
|
||||
|
||||
|
||||
def test_request_stream(app):
|
||||
'''test for complex application'''
|
||||
bp = Blueprint('test_blueprint_request_stream')
|
||||
"""test for complex application"""
|
||||
bp = Blueprint("test_blueprint_request_stream")
|
||||
|
||||
class SimpleView(HTTPMethodView):
|
||||
|
||||
def get(self, request):
|
||||
assert request.stream is None
|
||||
return text('OK')
|
||||
return text("OK")
|
||||
|
||||
@stream_decorator
|
||||
async def post(self, request):
|
||||
assert isinstance(request.stream, StreamBuffer)
|
||||
result = ''
|
||||
result = ""
|
||||
while True:
|
||||
body = await request.stream.read()
|
||||
if body is None:
|
||||
break
|
||||
result += body.decode('utf-8')
|
||||
result += body.decode("utf-8")
|
||||
return text(result)
|
||||
|
||||
@app.post('/stream', stream=True)
|
||||
@app.post("/stream", stream=True)
|
||||
async def handler(request):
|
||||
assert isinstance(request.stream, StreamBuffer)
|
||||
|
||||
@@ -370,84 +377,85 @@ def test_request_stream(app):
|
||||
body = await request.stream.read()
|
||||
if body is None:
|
||||
break
|
||||
await response.write(body.decode('utf-8'))
|
||||
await response.write(body.decode("utf-8"))
|
||||
|
||||
return stream(streaming)
|
||||
|
||||
@app.get('/get')
|
||||
@app.get("/get")
|
||||
async def get(request):
|
||||
assert request.stream is None
|
||||
return text('OK')
|
||||
return text("OK")
|
||||
|
||||
@bp.post('/bp_stream', stream=True)
|
||||
@bp.post("/bp_stream", stream=True)
|
||||
async def bp_stream(request):
|
||||
assert isinstance(request.stream, StreamBuffer)
|
||||
result = ''
|
||||
result = ""
|
||||
while True:
|
||||
body = await request.stream.read()
|
||||
if body is None:
|
||||
break
|
||||
result += body.decode('utf-8')
|
||||
result += body.decode("utf-8")
|
||||
return text(result)
|
||||
|
||||
@bp.get('/bp_get')
|
||||
@bp.get("/bp_get")
|
||||
async def bp_get(request):
|
||||
assert request.stream is None
|
||||
return text('OK')
|
||||
return text("OK")
|
||||
|
||||
def get_handler(request):
|
||||
assert request.stream is None
|
||||
return text('OK')
|
||||
return text("OK")
|
||||
|
||||
async def post_handler(request):
|
||||
assert isinstance(request.stream, StreamBuffer)
|
||||
result = ''
|
||||
result = ""
|
||||
while True:
|
||||
body = await request.stream.read()
|
||||
if body is None:
|
||||
break
|
||||
result += body.decode('utf-8')
|
||||
result += body.decode("utf-8")
|
||||
return text(result)
|
||||
|
||||
app.add_route(SimpleView.as_view(), '/method_view')
|
||||
app.add_route(SimpleView.as_view(), "/method_view")
|
||||
|
||||
view = CompositionView()
|
||||
view.add(['GET'], get_handler)
|
||||
view.add(['POST'], post_handler, stream=True)
|
||||
view.add(["GET"], get_handler)
|
||||
view.add(["POST"], post_handler, stream=True)
|
||||
|
||||
app.blueprint(bp)
|
||||
|
||||
app.add_route(view, '/composition_view')
|
||||
app.add_route(view, "/composition_view")
|
||||
|
||||
assert app.is_request_stream is True
|
||||
|
||||
request, response = app.test_client.get('/method_view')
|
||||
request, response = app.test_client.get("/method_view")
|
||||
assert response.status == 200
|
||||
assert response.text == 'OK'
|
||||
assert response.text == "OK"
|
||||
|
||||
request, response = app.test_client.post('/method_view', data=data)
|
||||
request, response = app.test_client.post("/method_view", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == data
|
||||
|
||||
request, response = app.test_client.get('/composition_view')
|
||||
request, response = app.test_client.get("/composition_view")
|
||||
assert response.status == 200
|
||||
assert response.text == 'OK'
|
||||
assert response.text == "OK"
|
||||
|
||||
request, response = app.test_client.post('/composition_view', data=data)
|
||||
request, response = app.test_client.post("/composition_view", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == data
|
||||
|
||||
request, response = app.test_client.get('/get')
|
||||
request, response = app.test_client.get("/get")
|
||||
assert response.status == 200
|
||||
assert response.text == 'OK'
|
||||
assert response.text == "OK"
|
||||
|
||||
request, response = app.test_client.post('/stream', data=data)
|
||||
request, response = app.test_client.post("/stream", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == data
|
||||
|
||||
request, response = app.test_client.get('/bp_get')
|
||||
request, response = app.test_client.get("/bp_get")
|
||||
assert response.status == 200
|
||||
assert response.text == 'OK'
|
||||
assert response.text == "OK"
|
||||
|
||||
request, response = app.test_client.post('/bp_stream', data=data)
|
||||
request, response = app.test_client.post("/bp_stream", data=data)
|
||||
assert response.status == 200
|
||||
assert response.text == data
|
||||
|
||||
Reference in New Issue
Block a user