def test_bad_request_custom_msg(self):
"""Test BadRequest/400 exception with a custom error message passed as a kwarg."""
exceptions = AddExceptions(self.app, statsd=MagicMock())
bad_request = exceptions.bad_request(message='Oh noes!')
self.assertIsInstance(bad_request, extension.BadRequest)
self.assertDictEqual(bad_request.to_dict(), {'message': 'Oh noes!'})
exceptions.statsd.incr.assert_called_once_with(
extension.DEFAULT_PREFIX + '.400')
def test_conflict(self):
"""Test Conflict/409 exception."""
exceptions = AddExceptions(self.app, statsd=MagicMock())
conflict = exceptions.conflict()
self.assertIsInstance(conflict, extension.Conflict)
self.assertDictEqual(conflict.to_dict(), {'message': 'Conflict'})
exceptions.statsd.incr.assert_called_once_with(
extension.DEFAULT_PREFIX + '.409')
def test_gone(self):
"""Test Gone/410 exception."""
exceptions = AddExceptions(self.app, statsd=MagicMock())
gone = exceptions.gone()
self.assertIsInstance(gone, extension.Gone)
self.assertDictEqual(gone.to_dict(), {'message': 'Gone'})
exceptions.statsd.incr.assert_called_once_with(
extension.DEFAULT_PREFIX + '.410')
def test_forbidden(self):
"""Test Forbidden/403 exception."""
exceptions = AddExceptions(self.app, statsd=MagicMock())
forbidden = exceptions.forbidden()
self.assertIsInstance(forbidden, extension.Forbidden)
self.assertDictEqual(forbidden.to_dict(), {'message': 'Forbidden'})
exceptions.statsd.incr.assert_called_once_with(
extension.DEFAULT_PREFIX + '.403')
def test_bad_request(self):
"""Test BadRequest/400 exception."""
exceptions = AddExceptions(self.app, statsd=MagicMock())
bad_request = exceptions.bad_request()
self.assertIsInstance(bad_request, extension.BadRequest)
self.assertDictEqual(bad_request.to_dict(),
{'message': 'Invalid request parameters'})
exceptions.statsd.incr.assert_called_once_with(
extension.DEFAULT_PREFIX + '.400')
def test_failed_dependency(self):
"""Test FailedDependency/424 exception."""
exceptions = AddExceptions(self.app, statsd=MagicMock())
failed_dependency = exceptions.failed_dependency()
self.assertIsInstance(failed_dependency, extension.FailedDependency)
self.assertDictEqual(failed_dependency.to_dict(),
{'message': 'Failed Dependency'})
exceptions.statsd.incr.assert_called_once_with(
extension.DEFAULT_PREFIX + '.424')
def test_unsupported_media(self):
"""Test UnsupportedMedia/415 exception."""
exceptions = AddExceptions(self.app, statsd=MagicMock())
unsupported_media = exceptions.unsupported_media()
self.assertIsInstance(unsupported_media, extension.UnsupportedMedia)
self.assertDictEqual(unsupported_media.to_dict(),
{'message': 'Unsupported Media'})
exceptions.statsd.incr.assert_called_once_with(
extension.DEFAULT_PREFIX + '.415')
def test_not_found(self):
"""Test NotFound/404 exception."""
exceptions = AddExceptions(self.app, statsd=MagicMock())
not_found = exceptions.not_found()
self.assertIsInstance(not_found, extension.NotFound)
self.assertDictEqual(not_found.to_dict(),
{'message': 'Resource not found'})
exceptions.statsd.incr.assert_called_once_with(
extension.DEFAULT_PREFIX + '.404')
def test_unauthorized(self):
"""Test Unauthorized/401 exception."""
exceptions = AddExceptions(self.app, statsd=MagicMock())
unauthorized = exceptions.unauthorized()
self.assertIsInstance(unauthorized, extension.Unauthorized)
self.assertDictEqual(unauthorized.to_dict(),
{'message': 'Unauthorized'})
exceptions.statsd.incr.assert_called_once_with(
extension.DEFAULT_PREFIX + '.401')
def test_bad_request_no_msg(self):
"""Test BadRequest/400 exception with no message."""
self.app.config['EXCEPTION_MESSAGE'] = False
exceptions = AddExceptions(self.app, statsd=MagicMock())
bad_request = exceptions.bad_request()
self.assertIsInstance(bad_request, extension.BadRequest)
self.assertDictEqual(bad_request.to_dict(), {})
exceptions.statsd.incr.assert_called_once_with(
extension.DEFAULT_PREFIX + '.400')
def test_unprocessable_entity(self):
"""Test UnprocessableEntity/422 exception."""
exceptions = AddExceptions(self.app, statsd=MagicMock())
unprocessable_entity = exceptions.unprocessable_entity()
self.assertIsInstance(unprocessable_entity,
extension.UnprocessableEntity)
self.assertDictEqual(unprocessable_entity.to_dict(),
{'message': 'Unprocessable Entity'})
exceptions.statsd.incr.assert_called_once_with(
extension.DEFAULT_PREFIX + '.422')
def test_init_app_kwarg_config(self):
"""Test custom configs passed via kwargs to init_app."""
config = {
'EXCEPTION_MESSAGE': False,
'EXCEPTION_PREFIX': 'foo',
}
exceptions = AddExceptions()
exceptions.init_app(self.app, config)
self.assertEqual(False, exceptions.messages)
self.assertEqual('foo', exceptions.prefix)
self.assertEqual(None, exceptions.statsd)
def test_custom_app_config(self):
"""Test custom configs set on app."""
self.app.config['EXCEPTION_MESSAGE'] = False
self.app.config['EXCEPTION_PREFIX'] = 'foo'
exceptions = AddExceptions(self.app, statsd=MagicMock())
self.assertEqual(False, exceptions.messages)
self.assertEqual('foo', exceptions.prefix)
self.assertIsInstance(exceptions.statsd, MagicMock)
def test_init_kwarg_config(self):
"""Test custom configs passed via kwargs."""
config = {
'EXCEPTION_MESSAGE': False,
'EXCEPTION_PREFIX': 'foo',
}
exceptions = AddExceptions(self.app, config, statsd=MagicMock())
self.assertEqual(False, exceptions.messages)
self.assertEqual('foo', exceptions.prefix)
self.assertIsInstance(exceptions.statsd, MagicMock)
def test_default_config(self):
"""Test the default configs."""
exceptions = AddExceptions(self.app)
self.assertEqual(True, exceptions.messages)
self.assertEqual(extension.DEFAULT_PREFIX, exceptions.prefix)
self.assertEqual(None, exceptions.statsd)
def create_app():
"""Create a Flask app for context."""
app = Flask(__name__)
exceptions = AddExceptions()
exceptions.init_app(app)
return app
