60 lines
1.9 KiB
Go
60 lines
1.9 KiB
Go
package logging
|
|
|
|
import (
|
|
"bytes"
|
|
"log"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestIsLogLevelEnabled(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
loggerLevel string
|
|
messageLevel string
|
|
want bool
|
|
}{
|
|
{name: "debug logger accepts debug", loggerLevel: "debug", messageLevel: "debug", want: true},
|
|
{name: "debug logger accepts info", loggerLevel: "debug", messageLevel: "info", want: true},
|
|
{name: "info logger rejects debug", loggerLevel: "info", messageLevel: "debug", want: false},
|
|
{name: "info logger accepts info", loggerLevel: "info", messageLevel: "info", want: true},
|
|
{name: "warn logger rejects info", loggerLevel: "warn", messageLevel: "info", want: false},
|
|
{name: "warn logger accepts error", loggerLevel: "warn", messageLevel: "error", want: true},
|
|
{name: "error logger accepts only error", loggerLevel: "error", messageLevel: "error", want: true},
|
|
{name: "error logger rejects warn", loggerLevel: "error", messageLevel: "warn", want: false},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
logger := NewServiceLogger(tt.loggerLevel)
|
|
if got := logger.isLogLevelEnabled(tt.messageLevel); got != tt.want {
|
|
t.Fatalf("isLogLevelEnabled(%q) = %v, want %v", tt.messageLevel, got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDebug_NotEmittedWhenLoggerLevelInfo(t *testing.T) {
|
|
logger := NewServiceLogger("info")
|
|
var buf bytes.Buffer
|
|
logger.out = log.New(&buf, "", 0)
|
|
|
|
logger.Debug("service", "debug message", map[string]string{"k": "v"})
|
|
|
|
if buf.Len() != 0 {
|
|
t.Fatalf("expected no output for debug at info level, got: %s", buf.String())
|
|
}
|
|
}
|
|
|
|
func TestLog_UppercaseDebug_NotEmittedWhenLoggerLevelInfo(t *testing.T) {
|
|
logger := NewServiceLogger("info")
|
|
var buf bytes.Buffer
|
|
logger.out = log.New(&buf, "", 0)
|
|
|
|
logger.Log("service", "DEBUG", "debug message", nil)
|
|
|
|
if strings.TrimSpace(buf.String()) != "" {
|
|
t.Fatalf("expected no output for uppercase DEBUG at info level, got: %s", buf.String())
|
|
}
|
|
}
|