152 lines
3.1 KiB
Go
152 lines
3.1 KiB
Go
package logging
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/mitchellh/cli"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestLogger_SetupBasic(t *testing.T) {
|
|
t.Parallel()
|
|
require := require.New(t)
|
|
cfg := &Config{
|
|
LogLevel: "INFO",
|
|
}
|
|
ui := cli.NewMockUi()
|
|
|
|
logger, gatedWriter, writer, ok := Setup(cfg, ui)
|
|
require.True(ok)
|
|
require.NotNil(gatedWriter)
|
|
require.NotNil(writer)
|
|
require.NotNil(logger)
|
|
}
|
|
|
|
func TestLogger_SetupInvalidLogLevel(t *testing.T) {
|
|
t.Parallel()
|
|
require := require.New(t)
|
|
cfg := &Config{}
|
|
ui := cli.NewMockUi()
|
|
|
|
_, _, _, ok := Setup(cfg, ui)
|
|
require.False(ok)
|
|
require.Contains(ui.ErrorWriter.String(), "Invalid log level")
|
|
}
|
|
|
|
func TestLogger_SetupLoggerErrorLevel(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
cases := []struct {
|
|
desc string
|
|
before func(*Config)
|
|
}{
|
|
{
|
|
desc: "ERR log level",
|
|
before: func(cfg *Config) {
|
|
cfg.LogLevel = "ERR"
|
|
},
|
|
},
|
|
{
|
|
desc: "ERROR log level",
|
|
before: func(cfg *Config) {
|
|
cfg.LogLevel = "ERROR"
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
t.Run(c.desc, func(t *testing.T) {
|
|
var cfg Config
|
|
|
|
c.before(&cfg)
|
|
require := require.New(t)
|
|
ui := cli.NewMockUi()
|
|
|
|
logger, gatedWriter, _, ok := Setup(&cfg, ui)
|
|
require.True(ok)
|
|
require.NotNil(logger)
|
|
require.NotNil(gatedWriter)
|
|
|
|
gatedWriter.Flush()
|
|
|
|
logger.Error("test error msg")
|
|
logger.Info("test info msg")
|
|
|
|
require.Contains(ui.OutputWriter.String(), "[ERROR] test error msg")
|
|
require.NotContains(ui.OutputWriter.String(), "[INFO] test info msg")
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestLogger_SetupLoggerDebugLevel(t *testing.T) {
|
|
t.Parallel()
|
|
require := require.New(t)
|
|
cfg := &Config{
|
|
LogLevel: "DEBUG",
|
|
}
|
|
ui := cli.NewMockUi()
|
|
|
|
logger, gatedWriter, _, ok := Setup(cfg, ui)
|
|
require.True(ok)
|
|
require.NotNil(logger)
|
|
require.NotNil(gatedWriter)
|
|
|
|
gatedWriter.Flush()
|
|
|
|
logger.Info("test info msg")
|
|
logger.Debug("test debug msg")
|
|
|
|
require.Contains(ui.OutputWriter.String(), "[INFO] test info msg")
|
|
require.Contains(ui.OutputWriter.String(), "[DEBUG] test debug msg")
|
|
}
|
|
|
|
func TestLogger_SetupLoggerWithName(t *testing.T) {
|
|
t.Parallel()
|
|
require := require.New(t)
|
|
cfg := &Config{
|
|
LogLevel: "DEBUG",
|
|
Name: "test-system",
|
|
}
|
|
ui := cli.NewMockUi()
|
|
|
|
logger, gatedWriter, _, ok := Setup(cfg, ui)
|
|
require.True(ok)
|
|
require.NotNil(logger)
|
|
require.NotNil(gatedWriter)
|
|
|
|
gatedWriter.Flush()
|
|
|
|
logger.Warn("test warn msg")
|
|
|
|
require.Contains(ui.OutputWriter.String(), "[WARN] test-system: test warn msg")
|
|
}
|
|
|
|
func TestLogger_SetupLoggerWithJSON(t *testing.T) {
|
|
t.Parallel()
|
|
require := require.New(t)
|
|
cfg := &Config{
|
|
LogLevel: "DEBUG",
|
|
LogJSON: true,
|
|
Name: "test-system",
|
|
}
|
|
ui := cli.NewMockUi()
|
|
|
|
logger, gatedWriter, _, ok := Setup(cfg, ui)
|
|
require.True(ok)
|
|
require.NotNil(logger)
|
|
require.NotNil(gatedWriter)
|
|
|
|
gatedWriter.Flush()
|
|
|
|
logger.Warn("test warn msg")
|
|
|
|
var jsonOutput map[string]string
|
|
err := json.Unmarshal(ui.OutputWriter.Bytes(), &jsonOutput)
|
|
require.NoError(err)
|
|
require.Contains(jsonOutput, "@level")
|
|
require.Equal(jsonOutput["@level"], "warn")
|
|
require.Contains(jsonOutput, "@message")
|
|
require.Equal(jsonOutput["@message"], "test warn msg")
|
|
}
|