blob: 1e47e4a6520504091664f203b9514edcdfeee14d (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
package logging_test
import (
"encoding/json"
"log/slog"
"slices"
"sync"
"testing"
"testing/slogtest"
"go.sudomsg.com/kit/logging"
)
type logRecorder struct {
mu sync.Mutex
encoder logging.Encoder
records []string
}
var _ logging.Sink = &logRecorder{}
func (h *logRecorder) Append(r slog.Record) error {
h.mu.Lock()
defer h.mu.Unlock()
it := logging.RecordAll(r, nil)
entry, err := h.encoder(it)
if err != nil {
return err
}
h.records = append(h.records, entry)
return nil
}
func (h *logRecorder) Enabled(level slog.Level) bool {
return true // Capture all logs
}
func (h *logRecorder) Records() []string {
h.mu.Lock()
defer h.mu.Unlock()
return slices.Clone(h.records)
}
func TestJSONEncoder(t *testing.T) {
var lastMock *logRecorder
slogtest.Run(t,
func(t *testing.T) slog.Handler {
lastMock = &logRecorder{
encoder: logging.JSONEncoder,
}
return logging.NewSinkHandler(lastMock)
},
func(t *testing.T) map[string]any {
t.Helper()
recs := lastMock.Records()
if len(recs) == 0 {
t.Fatalf("no records captured")
}
rec := recs[len(recs)-1]
m := make(map[string]any)
if err := json.Unmarshal([]byte(rec), &m); err != nil {
t.Fatalf("%v", err)
}
return m
},
)
}
|
