stdlogger_test.go

package logr_test

import (
	"bytes"
	"log"
	"os"
	"testing"

	"github.com/mattermost/logr/v2"
	"github.com/mattermost/logr/v2/formatters"
	"github.com/mattermost/logr/v2/targets"
	"github.com/stretchr/testify/require"
)

func TestNewStdLogger(t *testing.T) {
	buf := &bytes.Buffer{}

	lgr, err := logr.New(logr.StackFilter("log"))
	require.NoError(t, err)

	customLevel := logr.Level{ID: 1001, Name: "std", Color: logr.Green}

	filter := logr.NewCustomFilter(customLevel)
	formatter := &formatters.Plain{Delim: " ", EnableColor: true}

	tgt := targets.NewWriterTarget(buf)
	err = lgr.AddTarget(tgt, "buf", filter, formatter, 1000)
	require.NoError(t, err)

	stdLogger := lgr.NewLogger().With(logr.String("foo", "bar hop")).StdLogger(customLevel)

	stdLogger.Println("The Spirit Of Radio")

	err = lgr.Shutdown()
	require.NoError(t, err)

	output := buf.String()
	require.Contains(t, output, "foo")
	require.Contains(t, output, "\"bar hop\"")
	require.Contains(t, output, "The Spirit Of Radio")

	t.Log(output)
}

func TestRedirectStdLog(t *testing.T) {
	buf := &bytes.Buffer{}

	lgr, err := logr.New(logr.StackFilter("log"))
	require.NoError(t, err)

	filter := &logr.StdFilter{logr.Info, logr.Error}
	formatter := &formatters.Plain{Delim: " ", MinMessageLen: 40}

	tgt := targets.NewWriterTarget(buf)
	err = lgr.AddTarget(tgt, "buf", filter, formatter, 1000)
	require.NoError(t, err)

	tgt = targets.NewWriterTarget(os.Stdout)
	err = lgr.AddTarget(tgt, "stdout", filter, formatter, 1000)
	require.NoError(t, err)

	// remember old settings.
	flags := log.Flags()
	prefix := log.Prefix()

	restoreFunc := lgr.RedirectStdLog(logr.Info, logr.String("foo", "bar stool"))

	log.Println("Peaky Blinders!")

	err = lgr.Shutdown()
	require.NoError(t, err)

	output := buf.String()
	require.Contains(t, output, "foo")
	require.Contains(t, output, "bar stool")
	require.Contains(t, output, "Peaky Blinders!")

	restoreFunc()

	// check settings restored
	require.Equal(t, flags, log.Flags())
	require.Equal(t, prefix, log.Prefix())
}