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_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)