consul/logging/logger_test.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")
}