engine_test.go

package fuego

import (
	"errors"
	"fmt"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestWithErrorHandler(t *testing.T) {
	t.Run("default engine", func(t *testing.T) {
		e := NewEngine()
		err := NotFoundError{
			Err: errors.New("Not Found :c"),
		}
		errResponse := e.ErrorHandler(err)
		require.ErrorAs(t, errResponse, &HTTPError{})
	})
	t.Run("custom handler", func(t *testing.T) {
		e := NewEngine(
			WithErrorHandler(func(err error) error {
				return fmt.Errorf("%w foobar", err)
			}),
		)
		err := NotFoundError{
			Err: errors.New("Not Found :c"),
		}
		errResponse := e.ErrorHandler(err)
		require.ErrorAs(t, errResponse, &HTTPError{})
		require.ErrorContains(t, errResponse, "Not Found :c foobar")
	})
	t.Run("should be fatal", func(t *testing.T) {
		require.Panics(t, func() {
			NewEngine(
				WithErrorHandler(nil),
			)
		})
	})

	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())
	})

	t.Run("nil returning handler", func(t *testing.T) {
		e := NewEngine(
			WithErrorHandler(func(err error) error {
				return nil
			}),
		)
		err := NotFoundError{
			Err: errors.New("Not Found"),
		}
		errResponse := e.ErrorHandler(err)
		require.Nil(t, errResponse, "error handler can return nil, which might lead to unexpected behavior")
	})
}

func TestWithRequestContentType(t *testing.T) {
	t.Run("base", func(t *testing.T) {
		e := NewEngine()
		require.Nil(t, e.requestContentTypes)
	})

	t.Run("input", func(t *testing.T) {
		arr := []string{"application/json", "application/xml"}
		e := NewEngine(WithRequestContentType("application/json", "application/xml"))
		require.ElementsMatch(t, arr, e.requestContentTypes)
	})

	t.Run("ensure applied to route", func(t *testing.T) {
		s := NewServer(WithEngineOptions(
			WithRequestContentType("application/json", "application/xml")),
		)
		route := Post(s, "/test", dummyController)

		content := route.Operation.RequestBody.Value.Content
		require.NotNil(t, content["application/json"])
		assert.Equal(t, "#/components/schemas/ReqBody", content["application/json"].Schema.Ref)

		require.NotNil(t, content["application/xml"])
		assert.Equal(t, "#/components/schemas/ReqBody", content["application/xml"].Schema.Ref)

		require.Nil(t, content["application/x-yaml"])

		_, ok := s.OpenAPI.Description().Components.RequestBodies["ReqBody"]
		require.False(t, ok)
	})
}