package main_test

import (
	"bytes"
	"encoding/json"
	"log/slog"
	"testing"
	"testing/slogtest"

	"github.com/rs/zerolog"
	slogzerolog "github.com/samber/slog-zerolog/v2"
)

func TestNewHandler(t *testing.T) {
	var buf bytes.Buffer

	newHandler := func(t *testing.T) slog.Handler {
		logger := zerolog.New(&buf)

		return slogzerolog.Option{
			Logger: &logger,
			// Level:  slog.LevelInfo,
		}.NewZerologHandler()
	}

	result := func(t *testing.T) map[string]any {
		defer buf.Reset()

		var m map[string]any
		if err := json.Unmarshal(buf.Bytes(), &m); err != nil {
			t.Fatal(err)
		}

		if val, ok := m["message"]; ok {
			// zerolog use "message" as the key, key replacement is needed
			m[slog.MessageKey] = val
			delete(m, "message")
		}

		if val, ok := m["time"]; ok && val == "0001-01-01T00:00:00Z" {
			// zerolog does not ignore zero time, manual removal is needed
			delete(m, "time")
		}

		return m
	}

	slogtest.Run(t, newHandler, result)
}