logger_test.go

package log

import (
	"bytes"
	"fmt"
	"strings"
	"testing"

	"github.com/sirupsen/logrus"
	"github.com/stretchr/testify/require"
)

func TestLoggerNew(t *testing.T) {
	require := require.New(t)

	f := &LoggerFactory{Format: "text", Level: "debug"}
	l, err := f.New(nil)
	require.NoError(err)

	l = l.New(Fields{"foo": "qux"})
	l1, ok := l.(*logger)
	require.True(ok)
	require.Equal(logrus.Fields{"foo": "qux"}, l1.Entry.Data)

	l = l.New(Fields{"bar": "baz"})
	l2, ok := l.(*logger)
	require.True(ok)
	require.Equal(logrus.Fields{
		"foo": "qux",
		"bar": "baz",
	}, l2.Entry.Data)
}

func TestLogger_With(t *testing.T) {
	require := require.New(t)

	f := &LoggerFactory{Format: "text", Level: "debug"}
	l, err := f.New(nil)
	require.NoError(err)

	l = l.With(Fields{"foo": "qux"})
	l1, ok := l.(*logger)
	require.True(ok)
	require.Equal(logrus.Fields{"foo": "qux"}, l1.Entry.Data)
}

func TestLogger_Errorf(t *testing.T) {
	require := require.New(t)

	f := &LoggerFactory{Format: "text", Level: "debug"}
	l, err := f.New(nil)
	require.NoError(err)

	logger, ok := l.(*logger)
	require.True(ok)

	buf := bytes.NewBuffer(nil)
	logger.Logger.Out = buf

	l.Errorf(fmt.Errorf("foo"), "qux %d", 42)
	require.True(strings.Contains(buf.String(), "error=foo"))
	require.True(strings.Contains(buf.String(), "msg=\"qux 42\""))
}