package gormkit

import (
	"bytes"
	"context"
	"errors"
	"log/slog"
	"testing"
	"time"

	"gorm.io/gorm/logger"
)

func TestTraceOnlyFormatsQueriesThatWillBeLogged(t *testing.T) {
	for _, tc := range []struct {
		name     string
		gorm     logger.LogLevel
		app      slog.Level
		duration time.Duration
		err      error
		want     bool
	}{
		{"silent", logger.Silent, slog.LevelInfo, time.Second, errors.New("failure"), false},
		{"successful error-only", logger.Error, slog.LevelInfo, time.Second, nil, false},
		{"fast warning-only", logger.Warn, slog.LevelInfo, time.Millisecond, nil, false},
		{"not-found warning-only", logger.Warn, slog.LevelInfo, time.Millisecond, logger.ErrRecordNotFound, false},
		{"application filter", logger.Info, slog.LevelError, time.Millisecond, nil, false},
		{"slow", logger.Warn, slog.LevelInfo, time.Second, nil, true},
		{"error", logger.Error, slog.LevelInfo, time.Millisecond, errors.New("failure"), true},
		{"info", logger.Info, slog.LevelInfo, time.Millisecond, nil, true},
	} {
		t.Run(tc.name, func(t *testing.T) {
			var buffer bytes.Buffer
			log := NewLogger(slog.New(slog.NewJSONHandler(&buffer, &slog.HandlerOptions{Level: tc.app})), tc.gorm)
			calls := 0
			log.Trace(context.Background(), time.Now().Add(-tc.duration), func() (string, int64) {
				calls++
				return "SELECT 1", 1
			}, tc.err)
			if (calls == 1) != tc.want || (buffer.Len() > 0) != tc.want {
				t.Fatalf("format calls=%d, output=%q, want logging=%v", calls, buffer.String(), tc.want)
			}
		})
	}
}