aboutsummaryrefslogtreecommitdiff
path: root/writers/json/json_test.go
blob: e6743a2e7d95cfc6dd66c2326a79da4fed23c257 (plain)
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package json

import (
	"bytes"
	"encoding/json"
	"reflect"
	"testing"
	"time"

	"src.userspace.com.au/felix/logger/message"
)

func TestWriter(t *testing.T) {
	now, _ := time.Parse(time.RFC3339, "2019-05-03T13:38:29.987249+10:00")
	var tests = []struct {
		in       message.Message
		expected map[string]interface{}
	}{
		{
			in: message.Message{
				Time:    now,
				Name:    "test",
				Level:   message.ERROR,
				Content: "msg",
			},
			expected: map[string]interface{}{
				"_level": "error", "_name": "test", "_message": "msg",
				"_time": "2019-05-03T13:38:29.987249+10:00",
			},
		},
		{
			in: message.Message{
				Time:    now,
				Name:    "test",
				Level:   message.ERROR,
				Content: "msg",
				Extras:  []interface{}{"one"},
			},
			expected: map[string]interface{}{
				"_level": "error", "_name": "test", "_message": "msg", "extra00": "one",
				"_time": "2019-05-03T13:38:29.987249+10:00",
			},
		},
		{
			in: message.Message{
				Time:    now,
				Name:    "test",
				Level:   message.ERROR,
				Content: "msg",
				Fields:  map[string]interface{}{"one": "1"},
			},
			expected: map[string]interface{}{
				"_level": "error", "_name": "test", "_message": "msg", "one": "1",
				"_time": "2019-05-03T13:38:29.987249+10:00",
			},
		},
		{
			in: message.Message{
				Time:    now,
				Name:    "test",
				Level:   message.ERROR,
				Content: "msg", Extras: []interface{}{"one", "1", "two", "2", "three", 3, "fo ur", "# 4"},
			},
			expected: map[string]interface{}{
				"_level": "error", "_name": "test", "_message": "msg", "one": "1", "two": "2", "three": float64(3), "fo ur": "# 4",
				"_time": "2019-05-03T13:38:29.987249+10:00",
			},
		},
		{
			in: message.Message{
				Time:    now,
				Name:    "test",
				Level:   message.DEBUG,
				Content: "msg",
				Extras:  []interface{}{"one"}, Fields: map[string]interface{}{"f1": "v1"},
			},
			expected: map[string]interface{}{
				"_level": "debug", "_name": "test", "_message": "msg", "f1": "v1", "extra00": "one",
				"_time": "2019-05-03T13:38:29.987249+10:00",
			},
		},
	}

	var buf bytes.Buffer
	l, err := New(SetOutput(&buf))
	if err != nil {
		panic(err)
	}

	for _, tt := range tests {
		l.Write(tt.in)

		var raw map[string]interface{}
		if err := json.Unmarshal(buf.Bytes(), &raw); err != nil {
			t.Fatal(err)
		}
		buf.Reset()

		if !reflect.DeepEqual(raw, tt.expected) {
			t.Errorf("got %q, expected %q\n", raw, tt.expected)
		}
	}
}