forked from orbit-oss/flask
Fix formatting errors
This commit is contained in:
parent
7d506f2408
commit
0bac2ade91
1 changed files with 80 additions and 82 deletions
|
|
@ -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):
|
class ForbiddenSubclassRegistered(Forbidden):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
class ForbiddenSubclassUnregistered(Forbidden):
|
class ForbiddenSubclassUnregistered(Forbidden):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
@app.errorhandler(403)
|
@app.errorhandler(403)
|
||||||
def code_exception_handler(e):
|
def code_exception_handler(e):
|
||||||
assert isinstance(e, Forbidden)
|
assert isinstance(e, Forbidden)
|
||||||
return 'forbidden'
|
return 'forbidden'
|
||||||
|
|
||||||
@app.errorhandler(ForbiddenSubclassRegistered)
|
@app.errorhandler(ForbiddenSubclassRegistered)
|
||||||
def subclass_exception_handler(e):
|
def subclass_exception_handler(e):
|
||||||
assert isinstance(e, ForbiddenSubclassRegistered)
|
assert isinstance(e, ForbiddenSubclassRegistered)
|
||||||
return 'forbidden-registered'
|
return 'forbidden-registered'
|
||||||
|
|
||||||
@app.route('/forbidden')
|
@app.route('/forbidden')
|
||||||
def forbidden_test():
|
def forbidden_test():
|
||||||
raise Forbidden()
|
raise Forbidden()
|
||||||
|
|
||||||
@app.route('/forbidden-registered')
|
@app.route('/forbidden-registered')
|
||||||
def registered_test():
|
def registered_test():
|
||||||
raise ForbiddenSubclassRegistered()
|
raise ForbiddenSubclassRegistered()
|
||||||
|
|
||||||
@app.route('/forbidden-unregistered')
|
@app.route('/forbidden-unregistered')
|
||||||
def unregistered_test():
|
def unregistered_test():
|
||||||
raise ForbiddenSubclassUnregistered()
|
raise ForbiddenSubclassUnregistered()
|
||||||
|
|
||||||
|
c = app.test_client()
|
||||||
|
|
||||||
c = app.test_client()
|
assert c.get('/forbidden').data == b'forbidden'
|
||||||
|
assert c.get('/forbidden-unregistered').data == b'forbidden'
|
||||||
assert c.get('/forbidden').data == b'forbidden'
|
assert c.get('/forbidden-registered').data == b'forbidden-registered'
|
||||||
assert c.get('/forbidden-unregistered').data == b'forbidden'
|
|
||||||
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…
Add table
Add a link
Reference in a new issue