Browse Source

Fix formatting errors

pull/1431/head
Markus Unterwaditzer 10 years ago
parent
commit
0bac2ade91
  1. 162
      tests/test_user_error_handler.py

162
tests/test_user_error_handler.py

@ -4,110 +4,108 @@ import flask
def test_error_handler_subclass(): def test_error_handler_subclass():
app = flask.Flask(__name__) app = flask.Flask(__name__)
class ParentException(Exception): class ParentException(Exception):
pass pass
class ChildExceptionUnregistered(ParentException): class ChildExceptionUnregistered(ParentException):
pass pass
class ChildExceptionRegistered(ParentException): class ChildExceptionRegistered(ParentException):
pass pass
@app.errorhandler(ParentException) @app.errorhandler(ParentException)
def parent_exception_handler(e): def parent_exception_handler(e):
assert isinstance(e, ParentException) assert isinstance(e, ParentException)
return 'parent' return 'parent'
@app.errorhandler(ChildExceptionRegistered) @app.errorhandler(ChildExceptionRegistered)
def child_exception_handler(e): def child_exception_handler(e):
assert isinstance(e, ChildExceptionRegistered) assert isinstance(e, ChildExceptionRegistered)
return 'child-registered' return 'child-registered'
@app.route('/parent') @app.route('/parent')
def parent_test(): def parent_test():
raise ParentException() raise ParentException()
@app.route('/child-unregistered') @app.route('/child-unregistered')
def unregistered_test(): def unregistered_test():
raise ChildExceptionUnregistered() raise ChildExceptionUnregistered()
@app.route('/child-registered') @app.route('/child-registered')
def registered_test(): def registered_test():
raise ChildExceptionRegistered() raise ChildExceptionRegistered()
c = app.test_client()
c = app.test_client() assert c.get('/parent').data == b'parent'
assert c.get('/child-unregistered').data == b'parent'
assert c.get('/parent').data == b'parent' assert c.get('/child-registered').data == b'child-registered'
assert c.get('/child-unregistered').data == b'parent'
assert c.get('/child-registered').data == b'child-registered'
def test_error_handler_http_subclass(): def test_error_handler_http_subclass():
app = flask.Flask(__name__) app = flask.Flask(__name__)
class ForbiddenSubclassRegistered(Forbidden):
pass
class ForbiddenSubclassUnregistered(Forbidden): class ForbiddenSubclassRegistered(Forbidden):
pass pass
@app.errorhandler(403) class ForbiddenSubclassUnregistered(Forbidden):
def code_exception_handler(e): pass
assert isinstance(e, Forbidden)
return 'forbidden'
@app.errorhandler(ForbiddenSubclassRegistered) @app.errorhandler(403)
def subclass_exception_handler(e): def code_exception_handler(e):
assert isinstance(e, ForbiddenSubclassRegistered) assert isinstance(e, Forbidden)
return 'forbidden-registered' return 'forbidden'
@app.route('/forbidden') @app.errorhandler(ForbiddenSubclassRegistered)
def forbidden_test(): def subclass_exception_handler(e):
raise Forbidden() assert isinstance(e, ForbiddenSubclassRegistered)
return 'forbidden-registered'
@app.route('/forbidden-registered') @app.route('/forbidden')
def registered_test(): def forbidden_test():
raise ForbiddenSubclassRegistered() raise Forbidden()
@app.route('/forbidden-unregistered') @app.route('/forbidden-registered')
def unregistered_test(): def registered_test():
raise ForbiddenSubclassUnregistered() raise ForbiddenSubclassRegistered()
@app.route('/forbidden-unregistered')
def unregistered_test():
raise ForbiddenSubclassUnregistered()
c = app.test_client() c = app.test_client()
assert c.get('/forbidden').data == b'forbidden' assert c.get('/forbidden').data == b'forbidden'
assert c.get('/forbidden-unregistered').data == b'forbidden' assert c.get('/forbidden-unregistered').data == b'forbidden'
assert c.get('/forbidden-registered').data == b'forbidden-registered' assert c.get('/forbidden-registered').data == b'forbidden-registered'
def test_error_handler_blueprint(): def test_error_handler_blueprint():
bp = flask.Blueprint('bp', __name__) bp = flask.Blueprint('bp', __name__)
@bp.errorhandler(500) @bp.errorhandler(500)
def bp_exception_handler(e): def bp_exception_handler(e):
return 'bp-error' return 'bp-error'
@bp.route('/error') @bp.route('/error')
def bp_test(): def bp_test():
raise InternalServerError() raise InternalServerError()
app = flask.Flask(__name__) app = flask.Flask(__name__)
@app.errorhandler(500) @app.errorhandler(500)
def app_exception_handler(e): def app_exception_handler(e):
return 'app-error' return 'app-error'
@app.route('/error') @app.route('/error')
def app_test(): def app_test():
raise InternalServerError() raise InternalServerError()
app.register_blueprint(bp, url_prefix='/bp') app.register_blueprint(bp, url_prefix='/bp')
c = app.test_client() c = app.test_client()
assert c.get('/error').data == b'app-error' assert c.get('/error').data == b'app-error'
assert c.get('/bp/error').data == b'bp-error' assert c.get('/bp/error').data == b'bp-error'

Loading…
Cancel
Save