Run linter

This commit is contained in:
Adam Hopkins
2018-12-30 13:18:06 +02:00
parent 040468755c
commit 05dd3b2e9d
47 changed files with 3092 additions and 2852 deletions

View File

@@ -9,6 +9,7 @@ from sanic.response import HTTPResponse, text
# GET
# ------------------------------------------------------------ #
def test_middleware_request(app):
results = []
@@ -16,168 +17,164 @@ def test_middleware_request(app):
async def handler1(request):
results.append(request)
@app.route('/')
@app.route("/")
async def handler2(request):
return text('OK')
return text("OK")
request, response = app.test_client.get('/')
request, response = app.test_client.get("/")
assert response.text == 'OK'
assert response.text == "OK"
assert type(results[0]) is Request
def test_middleware_response(app):
results = []
@app.middleware('request')
@app.middleware("request")
async def process_request(request):
results.append(request)
@app.middleware('response')
@app.middleware("response")
async def process_response(request, response):
results.append(request)
results.append(response)
@app.route('/')
@app.route("/")
async def handler(request):
return text('OK')
return text("OK")
request, response = app.test_client.get('/')
request, response = app.test_client.get("/")
assert response.text == 'OK'
assert response.text == "OK"
assert type(results[0]) is Request
assert type(results[1]) is Request
assert isinstance(results[2], HTTPResponse)
def test_middleware_response_exception(app):
result = {'status_code': None}
result = {"status_code": None}
@app.middleware('response')
@app.middleware("response")
async def process_response(request, response):
result['status_code'] = response.status
result["status_code"] = response.status
return response
@app.exception(NotFound)
async def error_handler(request, exception):
return text('OK', exception.status_code)
return text("OK", exception.status_code)
@app.route('/')
@app.route("/")
async def handler(request):
return text('FAIL')
return text("FAIL")
request, response = app.test_client.get('/page_not_found')
assert response.text == 'OK'
assert result['status_code'] == 404
request, response = app.test_client.get("/page_not_found")
assert response.text == "OK"
assert result["status_code"] == 404
def test_middleware_response_raise_cancelled_error(app, caplog):
@app.middleware('response')
@app.middleware("response")
async def process_response(request, response):
raise CancelledError('CancelledError at response middleware')
raise CancelledError("CancelledError at response middleware")
@app.get('/')
@app.get("/")
def handler(request):
return text('OK')
return text("OK")
with caplog.at_level(logging.ERROR):
reqrequest, response = app.test_client.get('/')
reqrequest, response = app.test_client.get("/")
assert response.status == 503
assert caplog.record_tuples[0] == (
'sanic.root',
"sanic.root",
logging.ERROR,
'Exception occurred while handling uri: \'http://127.0.0.1:42101/\''
"Exception occurred while handling uri: 'http://127.0.0.1:42101/'",
)
def test_middleware_response_raise_exception(app, caplog):
@app.middleware('response')
@app.middleware("response")
async def process_response(request, response):
raise Exception('Exception at response middleware')
raise Exception("Exception at response middleware")
with caplog.at_level(logging.ERROR):
reqrequest, response = app.test_client.get('/')
reqrequest, response = app.test_client.get("/")
assert response.status == 404
assert caplog.record_tuples[0] == (
'sanic.root',
"sanic.root",
logging.ERROR,
'Exception occurred while handling uri: \'http://127.0.0.1:42101/\''
"Exception occurred while handling uri: 'http://127.0.0.1:42101/'",
)
assert caplog.record_tuples[1] == (
'sanic.error',
"sanic.error",
logging.ERROR,
'Exception occurred in one of response middleware handlers'
"Exception occurred in one of response middleware handlers",
)
def test_middleware_override_request(app):
@app.middleware
async def halt_request(request):
return text('OK')
return text("OK")
@app.route('/')
@app.route("/")
async def handler(request):
return text('FAIL')
return text("FAIL")
response = app.test_client.get('/', gather_request=False)
response = app.test_client.get("/", gather_request=False)
assert response.status == 200
assert response.text == 'OK'
assert response.text == "OK"
def test_middleware_override_response(app):
@app.middleware('response')
@app.middleware("response")
async def process_response(request, response):
return text('OK')
return text("OK")
@app.route('/')
@app.route("/")
async def handler(request):
return text('FAIL')
return text("FAIL")
request, response = app.test_client.get('/')
request, response = app.test_client.get("/")
assert response.status == 200
assert response.text == 'OK'
assert response.text == "OK"
def test_middleware_order(app):
order = []
@app.middleware('request')
@app.middleware("request")
async def request1(request):
order.append(1)
@app.middleware('request')
@app.middleware("request")
async def request2(request):
order.append(2)
@app.middleware('request')
@app.middleware("request")
async def request3(request):
order.append(3)
@app.middleware('response')
@app.middleware("response")
async def response1(request, response):
order.append(6)
@app.middleware('response')
@app.middleware("response")
async def response2(request, response):
order.append(5)
@app.middleware('response')
@app.middleware("response")
async def response3(request, response):
order.append(4)
@app.route('/')
@app.route("/")
async def handler(request):
return text('OK')
return text("OK")
request, response = app.test_client.get('/')
request, response = app.test_client.get("/")
assert response.status == 200
assert order == [1, 2, 3, 4, 5, 6]