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
|
package stdlog
import (
"bytes"
"io"
"testing"
"github.com/alexcesaro/log"
"github.com/alexcesaro/log/logtest"
)
var (
testStream = new(bytes.Buffer)
testlogToStderr = false
)
func TestGetFromFlags(t *testing.T) {
logger := getLogger("info", "none")
logger.Debug(logtest.Messages[log.Debug])
logtest.AssertNotContain(t, testStream, logtest.Messages[log.Debug])
logger.Info(logtest.Messages[log.Info])
logtest.AssertContains(t, testStream, logtest.Messages[log.Info])
logtest.AssertLineCount(t, testStream, 1)
logger = getLogger("info", "error")
logger.Info(logtest.Messages[log.Info])
logtest.AssertNotContain(t, testStream, logtest.Messages[log.Info])
logger.Error(logtest.Messages[log.Error])
logtest.AssertContains(t, testStream, logtest.Messages[log.Error])
logtest.AssertLineCount(t, testStream, 2)
}
func getLogger(threshold, flushThreshold string) log.Logger {
getStream = func(logToStderr bool) io.Writer {
return testStream
}
thresholdName = &threshold
logToStderr = &testlogToStderr
flushThresholdName = &flushThreshold
testStream.Reset()
logger = nil
return GetFromFlags()
}
|
