logger_test.go

package zap

import (
	"bytes"
	"encoding/json"
	"strings"
	"testing"

	"go.uber.org/zap"
	"go.uber.org/zap/zapcore"
	"logur.dev/logur"
	"logur.dev/logur/conformance"
)

// nolint: gochecknoglobals
var levelMap = map[logur.Level]zapcore.Level{
	logur.Trace: zap.DebugLevel,
	logur.Debug: zap.DebugLevel,
	logur.Info:  zap.InfoLevel,
	logur.Warn:  zap.WarnLevel,
	logur.Error: zap.ErrorLevel,
}

func TestLogger(t *testing.T) {
	suite := conformance.TestSuite{
		NoTraceLevel: true,
		LoggerFactory: func(level logur.Level) (logur.Logger, conformance.TestLogger) {
			var buf bytes.Buffer

			logger := zap.New(
				zapcore.NewCore(
					zapcore.NewConsoleEncoder(zap.NewDevelopmentEncoderConfig()),
					zapcore.AddSync(&buf),
					levelMap[level],
				),
			)

			return New(logger), conformance.TestLoggerFunc(func() []logur.LogEvent {
				lines := strings.Split(strings.TrimSuffix(buf.String(), "\n"), "\n")

				events := make([]logur.LogEvent, len(lines))

				for key, line := range lines {
					log := strings.SplitN(line, "\t", 4)

					level, _ := logur.ParseLevel(strings.ToLower(log[1]))

					var fields map[string]interface{}

					if len(log) > 3 {
						err := json.Unmarshal([]byte(log[3]), &fields)
						if err != nil {
							panic(err)
						}
					}

					events[key] = logur.LogEvent{
						Line:   log[2],
						Level:  level,
						Fields: fields,
					}
				}

				return events
			})
		},
	}

	suite.Run(t)
}