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
|
package pp
import (
"bytes"
"testing"
)
func TestDefaultOutput(t *testing.T) {
testOutput := new(bytes.Buffer)
init := GetDefaultOutput()
SetDefaultOutput(testOutput)
if GetDefaultOutput() != testOutput {
t.Errorf("failed to SetOutput")
}
if len(testOutput.String()) != 0 {
t.Errorf("testOutput should be initialized")
}
Print("abcde")
if len(testOutput.String()) == 0 {
t.Errorf("Expected Print output to testOutput, testOutput is %s", testOutput.String())
}
if init == GetDefaultOutput() {
t.Errorf("it should be changed DefaultOutput")
}
ResetDefaultOutput()
if init != GetDefaultOutput() {
t.Errorf("it should be reset to initial default output")
}
}
func TestColorScheme(t *testing.T) {
SetColorScheme(ColorScheme{})
if defaultPrettyPrinter.currentScheme.FieldName == 0 {
t.FailNow()
}
}
func TestWithLineInfo(t *testing.T) {
outputWithoutLineInfo := new(bytes.Buffer)
SetDefaultOutput(outputWithoutLineInfo)
Print("abcde")
outputWithLineInfo := new(bytes.Buffer)
SetDefaultOutput(outputWithLineInfo)
WithLineInfo = true
Print("abcde")
ResetDefaultOutput()
if bytes.Equal(outputWithLineInfo.Bytes(), outputWithoutLineInfo.Bytes()) {
t.Errorf("outputWithLineInfo should not have the same contents than outputWithoutLineInfo")
}
}
func TestWithLineInfoBackwardsCompatible(t *testing.T) {
// Test that the global accessible field `WithLineInfo` does not mutate other instances
outputWithLineInfo := new(bytes.Buffer)
SetDefaultOutput(outputWithLineInfo)
WithLineInfo = true
Print("abcde")
outputWithoutLineInfo := new(bytes.Buffer)
pp := New()
pp.SetOutput(outputWithoutLineInfo)
pp.Print("abcde")
if bytes.Equal(outputWithLineInfo.Bytes(), outputWithoutLineInfo.Bytes()) {
t.Errorf("outputWithLineInfo should not have the same contents than outputWithoutLineInfo")
}
ResetDefaultOutput()
}
|
