...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15 package debug
16
17 import (
18 "os"
19 "testing"
20
21 "github.com/stretchr/testify/assert"
22 "github.com/stretchr/testify/require"
23 )
24
25 func TestDebug(t *testing.T) {
26 tmpFile, err := os.CreateTemp("", "debug-test")
27 require.NoError(t, err)
28
29 output = tmpFile
30
31 tmpName := tmpFile.Name()
32 defer func() {
33 _ = os.Remove(tmpName)
34 }()
35
36 testLogger := GetLogger("test", true)
37
38 testLogger("A debug: %s", "a string")
39 tmpFile.Close()
40
41 flushed, err := os.ReadFile(tmpName)
42 require.NoError(t, err)
43
44 assert.Contains(t, string(flushed), "A debug: a string")
45
46 tmpEmptyFile, err := os.CreateTemp("", "debug-test")
47 require.NoError(t, err)
48 tmpEmpty := tmpEmptyFile.Name()
49 defer func() {
50 _ = os.Remove(tmpEmpty)
51 }()
52
53 testLogger = GetLogger("test", false)
54 testLogger("A debug: %s", "a string")
55 tmpFile.Close()
56
57 flushed, err = os.ReadFile(tmpEmpty)
58 require.NoError(t, err)
59
60 assert.Empty(t, flushed)
61 }
62
View as plain text