orbit-oss/flask
4
5
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add assert message for errorhandler exception type

Browse source
This commit is contained in:
jackwardell 2020-04-16 17:01:51 +01:00 • committed by David Lord
parent 64790568ed
commit 148c89bccd
2 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/test_user_error_handler.py
View file

@ -11,11 +11,19 @@ def test_error_handler_no_match(app, client):
class CustomException(Exception):
pass
class UnacceptableCustomException(BaseException):
pass
@app.errorhandler(CustomException)
def custom_exception_handler(e):
assert isinstance(e, CustomException)
return "custom"
with pytest.raises(
AssertionError, match="Custom exceptions must be subclasses of Exception."
):
app.register_error_handler(UnacceptableCustomException, None)
@app.errorhandler(500)
def handle_500(e):
assert isinstance(e, InternalServerError)