54 lines
1.2 KiB
Python
54 lines
1.2 KiB
Python
from sanic import Sanic
|
|
from sanic.response import text
|
|
from sanic.exceptions import InvalidUsage, ServerError, NotFound
|
|
from sanic.utils import sanic_endpoint_test
|
|
|
|
# ------------------------------------------------------------ #
|
|
# GET
|
|
# ------------------------------------------------------------ #
|
|
|
|
exception_app = Sanic('test_exceptions')
|
|
|
|
|
|
@exception_app.route('/')
|
|
def handler(request):
|
|
return text('OK')
|
|
|
|
|
|
@exception_app.route('/error')
|
|
def handler_error(request):
|
|
raise ServerError("OK")
|
|
|
|
|
|
@exception_app.route('/404')
|
|
def handler_404(request):
|
|
raise NotFound("OK")
|
|
|
|
|
|
@exception_app.route('/invalid')
|
|
def handler_invalid(request):
|
|
raise InvalidUsage("OK")
|
|
|
|
|
|
def test_no_exception():
|
|
request, response = sanic_endpoint_test(exception_app)
|
|
assert response.status == 200
|
|
assert response.text == 'OK'
|
|
|
|
|
|
def test_server_error_exception():
|
|
request, response = sanic_endpoint_test(exception_app, uri='/error')
|
|
assert response.status == 500
|
|
|
|
|
|
def test_invalid_usage_exception():
|
|
request, response = sanic_endpoint_test(exception_app, uri='/invalid')
|
|
assert response.status == 400
|
|
|
|
|
|
def test_not_found_exception():
|
|
request, response = sanic_endpoint_test(exception_app, uri='/404')
|
|
assert response.status == 404
|
|
|
|
|