ba9b432993
* Default error handler now only logs traceback on 500 errors and all responses are HTML formatted. * Tests passing. * Ability to flag any exception object with self.quiet = True following @ashleysommer suggestion. * Refactor HTML formatting into errorpages.py. String escapes for debug tracebacks. * Remove extra includes * Auto-set quiet flag also when decorator is used. * Cleanup, make error pages (probably) HTML5-compliant and similar for debug and non-debug modes. * Fix lookup of non-existant status codes * No logging of 503 errors after all. * lint
70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
import asyncio
|
|
|
|
from sanic import Sanic
|
|
from sanic.exceptions import ServiceUnavailable
|
|
from sanic.response import text
|
|
|
|
|
|
response_timeout_app = Sanic("test_response_timeout")
|
|
response_timeout_default_app = Sanic("test_response_timeout_default")
|
|
response_handler_cancelled_app = Sanic("test_response_handler_cancelled")
|
|
|
|
response_timeout_app.config.RESPONSE_TIMEOUT = 1
|
|
response_timeout_default_app.config.RESPONSE_TIMEOUT = 1
|
|
response_handler_cancelled_app.config.RESPONSE_TIMEOUT = 1
|
|
|
|
|
|
@response_timeout_app.route("/1")
|
|
async def handler_1(request):
|
|
await asyncio.sleep(2)
|
|
return text("OK")
|
|
|
|
|
|
@response_timeout_app.exception(ServiceUnavailable)
|
|
def handler_exception(request, exception):
|
|
return text("Response Timeout from error_handler.", 503)
|
|
|
|
|
|
def test_server_error_response_timeout():
|
|
request, response = response_timeout_app.test_client.get("/1")
|
|
assert response.status == 503
|
|
assert response.text == "Response Timeout from error_handler."
|
|
|
|
|
|
@response_timeout_default_app.route("/1")
|
|
async def handler_2(request):
|
|
await asyncio.sleep(2)
|
|
return text("OK")
|
|
|
|
|
|
def test_default_server_error_response_timeout():
|
|
request, response = response_timeout_default_app.test_client.get("/1")
|
|
assert response.status == 503
|
|
assert "Response Timeout" in response.text
|
|
|
|
|
|
response_handler_cancelled_app.flag = False
|
|
|
|
|
|
@response_handler_cancelled_app.exception(asyncio.CancelledError)
|
|
def handler_cancelled(request, exception):
|
|
# If we get a CancelledError, it means sanic has already sent a response,
|
|
# we should not ever have to handle a CancelledError.
|
|
response_handler_cancelled_app.flag = True
|
|
return text("App received CancelledError!", 500)
|
|
# The client will never receive this response, because the socket
|
|
# is already closed when we get a CancelledError.
|
|
|
|
|
|
@response_handler_cancelled_app.route("/1")
|
|
async def handler_3(request):
|
|
await asyncio.sleep(2)
|
|
return text("OK")
|
|
|
|
|
|
def test_response_handler_cancelled():
|
|
request, response = response_handler_cancelled_app.test_client.get("/1")
|
|
assert response.status == 503
|
|
assert "Response Timeout" in response.text
|
|
assert response_handler_cancelled_app.flag is False
|