mirror of
https://github.com/OpenDiablo2/OpenDiablo2
synced 2024-09-06 11:34:15 -04:00
116 lines
2.5 KiB
Go
116 lines
2.5 KiB
Go
|
package d2common
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
type testWriter struct {
|
||
|
data []byte
|
||
|
}
|
||
|
|
||
|
func (tw *testWriter) Write(msg []byte) (int, error) {
|
||
|
tw.data = msg
|
||
|
|
||
|
return len(msg), nil
|
||
|
}
|
||
|
|
||
|
func Test_logger_SetLevel(t *testing.T) {
|
||
|
l := &Logger{Writer: &testWriter{}}
|
||
|
|
||
|
tests := []struct {
|
||
|
level LogLevel
|
||
|
}{
|
||
|
{LogLevelNone},
|
||
|
{LogLevelError},
|
||
|
{LogLevelWarning},
|
||
|
{LogLevelInfo},
|
||
|
{LogLevelDebug},
|
||
|
}
|
||
|
|
||
|
for idx := range tests {
|
||
|
targetLevel := tests[idx].level
|
||
|
l.SetLevel(targetLevel)
|
||
|
|
||
|
if l.level != targetLevel {
|
||
|
t.Error("unexpected log level")
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func Test_logger_LogLevels(t *testing.T) {
|
||
|
w := &testWriter{}
|
||
|
l := &Logger{Writer: w}
|
||
|
|
||
|
noMessage := ""
|
||
|
message := "test"
|
||
|
expectedError := fmt.Sprintf(LogFmtError, message)
|
||
|
expectedWarning := fmt.Sprintf(LogFmtWarning, message)
|
||
|
expectedInfo := fmt.Sprintf(LogFmtInfo, message)
|
||
|
expectedDebug := fmt.Sprintf(LogFmtDebug, message)
|
||
|
|
||
|
// for each log level we set, we will use different log methods (info, warning, etc) and check
|
||
|
// what the output in the writer is (clearing the writer data before each test)
|
||
|
tests := []struct {
|
||
|
logLevel LogLevel
|
||
|
expect map[LogLevel]string
|
||
|
}{
|
||
|
{LogLevelDebug, map[LogLevel]string{
|
||
|
LogLevelError: expectedError,
|
||
|
LogLevelWarning: expectedWarning,
|
||
|
LogLevelInfo: expectedInfo,
|
||
|
LogLevelDebug: expectedDebug,
|
||
|
}},
|
||
|
{LogLevelInfo, map[LogLevel]string{
|
||
|
LogLevelError: expectedError,
|
||
|
LogLevelWarning: expectedWarning,
|
||
|
LogLevelInfo: expectedInfo,
|
||
|
LogLevelDebug: noMessage,
|
||
|
}},
|
||
|
{LogLevelWarning, map[LogLevel]string{
|
||
|
LogLevelError: expectedError,
|
||
|
LogLevelWarning: expectedWarning,
|
||
|
LogLevelInfo: noMessage,
|
||
|
LogLevelDebug: noMessage,
|
||
|
}},
|
||
|
{LogLevelError, map[LogLevel]string{
|
||
|
LogLevelError: expectedError,
|
||
|
LogLevelWarning: noMessage,
|
||
|
LogLevelInfo: noMessage,
|
||
|
LogLevelDebug: noMessage,
|
||
|
}},
|
||
|
{LogLevelNone, map[LogLevel]string{
|
||
|
LogLevelError: noMessage,
|
||
|
LogLevelWarning: noMessage,
|
||
|
LogLevelInfo: noMessage,
|
||
|
LogLevelDebug: noMessage,
|
||
|
}},
|
||
|
}
|
||
|
|
||
|
for idx := range tests {
|
||
|
level := tests[idx].logLevel
|
||
|
l.SetLevel(level)
|
||
|
|
||
|
for levelTry, msgExpect := range tests[idx].expect {
|
||
|
w.data = make([]byte, 0)
|
||
|
|
||
|
switch levelTry {
|
||
|
case LogLevelError:
|
||
|
l.Error(message)
|
||
|
case LogLevelWarning:
|
||
|
l.Warning(message)
|
||
|
case LogLevelInfo:
|
||
|
l.Info(message)
|
||
|
case LogLevelDebug:
|
||
|
l.Debug(message)
|
||
|
}
|
||
|
|
||
|
msgGot := string(w.data)
|
||
|
|
||
|
if msgGot != msgExpect {
|
||
|
t.Errorf("unexpected log message: expected `%s` but got `%s`", msgExpect, msgGot)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|