26 lines
624 B
Go
26 lines
624 B
Go
|
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))
|
||
|
|
}
|