saasitone/pkg/middleware/log_test.go

package middleware

import (
	"context"
	"log/slog"
	"testing"

	"github.com/labstack/echo/v4"
	echomw "github.com/labstack/echo/v4/middleware"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"git.grosinger.net/tgrosinger/saasitone/pkg/log"
	"git.grosinger.net/tgrosinger/saasitone/pkg/tests"
)

type mockLogHandler struct {
	msg   string
	level string
	group string
	attr  []slog.Attr
}

func (m *mockLogHandler) Enabled(_ context.Context, l slog.Level) bool {
	return true
}

func (m *mockLogHandler) Handle(_ context.Context, r slog.Record) error {
	m.level = r.Level.String()
	m.msg = r.Message
	return nil
}

func (m *mockLogHandler) WithAttrs(as []slog.Attr) slog.Handler {
	if m.attr == nil {
		m.attr = make([]slog.Attr, 0)
	}
	m.attr = append(m.attr, as...)
	return m
}

func (m *mockLogHandler) WithGroup(name string) slog.Handler {
	m.group = name
	return m
}

func (m *mockLogHandler) GetAttr(key string) string {
	if m.attr == nil {
		return ""
	}
	for _, attr := range m.attr {
		if attr.Key == key {
			return attr.Value.String()
		}
	}

	return ""
}

func TestLogRequestID(t *testing.T) {
	ctx, _ := tests.NewContext(c.Web, "/")

	h := new(mockLogHandler)
	logger := slog.New(h)
	log.Set(ctx, logger)

	require.NoError(t, tests.ExecuteMiddleware(ctx, echomw.RequestID()))
	require.NoError(t, tests.ExecuteMiddleware(ctx, SetLogger()))

	log.Ctx(ctx).Info("test")
	rID := ctx.Response().Header().Get(echo.HeaderXRequestID)
	assert.Equal(t, rID, h.GetAttr("request_id"))
}

func TestLogRequest(t *testing.T) {
	statusCode := 200
	h := new(mockLogHandler)

	exec := func() {
		ctx, _ := tests.NewContext(c.Web, "http://test.localhost/abc?d=1&e=2")
		logger := slog.New(h).With("previous", "param")
		log.Set(ctx, logger)
		ctx.Request().Header.Set("Referer", "ref.com")
		ctx.Request().Header.Set(echo.HeaderXRealIP, "21.12.12.21")

		require.NoError(t, tests.ExecuteHandler(ctx, func(ctx echo.Context) error {
			return ctx.String(statusCode, "hello")
		},
			SetLogger(),
			LogRequest(),
		))
	}

	exec()
	assert.Equal(t, "param", h.GetAttr("previous"))
	assert.Equal(t, "21.12.12.21", h.GetAttr("ip"))
	assert.Equal(t, "test.localhost", h.GetAttr("host"))
	assert.Equal(t, "ref.com", h.GetAttr("referer"))
	assert.Equal(t, "200", h.GetAttr("status"))
	assert.Equal(t, "0", h.GetAttr("bytes_in"))
	assert.Equal(t, "5", h.GetAttr("bytes_out"))
	assert.NotEmpty(t, h.GetAttr("latency"))
	assert.Equal(t, "INFO", h.level)
	assert.Equal(t, "GET /abc?d=1&e=2", h.msg)

	statusCode = 500
	exec()
	assert.Equal(t, "ERROR", h.level)
}