def test_html_handler():
app = flask.Flask(__name__, template_folder='../{{ cookiecutter.project_slug }}/templates')
register_error_handler(app, html_handler)
@app.route('/good')
def good():
return 'OK'
@app.route('/418')
def teapot():
flask.abort(418)
@app.route('/internal')
def divide_by_zero():
1/0
with app.test_client() as client:
rv = client.get('/good')
assert rv.status_code == 200
rv = client.get('/418')
assert b'<title>{{ cookiecutter.project_name }} - I'm a teapot</title>' in rv.data
assert b'>Error 418</h1>' in rv.data
assert rv.status_code == 418
rv = client.get('/internal')
assert b'<title>{{ cookiecutter.project_name }} - Internal Server Error</title>' in rv.data
assert b'>Error 500</h1>' in rv.data
assert rv.status_code == 500
def register_request_handler(app):
""" 注册请求处理器 """
# 注册错误请求处理函数
register_error_handler(app)
@app.before_request
def before_request_callback():
# FIXME: 添加你想要执行的操作
pass
@app.after_request
def after_request_callback(response):
# FIXME: 添加你想要执行的操作
return response
def test_blueprint_handler():
app = flask.Flask(__name__)
bp = flask.Blueprint('bp', __name__)
def app_handler(e):
return 'App: ' + e.name + '\n', e.code
register_error_handler(app, app_handler)
def bp_handler(e):
return 'Blueprint: ' + e.name + '\n', e.code
register_error_handler(bp, bp_handler)
@bp.route('/good')
def good():
return 'OK'
@bp.route('/418')
def teapot():
flask.abort(418)
@bp.route('/500')
def internal():
raise werkzeug.exceptions.InternalServerError()
@bp.route('/internal')
def divide_by_zero():
1/0
app.register_blueprint(bp)
with app.test_client() as client:
rv = client.get('/good')
assert rv.data == b'OK'
assert rv.status_code == 200
rv = client.get('/418')
assert rv.data == b'Blueprint: I\'m a teapot\n'
assert rv.status_code == 418
rv = client.get('/500')
assert rv.data == b'App: Internal Server Error\n'
assert rv.status_code == 500
rv = client.get('/internal')
assert rv.data == b'App: Internal Server Error\n'
assert rv.status_code == 500
def test_custom_handler():
app = flask.Flask(__name__)
def handler(e):
return e.name + '\n', e.code
register_error_handler(app, handler)
@app.route('/good')
def good():
return 'OK'
@app.route('/418')
def teapot():
flask.abort(418)
@app.route('/500')
def internal():
raise werkzeug.exceptions.InternalServerError()
@app.route('/internal')
def divide_by_zero():
1/0
with app.test_client() as client:
rv = client.get('/good')
assert rv.status_code == 200
rv = client.get('/418')
assert rv.data == b'I\'m a teapot\n'
assert rv.status_code == 418
rv = client.get('/500')
assert rv.data == b'Internal Server Error\n'
assert rv.status_code == 500
rv = client.get('/internal')
assert rv.data == b'Internal Server Error\n'
assert rv.status_code == 500
