You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
1.8 KiB
Go
88 lines
1.8 KiB
Go
package logging
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"log/slog"
|
|
"testing"
|
|
|
|
"git.runcible.io/learning/pulley/internal/assert"
|
|
)
|
|
|
|
func TestParseLevel(t *testing.T) {
|
|
|
|
testTable := []struct {
|
|
name string
|
|
level string
|
|
want slog.Level
|
|
}{
|
|
{
|
|
name: "Default",
|
|
level: "",
|
|
want: slog.LevelInfo,
|
|
},
|
|
{
|
|
name: "Lowercase",
|
|
level: "debug",
|
|
want: slog.LevelDebug,
|
|
},
|
|
{
|
|
name: "Mixed case",
|
|
level: "wArN",
|
|
want: slog.LevelWarn,
|
|
},
|
|
{
|
|
name: "Calitalized",
|
|
level: "ERROR",
|
|
want: slog.LevelError,
|
|
},
|
|
{
|
|
name: "Info level",
|
|
level: "info",
|
|
want: slog.LevelInfo,
|
|
},
|
|
}
|
|
|
|
for _, test := range testTable {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
got := parseLogLevel(test.level)
|
|
|
|
assert.Equal(t, got, test.want)
|
|
})
|
|
}
|
|
|
|
}
|
|
|
|
func TestInitLogging(t *testing.T) {
|
|
spy := bytes.Buffer{}
|
|
msg := "Observability"
|
|
|
|
logger := InitLogging("info", &spy, false)
|
|
logger.Info(msg)
|
|
|
|
content := spy.String()
|
|
assert.StringContains(t, content, msg)
|
|
|
|
}
|
|
|
|
func TestLoggerFromCtx(t *testing.T) {
|
|
defaultLogger := slog.Default()
|
|
subLogger := defaultLogger.With("testLogger", "this is a test")
|
|
|
|
t.Run("test logger returned is default", func(t *testing.T) {
|
|
assert.Equal(t, LoggerFromCtx(context.TODO()), defaultLogger)
|
|
})
|
|
|
|
t.Run("test logger returns saved logger", func(t *testing.T) {
|
|
assert.Equal(t, LoggerFromCtx(context.WithValue(context.TODO(), CtxKeyLogger, subLogger)), subLogger)
|
|
})
|
|
|
|
t.Run("test default logger when nil", func(t *testing.T) {
|
|
assert.Equal(t, LoggerFromCtx(context.WithValue(context.TODO(), CtxKeyLogger, nil)), defaultLogger)
|
|
})
|
|
|
|
t.Run("test default logger when wrong type", func(t *testing.T) {
|
|
assert.Equal(t, LoggerFromCtx(context.WithValue(context.TODO(), CtxKeyLogger, "bad type")), defaultLogger)
|
|
})
|
|
}
|