blob: d655875316bb4b97659fa42c5a21e7d5db1cf7e4 [file] [log] [blame]
// Copyright 2019 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package log
import (
"context"
"fmt"
"strings"
"testing"
"cloud.google.com/go/logging"
)
const (
debugMsg = "debugMsg"
infoMsg = "infoMsg"
errorMsg = "errorMsg"
)
// Do not run in parallel. It overrides currentLevel.
func TestSetLogLevel(t *testing.T) {
oldLevel := getLevel()
defer func() { currentLevel = oldLevel }()
tests := []struct {
name string
newLevel string
wantLevel logging.Severity
}{
{name: "default level", newLevel: "", wantLevel: logging.Default},
{name: "invalid level", newLevel: "xyz", wantLevel: logging.Default},
{name: "debug level", newLevel: "debug", wantLevel: logging.Debug},
{name: "info level", newLevel: "info", wantLevel: logging.Info},
{name: "warning level", newLevel: "warning", wantLevel: logging.Warning},
{name: "error level", newLevel: "error", wantLevel: logging.Error},
{name: "fatal level", newLevel: "fatal", wantLevel: logging.Critical},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
SetLevel(test.newLevel)
gotLevel := getLevel()
if test.wantLevel != gotLevel {
t.Errorf("Error: want=%s, got=%s", test.wantLevel, gotLevel)
}
})
}
}
// Do not run in parallel. It overrides logger with mockLogger.
func TestLogLevel(t *testing.T) {
oldLogger := logger
defer func() { logger = oldLogger }()
logger = &mockLogger{}
// logs below info(like debug) won't print
SetLevel("info")
tests := []struct {
name string
logFunc func(context.Context, any)
logMsg string
expected bool
}{
{name: "debug", logFunc: Debug, logMsg: debugMsg, expected: false},
{name: "info", logFunc: Info, logMsg: infoMsg, expected: true},
{name: "warning", logFunc: Warning, logMsg: infoMsg, expected: true},
{name: "error", logFunc: Error, logMsg: errorMsg, expected: true},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
test.logFunc(context.Background(), test.logMsg)
logs := logger.(*mockLogger).logs
got := strings.Contains(logs, test.logMsg)
if got != test.expected {
t.Errorf("expected : %v, got %v", test.expected, got)
}
})
}
}
// Do not run in parallel. It overrides logger with mockLogger.
func TestDefaultLogLevel(t *testing.T) {
oldLogger := logger
defer func() { logger = oldLogger }()
logger = &mockLogger{}
SetLevel("") // default behaviour; print everything
tests := []struct {
name string
logFunc func(context.Context, any)
logMsg string
}{
{name: "debug", logFunc: Debug, logMsg: debugMsg},
{name: "info", logFunc: Info, logMsg: infoMsg},
{name: "error", logFunc: Error, logMsg: errorMsg},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
test.logFunc(context.Background(), test.logMsg)
logs := logger.(*mockLogger).logs
if !strings.Contains(logs, test.logMsg) {
t.Errorf("%v not logged.", test.logMsg)
}
})
}
}
type mockLogger struct {
logs string
}
func (l *mockLogger) log(ctx context.Context, s logging.Severity, payload any) {
l.logs += fmt.Sprintf("%s: %+v", s, payload)
}