buffer_handler_test.go

package collect_test

import (
	"encoding/json"
	"log/slog"
	"testing"
	"testing/slogtest"

	"github.com/google/go-cmp/cmp"
	"github.com/osbuild/logging/pkg/collect"
)

func TestStandardLibraryHelper(t *testing.T) {
	var result []map[string]any
	th := collect.NewTestHandler(slog.LevelDebug, true, true, true)
	err := slogtest.TestHandler(th, func() []map[string]any {
		result = th.All()
		return parseLogEntries(t, result)
	})
	if err != nil {
		t.Error(err)
		json, _ := json.MarshalIndent(result, "", " ")
		t.Log(string(json))
	}
}

func parseLogEntries(_ *testing.T, ms []map[string]any) []map[string]any {
	return ms
}

func TestLast(t *testing.T) {
	h := collect.NewTestHandler(slog.LevelDebug, false, false, false)
	logger := slog.New(h)

	tests := []struct {
		f    func()
		want map[string]any
	}{
		{
			f: func() {
				logger.Debug("test", "key", "value")
			},
			want: map[string]any{
				"msg": "test",
				"key": "value",
			},
		},
		{
			f: func() {
				logger.Debug("test", slog.Group("g", "key", "value"))
			},
			want: map[string]any{
				"msg": "test",
				"g": map[string]any{
					"key": "value",
				},
			},
		},
		{
			f: func() {
				logger.WithGroup("g").Debug("test", "key", "value")
			},
			want: map[string]any{
				"msg": "test",
				"g": map[string]any{
					"key": "value",
				},
			},
		},
	}

	for _, tt := range tests {
		tt.f()
		got := h.Last()

		if !cmp.Equal(got, tt.want) {
			t.Errorf("Got: %v, want: %v", got, tt.want)
		}
	}
}