errors_test.go

package simba_test

import (
	"bytes"
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"log/slog"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"

	"github.com/sillen102/simba"
	"github.com/sillen102/simba/simbaContext"
	"gotest.tools/v3/assert"
)

func TestHandleError(t *testing.T) {
	t.Parallel()

	t.Run("log wrapped error", func(t *testing.T) {
		req := httptest.NewRequest(http.MethodGet, "/test", nil)
		w := httptest.NewRecorder()

		logBuffer := &bytes.Buffer{}
		logger := slog.New(slog.NewTextHandler(logBuffer, &slog.HandlerOptions{}))
		ctx := context.WithValue(req.Context(), simbaContext.LoggerKey, logger)
		req = req.WithContext(ctx)

		simba.HandleError(w, req, simba.WrapError(
			http.StatusInternalServerError,
			fmt.Errorf("outermost error: %w", fmt.Errorf("wrapping error: %w", errors.New("original error"))),
			"Internal server error"))

		assert.Equal(t, http.StatusInternalServerError, w.Code)
		assert.Equal(t, "application/json", w.Header().Get("Content-Type"))

		var errorResponse simba.ErrorResponse
		err := json.NewDecoder(w.Body).Decode(&errorResponse)
		assert.NilError(t, err)
		assert.Equal(t, http.StatusInternalServerError, errorResponse.Status)
		assert.Equal(t, "Internal server error", errorResponse.Message)

		expectedLog := "wrapping error: original error"
		assert.Assert(t, strings.Contains(logBuffer.String(), expectedLog))
	})

	t.Run("unauthorized does not show wrapped error", func(t *testing.T) {
		req := httptest.NewRequest(http.MethodGet, "/test", nil)
		w := httptest.NewRecorder()

		logBuffer := &bytes.Buffer{}
		logger := slog.New(slog.NewTextHandler(logBuffer, &slog.HandlerOptions{}))
		ctx := context.WithValue(req.Context(), simbaContext.LoggerKey, logger)
		req = req.WithContext(ctx)

		simba.HandleError(w, req, simba.WrapError(http.StatusUnauthorized, errors.New("wrapped error"), "Internal server error"))

		assert.Equal(t, http.StatusUnauthorized, w.Code)
		assert.Equal(t, "application/json", w.Header().Get("Content-Type"))

		var errorResponse simba.ErrorResponse
		err := json.NewDecoder(w.Body).Decode(&errorResponse)
		assert.NilError(t, err)
		assert.Equal(t, http.StatusUnauthorized, errorResponse.Status)
		assert.Equal(t, "unauthorized", errorResponse.Message) // hide details of the error

		expectedLog := "wrapped error"
		assert.Assert(t, strings.Contains(logBuffer.String(), expectedLog))
	})
}