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
|
package logger
import (
"os"
"regexp"
"strings"
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/labkit/log"
"gitlab.com/gitlab-org/gitlab-shell/internal/config"
)
func TestConfigure(t *testing.T) {
tmpFile, err := os.CreateTemp(os.TempDir(), "logtest-")
require.NoError(t, err)
defer tmpFile.Close()
config := config.Config{
LogFile: tmpFile.Name(),
LogFormat: "json",
}
closer := Configure(&config)
defer closer.Close()
log.Info("this is a test")
tmpFile.Close()
data, err := os.ReadFile(tmpFile.Name())
require.NoError(t, err)
require.True(t, strings.Contains(string(data), `msg":"this is a test"`))
}
func TestConfigureWithPermissionError(t *testing.T) {
tmpPath, err := os.MkdirTemp(os.TempDir(), "logtest-")
require.NoError(t, err)
defer os.RemoveAll(tmpPath)
config := config.Config{
LogFile: tmpPath,
LogFormat: "json",
}
closer := Configure(&config)
defer closer.Close()
log.Info("this is a test")
}
func TestLogInUTC(t *testing.T) {
tmpFile, err := os.CreateTemp(os.TempDir(), "logtest-")
require.NoError(t, err)
defer tmpFile.Close()
defer os.Remove(tmpFile.Name())
config := config.Config{
LogFile: tmpFile.Name(),
LogFormat: "json",
}
closer := Configure(&config)
defer closer.Close()
log.Info("this is a test")
data, err := os.ReadFile(tmpFile.Name())
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)
}
|