def test_html_handler(): app = flask.Flask(__name__, template_folder='../expenses/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>Expenses - 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>Expenses - Internal Server Error</title>' in rv.data assert b'>Error 500</h1>' in rv.data assert rv.status_code == 500
def test_html_handler(): app = flask.Flask(__name__, template_folder='../expenses/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>Expenses - 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>Expenses - Internal Server Error</title>' in rv.data assert b'>Error 500</h1>' in rv.data assert rv.status_code == 500
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_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
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