feat: disable error handler engine option (#342)

go-fuego · Jan 12, 2025 · 35d6595 · 35d6595
1 parent d6300a0
commit 35d6595
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/engine.go b/engine.go
@@ -82,6 +82,13 @@ func WithErrorHandler(errorHandler func(err error) error) func(*Engine) {
 	}
 }
 
+// DisableErrorHandler overrides ErrorHandler with a simple pass-through
+func DisableErrorHandler() func(*Engine) {
+	return func(e *Engine) {
+		e.ErrorHandler = func(err error) error { return err }
+	}
+}
+
 // OutputOpenAPISpec takes the OpenAPI spec and outputs it to a JSON file
 func (e *Engine) OutputOpenAPISpec() []byte {
 	e.OpenAPI.computeTags()

diff --git a/engine_test.go b/engine_test.go
@@ -37,4 +37,13 @@ func TestWithErrorHandler(t *testing.T) {
 			)
 		})
 	})
+
+	t.Run("disable error handler", func(t *testing.T) {
+		e := NewEngine(DisableErrorHandler())
+		err := NotFoundError{
+			Err: errors.New("Not Found"),
+		}
+		errResponse := e.ErrorHandler(err)
+		require.Equal(t, "Not Found", errResponse.Error())
+	})
 }