processor_test.go

package slog_test

import (
	"context"
	"fmt"
	"os"
	"testing"

	"github.com/gookit/goutil/byteutil"
	"github.com/gookit/goutil/testutil/assert"
	"github.com/gookit/slog"
)

func TestLogger_AddProcessor(t *testing.T) {
	buf := new(byteutil.Buffer)

	l := slog.NewJSONSugared(buf, slog.InfoLevel)
	l.AddProcessor(slog.AddHostname())
	l.Info("message")

	hostname, _ := os.Hostname()

	// {"channel":"application","data":{},"datetime":"2020/07/17 12:01:35","extra":{},"hostname":"InhereMac","level":"INFO","message":"message"}
	str := buf.String()
	buf.Reset()
	assert.Contains(t, str, `"level":"INFO"`)
	assert.Contains(t, str, `"message":"message"`)
	assert.Contains(t, str, fmt.Sprintf(`"hostname":"%s"`, hostname))

	l.ResetProcessors()
	l.PushProcessor(slog.MemoryUsage)
	l.Info("message2")

	// {"channel":"application","data":{},"datetime":"2020/07/16 16:40:18","extra":{"memoryUsage":326072},"level":"INFO","message":"message2"}
	str = buf.String()
	buf.Reset()
	assert.Contains(t, str, `"message":"message2"`)
	assert.Contains(t, str, `"memoryUsage":`)

	l.ResetProcessors()
	l.SetProcessors([]slog.Processor{slog.AddUniqueID("requestId")})
	l.Info("message3")
	str = buf.String()
	buf.Reset()
	assert.Contains(t, str, `"message":"message3"`)
	assert.Contains(t, str, `"requestId":`)
	fmt.Print(str)

	l.ResetProcessors()
	l.AddProcessors(slog.AppendCtxKeys("traceId", "userId"))
	l.Info("message4")
	str = buf.ResetAndGet()
	fmt.Print(str)
	assert.Contains(t, str, `"message":"message4"`)
	assert.NotContains(t, str, `"traceId"`)

	ctx := context.WithValue(context.Background(), "traceId", "traceId123abc456")
	l.WithCtx(ctx).Info("message5")
	str = buf.ResetAndGet()
	fmt.Print(str)
	assert.Contains(t, str, `"message":"message5"`)
	assert.Contains(t, str, `"traceId":"traceId123abc456"`)
}

func TestProcessable_AddProcessor(t *testing.T) {
	ps := &slog.Processable{}
	ps.AddProcessor(slog.MemoryUsage)

	r := newLogRecord("error message")
	ps.ProcessRecord(r)

	assert.NotEmpty(t, r.Extra)
	assert.Contains(t, r.Extra, "memoryUsage")
}