117 lines
3.4 KiB
Python
117 lines
3.4 KiB
Python
import pytest
|
|
|
|
import falcon
|
|
from falcon import testing
|
|
|
|
|
|
def capture_error(ex, req, resp, params):
|
|
resp.status = falcon.HTTP_723
|
|
resp.body = 'error: %s' % str(ex)
|
|
|
|
|
|
def handle_error_first(ex, req, resp, params):
|
|
resp.status = falcon.HTTP_200
|
|
resp.body = 'first error handler'
|
|
|
|
|
|
class CustomBaseException(Exception):
|
|
pass
|
|
|
|
|
|
class CustomException(CustomBaseException):
|
|
|
|
@staticmethod
|
|
def handle(ex, req, resp, params):
|
|
raise falcon.HTTPError(
|
|
falcon.HTTP_792,
|
|
u'Internet crashed!',
|
|
u'Catastrophic weather event',
|
|
href=u'http://example.com/api/inconvenient-truth',
|
|
href_text=u'Drill, baby drill!')
|
|
|
|
|
|
class ErroredClassResource(object):
|
|
|
|
def on_get(self, req, resp):
|
|
raise Exception('Plain Exception')
|
|
|
|
def on_head(self, req, resp):
|
|
raise CustomBaseException('CustomBaseException')
|
|
|
|
def on_delete(self, req, resp):
|
|
raise CustomException('CustomException')
|
|
|
|
|
|
@pytest.fixture
|
|
def client():
|
|
app = falcon.API()
|
|
app.add_route('/', ErroredClassResource())
|
|
return testing.TestClient(app)
|
|
|
|
|
|
class TestErrorHandler(object):
|
|
|
|
def test_caught_error(self, client):
|
|
client.app.add_error_handler(Exception, capture_error)
|
|
|
|
result = client.simulate_get()
|
|
assert result.text == 'error: Plain Exception'
|
|
|
|
result = client.simulate_head()
|
|
assert result.status_code == 723
|
|
assert not result.content
|
|
|
|
def test_uncaught_error(self, client):
|
|
client.app.add_error_handler(CustomException, capture_error)
|
|
with pytest.raises(Exception):
|
|
client.simulate_get()
|
|
|
|
def test_uncaught_error_else(self, client):
|
|
with pytest.raises(Exception):
|
|
client.simulate_get()
|
|
|
|
def test_converted_error(self, client):
|
|
client.app.add_error_handler(CustomException)
|
|
|
|
result = client.simulate_delete()
|
|
assert result.status_code == 792
|
|
assert result.json[u'title'] == u'Internet crashed!'
|
|
|
|
def test_handle_not_defined(self, client):
|
|
with pytest.raises(AttributeError):
|
|
client.app.add_error_handler(CustomBaseException)
|
|
|
|
def test_subclass_error(self, client):
|
|
client.app.add_error_handler(CustomBaseException, capture_error)
|
|
|
|
result = client.simulate_delete()
|
|
assert result.status_code == 723
|
|
assert result.text == 'error: CustomException'
|
|
|
|
def test_error_order_duplicate(self, client):
|
|
client.app.add_error_handler(Exception, capture_error)
|
|
client.app.add_error_handler(Exception, handle_error_first)
|
|
|
|
result = client.simulate_get()
|
|
assert result.text == 'first error handler'
|
|
|
|
def test_error_order_subclass(self, client):
|
|
client.app.add_error_handler(Exception, capture_error)
|
|
client.app.add_error_handler(CustomException, handle_error_first)
|
|
|
|
result = client.simulate_delete()
|
|
assert result.status_code == 200
|
|
assert result.text == 'first error handler'
|
|
|
|
result = client.simulate_get()
|
|
assert result.status_code == 723
|
|
assert result.text == 'error: Plain Exception'
|
|
|
|
def test_error_order_subclass_masked(self, client):
|
|
client.app.add_error_handler(CustomException, handle_error_first)
|
|
client.app.add_error_handler(Exception, capture_error)
|
|
|
|
result = client.simulate_delete()
|
|
assert result.status_code == 723
|
|
assert result.text == 'error: CustomException'
|