-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathtable_test.go
146 lines (136 loc) · 15.7 KB
/
table_test.go
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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
package charts
import (
"strconv"
"testing"
"github.com/stretchr/testify/require"
)
func makeDefaultTableChartOptions() TableChartOption {
return TableChartOption{
Padding: NewBoxEqual(10),
Header: []string{
"Name", "Age", "Address", "Tag", "Action",
},
Data: [][]string{
{
"John Brown",
"32",
"New York No. 1 Lake Park",
"nice, developer",
"Send Mail",
},
{
"Jim Green ",
"42",
"London No. 1 Lake Park",
"wow",
"Send Mail",
},
{
"Joe Black ",
"32",
"Sidney No. 1 Lake Park",
"cool, teacher",
"Send Mail",
},
},
}
}
func TestTableChart(t *testing.T) {
t.Parallel()
tests := []struct {
name string
theme ColorPalette
makeOptions func() TableChartOption
result string
errorExpected bool
}{
{
name: "default_theme_spans",
theme: nil, // use default
makeOptions: func() TableChartOption {
opt := makeDefaultTableChartOptions()
opt.Spans = []int{
1,
1,
2,
1,
// span and header do not match, and are automatically set to 1 at the end
// 1,
}
return opt
},
result: "<svg xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\" viewBox=\"0 0 600 400\"><path d=\"M 0 0\nL 600 0\nL 600 35\nL 0 35\nL 0 0\" style=\"stroke:none;fill:rgb(220,220,220)\"/><path d=\"M 0 35\nL 600 35\nL 600 90\nL 0 90\nL 0 35\" style=\"stroke:none;fill:white\"/><path d=\"M 0 90\nL 600 90\nL 600 125\nL 0 125\nL 0 90\" style=\"stroke:none;fill:rgb(245,245,245)\"/><path d=\"M 0 125\nL 600 125\nL 600 180\nL 0 180\nL 0 125\" style=\"stroke:none;fill:white\"/><text x=\"10\" y=\"22\" style=\"stroke:none;fill:rgb(80,80,80);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Name</text><text x=\"110\" y=\"22\" style=\"stroke:none;fill:rgb(80,80,80);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Age</text><text x=\"210\" y=\"22\" style=\"stroke:none;fill:rgb(80,80,80);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Address</text><text x=\"410\" y=\"22\" style=\"stroke:none;fill:rgb(80,80,80);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Tag</text><text x=\"510\" y=\"22\" style=\"stroke:none;fill:rgb(80,80,80);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Action</text><text x=\"10\" y=\"57\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">John</text><text x=\"10\" y=\"77\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Brown</text><text x=\"110\" y=\"57\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">32</text><text x=\"210\" y=\"57\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">New York No. 1 Lake Park</text><text x=\"410\" y=\"57\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">nice,</text><text x=\"410\" y=\"77\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">developer</text><text x=\"510\" y=\"57\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Send Mail</text><text x=\"10\" y=\"112\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Jim Green</text><text x=\"110\" y=\"112\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">42</text><text x=\"210\" y=\"112\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">London No. 1 Lake Park</text><text x=\"410\" y=\"112\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">wow</text><text x=\"510\" y=\"112\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Send Mail</text><text x=\"10\" y=\"147\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Joe Black</text><text x=\"110\" y=\"147\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">32</text><text x=\"210\" y=\"147\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Sidney No. 1 Lake Park</text><text x=\"410\" y=\"147\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">cool,</text><text x=\"410\" y=\"167\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">teacher</text><text x=\"510\" y=\"147\" style=\"stroke:none;fill:rgb(50,50,50);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Send Mail</text></svg>",
},
{
name: "dark_theme",
theme: GetTheme(ThemeVividDark),
makeOptions: makeDefaultTableChartOptions,
result: "<svg xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\" viewBox=\"0 0 600 400\"><path d=\"M 0 0\nL 600 0\nL 600 35\nL 0 35\nL 0 0\" style=\"stroke:none;fill:rgb(38,38,42)\"/><path d=\"M 0 35\nL 600 35\nL 600 90\nL 0 90\nL 0 35\" style=\"stroke:none;fill:rgb(24,24,28)\"/><path d=\"M 0 90\nL 600 90\nL 600 145\nL 0 145\nL 0 90\" style=\"stroke:none;fill:rgb(38,38,42)\"/><path d=\"M 0 145\nL 600 145\nL 600 200\nL 0 200\nL 0 145\" style=\"stroke:none;fill:rgb(24,24,28)\"/><text x=\"10\" y=\"22\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Name</text><text x=\"130\" y=\"22\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Age</text><text x=\"250\" y=\"22\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Address</text><text x=\"370\" y=\"22\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Tag</text><text x=\"490\" y=\"22\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Action</text><text x=\"10\" y=\"57\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">John Brown</text><text x=\"130\" y=\"57\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">32</text><text x=\"250\" y=\"57\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">New York No.</text><text x=\"250\" y=\"77\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">1 Lake Park</text><text x=\"370\" y=\"57\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">nice,</text><text x=\"370\" y=\"77\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">developer</text><text x=\"490\" y=\"57\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Send Mail</text><text x=\"10\" y=\"112\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Jim Green</text><text x=\"130\" y=\"112\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">42</text><text x=\"250\" y=\"112\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">London No. 1</text><text x=\"250\" y=\"132\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Lake Park</text><text x=\"370\" y=\"112\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">wow</text><text x=\"490\" y=\"112\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Send Mail</text><text x=\"10\" y=\"167\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Joe Black</text><text x=\"130\" y=\"167\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">32</text><text x=\"250\" y=\"167\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Sidney No. 1</text><text x=\"250\" y=\"187\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Lake Park</text><text x=\"370\" y=\"167\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">cool, teacher</text><text x=\"490\" y=\"167\" style=\"stroke:none;fill:rgb(216,217,218);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Send Mail</text></svg>",
},
{
name: "cell_modified",
makeOptions: func() TableChartOption {
opt := makeDefaultTableChartOptions()
opt.CellModifier = func(tc TableCell) TableCell {
if tc.Column%2 == 0 {
tc.FillColor = ColorWhite
tc.FontStyle.FontColor = ColorBlue
} else if tc.Row%2 == 0 {
tc.FillColor = ColorAqua
tc.FontStyle.FontColor = ColorBlack
} else {
tc.FillColor = ColorYellow
tc.FontStyle.FontColor = ColorPurple
}
return tc
}
return opt
},
result: "<svg xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\" viewBox=\"0 0 600 400\"><path d=\"M 0 0\nL 600 0\nL 600 35\nL 0 35\nL 0 0\" style=\"stroke:none;fill:rgb(220,220,220)\"/><path d=\"M 0 35\nL 600 35\nL 600 90\nL 0 90\nL 0 35\" style=\"stroke:none;fill:white\"/><path d=\"M 0 90\nL 600 90\nL 600 145\nL 0 145\nL 0 90\" style=\"stroke:none;fill:rgb(245,245,245)\"/><path d=\"M 0 145\nL 600 145\nL 600 200\nL 0 200\nL 0 145\" style=\"stroke:none;fill:white\"/><path d=\"M 10 10\nL 110 10\nL 110 25\nL 10 25\nL 10 10\" style=\"stroke:none;fill:white\"/><path d=\"M 130 10\nL 230 10\nL 230 25\nL 130 25\nL 130 10\" style=\"stroke:none;fill:aqua\"/><path d=\"M 250 10\nL 350 10\nL 350 25\nL 250 25\nL 250 10\" style=\"stroke:none;fill:white\"/><path d=\"M 370 10\nL 470 10\nL 470 25\nL 370 25\nL 370 10\" style=\"stroke:none;fill:aqua\"/><path d=\"M 490 10\nL 590 10\nL 590 25\nL 490 25\nL 490 10\" style=\"stroke:none;fill:white\"/><path d=\"M 10 45\nL 110 45\nL 110 80\nL 10 80\nL 10 45\" style=\"stroke:none;fill:white\"/><path d=\"M 130 45\nL 230 45\nL 230 80\nL 130 80\nL 130 45\" style=\"stroke:none;fill:yellow\"/><path d=\"M 250 45\nL 350 45\nL 350 80\nL 250 80\nL 250 45\" style=\"stroke:none;fill:white\"/><path d=\"M 370 45\nL 470 45\nL 470 80\nL 370 80\nL 370 45\" style=\"stroke:none;fill:yellow\"/><path d=\"M 490 45\nL 590 45\nL 590 80\nL 490 80\nL 490 45\" style=\"stroke:none;fill:white\"/><path d=\"M 10 100\nL 110 100\nL 110 135\nL 10 135\nL 10 100\" style=\"stroke:none;fill:white\"/><path d=\"M 130 100\nL 230 100\nL 230 135\nL 130 135\nL 130 100\" style=\"stroke:none;fill:aqua\"/><path d=\"M 250 100\nL 350 100\nL 350 135\nL 250 135\nL 250 100\" style=\"stroke:none;fill:white\"/><path d=\"M 370 100\nL 470 100\nL 470 135\nL 370 135\nL 370 100\" style=\"stroke:none;fill:aqua\"/><path d=\"M 490 100\nL 590 100\nL 590 135\nL 490 135\nL 490 100\" style=\"stroke:none;fill:white\"/><path d=\"M 10 155\nL 110 155\nL 110 190\nL 10 190\nL 10 155\" style=\"stroke:none;fill:white\"/><path d=\"M 130 155\nL 230 155\nL 230 190\nL 130 190\nL 130 155\" style=\"stroke:none;fill:yellow\"/><path d=\"M 250 155\nL 350 155\nL 350 190\nL 250 190\nL 250 155\" style=\"stroke:none;fill:white\"/><path d=\"M 370 155\nL 470 155\nL 470 190\nL 370 190\nL 370 155\" style=\"stroke:none;fill:yellow\"/><path d=\"M 490 155\nL 590 155\nL 590 190\nL 490 190\nL 490 155\" style=\"stroke:none;fill:white\"/><text x=\"10\" y=\"22\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Name</text><text x=\"130\" y=\"22\" style=\"stroke:none;fill:black;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Age</text><text x=\"250\" y=\"22\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Address</text><text x=\"370\" y=\"22\" style=\"stroke:none;fill:black;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Tag</text><text x=\"490\" y=\"22\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Action</text><text x=\"10\" y=\"57\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">John Brown</text><text x=\"130\" y=\"57\" style=\"stroke:none;fill:purple;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">32</text><text x=\"250\" y=\"57\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">New York No.</text><text x=\"250\" y=\"77\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">1 Lake Park</text><text x=\"370\" y=\"57\" style=\"stroke:none;fill:purple;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">nice,</text><text x=\"370\" y=\"77\" style=\"stroke:none;fill:purple;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">developer</text><text x=\"490\" y=\"57\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Send Mail</text><text x=\"10\" y=\"112\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Jim Green</text><text x=\"130\" y=\"112\" style=\"stroke:none;fill:black;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">42</text><text x=\"250\" y=\"112\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">London No. 1</text><text x=\"250\" y=\"132\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Lake Park</text><text x=\"370\" y=\"112\" style=\"stroke:none;fill:black;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">wow</text><text x=\"490\" y=\"112\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Send Mail</text><text x=\"10\" y=\"167\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Joe Black</text><text x=\"130\" y=\"167\" style=\"stroke:none;fill:purple;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">32</text><text x=\"250\" y=\"167\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Sidney No. 1</text><text x=\"250\" y=\"187\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Lake Park</text><text x=\"370\" y=\"167\" style=\"stroke:none;fill:purple;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">cool, teacher</text><text x=\"490\" y=\"167\" style=\"stroke:none;fill:blue;font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Send Mail</text></svg>",
},
{
name: "error_no_header",
makeOptions: func() TableChartOption {
opt := makeDefaultTableChartOptions()
opt.Header = nil
return opt
},
errorExpected: true,
},
}
for i, tt := range tests {
painterOptions := PainterOptions{
OutputFormat: ChartOutputSVG,
Width: 600,
Height: 400,
}
runName := strconv.Itoa(i) + "-" + tt.name
if tt.theme != nil {
t.Run(runName+"-theme_painter", func(t *testing.T) {
p := NewPainter(painterOptions, PainterThemeOption(tt.theme))
opt := tt.makeOptions()
validateTableChartRender(t, p, opt, tt.result, tt.errorExpected)
})
runName += "-theme_opt"
}
t.Run(runName, func(t *testing.T) {
p := NewPainter(painterOptions)
opt := tt.makeOptions()
opt.Theme = tt.theme
validateTableChartRender(t, p, opt, tt.result, tt.errorExpected)
})
}
}
func validateTableChartRender(t *testing.T, p *Painter, opt TableChartOption,
expectedResult string, errorExpected bool) {
t.Helper()
err := p.TableChart(opt)
if errorExpected {
require.Error(t, err)
return
}
require.NoError(t, err)
data, err := p.Bytes()
require.NoError(t, err)
assertEqualSVG(t, expectedResult, data)
}