-
Notifications
You must be signed in to change notification settings - Fork 0
/
format_internal_test.go
56 lines (49 loc) · 1.29 KB
/
format_internal_test.go
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
package logrusconfigurator
import (
"testing"
"github.com/sirupsen/logrus"
"github.com/stretchr/testify/require"
)
func TestGetLogrusFormat(t *testing.T) {
testCases := []struct {
input format
expected logrus.Formatter
expectError bool
}{
{formatJSON, &logrus.JSONFormatter{}, false},
{formatText, &logrus.TextFormatter{}, false},
{"invalid", nil, true},
}
for _, tc := range testCases {
t.Run(string(tc.input), func(t *testing.T) {
result, err := getLogrusFormat(tc.input)
if tc.expectError {
require.Error(t, err)
return
}
require.NoError(t, err)
require.IsType(t, tc.expected, result)
})
}
}
func TestSetFormat(t *testing.T) {
testCases := []struct {
format format
expectedFormat logrus.Formatter
expectError bool
}{
{formatJSON, &logrus.JSONFormatter{}, false},
{formatText, &logrus.TextFormatter{}, false},
{"Invalid", nil, true},
}
for _, tc := range testCases {
err := setFormat(tc.format)
if tc.expectError {
require.NotNil(t, err, "Expected error for format: "+string(tc.format))
} else {
require.Nil(t, err, "Unexpected error for format: "+string(tc.format))
actualFormatter := logrus.StandardLogger().Formatter
require.IsType(t, tc.expectedFormat, actualFormatter, "Formatter type mismatch")
}
}
}