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
|
package logger
import (
"os"
"regexp"
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/labkit/log"
"gitlab.com/gitlab-org/gitlab-shell/v14/internal/config"
)
func TestConfigure(t *testing.T) {
tmpFile := createTempFile(t)
config := config.Config{
LogFile: tmpFile,
LogFormat: "json",
}
closer := Configure(&config)
defer closer.Close()
log.Info("this is a test")
log.WithFields(log.Fields{}).Debug("debug log message")
data, err := os.ReadFile(tmpFile)
dataStr := string(data)
require.NoError(t, err)
require.Contains(t, dataStr, `"msg":"this is a test"`)
require.NotContains(t, dataStr, `"msg":"debug log message"`)
require.NotContains(t, dataStr, `"msg":"unknown log level`)
}
func TestConfigureWithDebugLogLevel(t *testing.T) {
tmpFile := createTempFile(t)
config := config.Config{
LogFile: tmpFile,
LogFormat: "json",
LogLevel: "debug",
}
closer := Configure(&config)
defer closer.Close()
log.WithFields(log.Fields{}).Debug("debug log message")
data, err := os.ReadFile(tmpFile)
require.NoError(t, err)
require.Contains(t, string(data), `msg":"debug log message"`)
}
func TestConfigureWithPermissionError(t *testing.T) {
tempDir := t.TempDir()
config := config.Config{
LogFile: tempDir,
LogFormat: "json",
}
closer := Configure(&config)
defer closer.Close()
log.Info("this is a test")
}
func TestLogInUTC(t *testing.T) {
tmpFile := createTempFile(t)
config := config.Config{
LogFile: tmpFile,
LogFormat: "json",
}
closer := Configure(&config)
defer closer.Close()
log.Info("this is a test")
data, err := os.ReadFile(tmpFile)
require.NoError(t, err)
utc := `[0-9]{4}-[0-9]{2}-[0-9]{2}T[0-9]{2}:[0-9]{2}:[0-9]{2}Z`
r, e := regexp.MatchString(utc, string(data))
require.NoError(t, e)
require.True(t, r)
}
func createTempFile(t *testing.T) string {
t.Helper()
tmpFile, err := os.CreateTemp(t.TempDir(), "logtest-")
require.NoError(t, err)
tmpFile.Close()
return tmpFile.Name()
}
|
