aboutsummaryrefslogtreecommitdiff
path: root/vendor/github.com/felix/logger/logger_test.go
blob: 3f6443077e67ebb6e7b0725e58951e03e3bac52b (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
104
105
106
107
108
109
110
111
112
113
114
package logger

import (
	"bytes"
	"strings"
	"testing"
)

func TestKeyValueWriter(t *testing.T) {
	var tests = []struct {
		in  []interface{}
		out string
	}{
		{
			in:  []interface{}{"one"},
			out: "[INFO ] test: message=one\n",
		},
		{
			in:  []interface{}{"one", "two", "2"},
			out: "[INFO ] test: message=one two=2\n",
		},
		{
			in:  []interface{}{"one", "two", "2", "three", 3},
			out: "[INFO ] test: message=one two=2 three=3\n",
		},
		{
			in:  []interface{}{"one", "two", "2", "three", 3, "fo ur", "# 4"},
			out: "[INFO ] test: message=one two=2 three=3 \"fo ur\"=\"# 4\"\n",
		},
	}

	for _, tt := range tests {
		var buf bytes.Buffer
		logger := New(&Options{
			Name:   "test",
			Output: &buf,
		})

		logger.Info(tt.in...)

		str := buf.String()

		// Chop timestamp
		dataIdx := strings.IndexByte(str, ' ')
		rest := str[dataIdx+1:]

		if rest != tt.out {
			t.Errorf("Info(%q) => %q, expected %q\n", tt.in, rest, tt.out)
		}
	}
}

func TestKeyValueWriterWithFields(t *testing.T) {
	var tests = []struct {
		in  []interface{}
		out string
	}{
		{
			in:  []interface{}{"one"},
			out: "[INFO ] test: message=one added=this\n",
		},
		{
			in:  []interface{}{"one", "two", "2"},
			out: "[INFO ] test: message=one two=2 added=this\n",
		},
		{
			in:  []interface{}{"one", "two", "2", "three", 3},
			out: "[INFO ] test: message=one two=2 three=3 added=this\n",
		},
		{
			in:  []interface{}{"one", "two", "2", "three", 3, "fo ur", "# 4"},
			out: "[INFO ] test: message=one two=2 three=3 \"fo ur\"=\"# 4\" added=this\n",
		},
	}
	for _, tt := range tests {
		var buf bytes.Buffer
		logger := New(&Options{
			Name:   "test",
			Output: &buf,
		}).WithFields("added", "this")

		logger.Info(tt.in...)

		str := buf.String()

		// Chop timestamp
		dataIdx := strings.IndexByte(str, ' ')
		rest := str[dataIdx+1:]

		if rest != tt.out {
			t.Errorf("Info(%q) => %q, expected %q\n", tt.in, rest, tt.out)
		}
	}
}

func TestLevels(t *testing.T) {
	logger := New(&Options{
		Name:  "test",
		Level: Debug,
	})

	if !logger.IsDebug() {
		t.Errorf("Level Debug check failed")
	}

	logger = New(&Options{
		Name:  "test",
		Level: Error,
	})

	if !logger.IsError() {
		t.Errorf("Level Error check failed")
	}
}