def test_not_exception(self, exception_handler):
"""Test ES connection exception is handled"""
exception_handler.return_value = None
result = custom_exception_handler("what??", None)
self.assertTrue(exception_handler.called)
self.assertEqual(result, None)
def test_502_error_exceptions(self, exception_handler):
"""Test ES connection exception is handled"""
exception_handler.return_value = None
result = custom_exception_handler(elasticsearch.exceptions.ConnectionError("oops"), None)
self.assertTrue(exception_handler.called)
self.assertEqual(result.status_code, 502)
def test_drf_handled_exception(self, exception_handler):
"""Test that we pass DRF recognized exceptions through unmodified"""
exception_handler.return_value = "it's handled"
result = custom_exception_handler(None, None)
self.assertTrue(exception_handler.called)
self.assertEqual(result, "it's handled")
def test_not_exception(self, exception_handler):
"""Test ES connection exception is handled"""
exception_handler.return_value = None
result = custom_exception_handler('what??', None)
self.assertTrue(exception_handler.called)
self.assertEqual(result, None)
def test_drf_handled_exception(self, exception_handler):
"""Test that we pass DRF recognized exceptions through unmodified"""
exception_handler.return_value = "it's handled"
result = custom_exception_handler(None, None)
self.assertTrue(exception_handler.called)
self.assertEqual(result, "it's handled")
def test_502_error_exceptions(self, exception_handler):
"""Test ES connection exception is handled"""
exception_handler.return_value = None
result = custom_exception_handler(
elasticsearch.exceptions.ConnectionError("oops"), None)
self.assertTrue(exception_handler.called)
self.assertEqual(result.status_code, 502)
def test_500_error_exceptions(self, exception_handler):
"""Test ES connection exception is handled"""
exception_handler.return_value = None
result = custom_exception_handler(elasticsearch.exceptions.SerializationError("oops"), None)
self.assertTrue(exception_handler.called)
self.assertEqual(result.status_code, 500)
result = custom_exception_handler(elasticsearch.exceptions.ImproperlyConfigured("oops"), None)
self.assertTrue(exception_handler.called)
self.assertEqual(result.status_code, 500)
result = custom_exception_handler(elasticsearch.exceptions.TransportError("oops"), None)
self.assertTrue(exception_handler.called)
self.assertEqual(result.status_code, 500)
exception_handler.return_value = None
result = custom_exception_handler(Exception("oops"), None)
self.assertTrue(exception_handler.called)
self.assertEqual(result.status_code, 500)
def test_500_error_exceptions(self, exception_handler):
"""Test ES connection exception is handled"""
exception_handler.return_value = None
result = custom_exception_handler(
elasticsearch.exceptions.SerializationError("oops"), None)
self.assertTrue(exception_handler.called)
self.assertEqual(result.status_code, 500)
result = custom_exception_handler(
elasticsearch.exceptions.ImproperlyConfigured("oops"), None)
self.assertTrue(exception_handler.called)
self.assertEqual(result.status_code, 500)
result = custom_exception_handler(
elasticsearch.exceptions.TransportError("oops"), None)
self.assertTrue(exception_handler.called)
self.assertEqual(result.status_code, 500)
exception_handler.return_value = None
result = custom_exception_handler(Exception("oops"), None)
self.assertTrue(exception_handler.called)
self.assertEqual(result.status_code, 500)
