logger_test.go

package kit

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

	"github.com/go-kit/kit/log"
	"logur.dev/logur"
	"logur.dev/logur/conformance"
)

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

			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 {
					var event map[string]interface{}

					err := json.Unmarshal([]byte(line), &event)
					if err != nil {
						panic(err)
					}

					level, _ := logur.ParseLevel(strings.ToLower(event["level"].(string)))
					msg := event["msg"].(string)

					delete(event, "level")
					delete(event, "msg")

					events[key] = logur.LogEvent{
						Line:   msg,
						Level:  level,
						Fields: event,
					}
				}

				return events
			})
		},
	}

	suite.Run(t)
}