package logging

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

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

func TestSetupLogger(t *testing.T) {
	for _, format := range []string{"text", "json"} {
		l := SetupLogger("info", format, "test-service", "v0.0.0")
		assert.NotNil(t, l)
		assert.Same(t, l, slog.Default())
	}
}

func TestContextLogger(t *testing.T) {
	base := SetupLogger("info", "text", "svc", "v1")
	assert.Same(t, base, LoggerFromContext(context.Background()))
	custom := slog.New(slog.NewTextHandler(nil, nil))
	ctx := ContextWithLogger(context.Background(), custom)
	assert.Same(t, custom, LoggerFromContext(ctx))
}