70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
|
package logger_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/go-flucky/flucky/pkg/logger"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestLogger(t *testing.T) {
|
||
|
|
require := require.New(t)
|
||
|
|
stdoutBuffer := new(bytes.Buffer)
|
||
|
|
stderrBuffer := new(bytes.Buffer)
|
||
|
|
|
||
|
|
// LogLevelDebug
|
||
|
|
l := logger.NewCustomLogger(logger.LogLevelDebug, stdoutBuffer, stderrBuffer)
|
||
|
|
l.Debug("DEBUG")
|
||
|
|
require.NotEmpty(stdoutBuffer.Bytes())
|
||
|
|
require.Empty(stderrBuffer.Bytes())
|
||
|
|
|
||
|
|
stdoutBuffer.Reset()
|
||
|
|
stderrBuffer.Reset()
|
||
|
|
|
||
|
|
// LogLevelInfo
|
||
|
|
l = logger.NewCustomLogger(logger.LogLevelInfo, stdoutBuffer, stderrBuffer)
|
||
|
|
|
||
|
|
l.Debug("DEBUG")
|
||
|
|
require.Empty(stdoutBuffer.Bytes())
|
||
|
|
require.Empty(stderrBuffer.Bytes())
|
||
|
|
|
||
|
|
stdoutBuffer.Reset()
|
||
|
|
stderrBuffer.Reset()
|
||
|
|
|
||
|
|
l.Info("INFO")
|
||
|
|
require.NotEmpty(stdoutBuffer.Bytes())
|
||
|
|
require.Empty(stderrBuffer.Bytes())
|
||
|
|
|
||
|
|
stringArray := strings.Split(stdoutBuffer.String(), ": ")
|
||
|
|
require.Equal(2, len(stringArray))
|
||
|
|
require.Equal("INFO\n", stringArray[1])
|
||
|
|
|
||
|
|
stdoutBuffer.Reset()
|
||
|
|
stderrBuffer.Reset()
|
||
|
|
|
||
|
|
// LogLevelWarn
|
||
|
|
l = logger.NewCustomLogger(logger.LogLevelWarn, stdoutBuffer, stderrBuffer)
|
||
|
|
l.Warn("WARN")
|
||
|
|
require.NotEmpty(stdoutBuffer.Bytes())
|
||
|
|
require.Empty(stderrBuffer.Bytes())
|
||
|
|
|
||
|
|
stringArray = strings.Split(stdoutBuffer.String(), ": ")
|
||
|
|
require.Equal(2, len(stringArray))
|
||
|
|
require.Equal("WARN\n", stringArray[1])
|
||
|
|
|
||
|
|
stdoutBuffer.Reset()
|
||
|
|
stderrBuffer.Reset()
|
||
|
|
|
||
|
|
// LogLevelError
|
||
|
|
l = logger.NewCustomLogger(logger.LogLevelError, stdoutBuffer, stderrBuffer)
|
||
|
|
l.Error("ERROR")
|
||
|
|
require.Empty(stdoutBuffer.Bytes())
|
||
|
|
require.NotEmpty(stderrBuffer.Bytes())
|
||
|
|
|
||
|
|
stringArray = strings.Split(stderrBuffer.String(), ": ")
|
||
|
|
require.Equal(2, len(stringArray))
|
||
|
|
require.Equal("ERROR\n", stringArray[1])
|
||
|
|
}
|