Add some test coverage
This commit is contained in:
@@ -30,6 +30,23 @@ def test_middleware_request(app):
|
||||
assert type(results[0]) is Request
|
||||
|
||||
|
||||
def test_middleware_request_as_convenience(app):
|
||||
results = []
|
||||
|
||||
@app.on_request
|
||||
async def handler1(request):
|
||||
results.append(request)
|
||||
|
||||
@app.route("/")
|
||||
async def handler2(request):
|
||||
return text("OK")
|
||||
|
||||
request, response = app.test_client.get("/")
|
||||
|
||||
assert response.text == "OK"
|
||||
assert type(results[0]) is Request
|
||||
|
||||
|
||||
def test_middleware_response(app):
|
||||
results = []
|
||||
|
||||
@@ -54,6 +71,54 @@ def test_middleware_response(app):
|
||||
assert isinstance(results[2], HTTPResponse)
|
||||
|
||||
|
||||
def test_middleware_response_as_convenience(app):
|
||||
results = []
|
||||
|
||||
@app.on_request
|
||||
async def process_request(request):
|
||||
results.append(request)
|
||||
|
||||
@app.on_response
|
||||
async def process_response(request, response):
|
||||
results.append(request)
|
||||
results.append(response)
|
||||
|
||||
@app.route("/")
|
||||
async def handler(request):
|
||||
return text("OK")
|
||||
|
||||
request, response = app.test_client.get("/")
|
||||
|
||||
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_as_convenience_called(app):
|
||||
results = []
|
||||
|
||||
@app.on_request()
|
||||
async def process_request(request):
|
||||
results.append(request)
|
||||
|
||||
@app.on_response()
|
||||
async def process_response(request, response):
|
||||
results.append(request)
|
||||
results.append(response)
|
||||
|
||||
@app.route("/")
|
||||
async def handler(request):
|
||||
return text("OK")
|
||||
|
||||
request, response = app.test_client.get("/")
|
||||
|
||||
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": "middleware not run"}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user