-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathfunnel_chart_test.go
140 lines (121 loc) · 9 KB
/
funnel_chart_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
package charts
import (
"strconv"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func makeBasicFunnelChartOption() FunnelChartOption {
return FunnelChartOption{
SeriesList: NewSeriesListFunnel([]float64{
100, 80, 60, 40, 20,
}),
Legend: LegendOption{
SeriesNames: []string{"Show", "Click", "Visit", "Inquiry", "Order"},
},
Title: TitleOption{
Text: "Funnel",
},
}
}
func TestNewFunnelChartOptionWithData(t *testing.T) {
t.Parallel()
opt := NewFunnelChartOptionWithData([]float64{12, 24, 48})
assert.Len(t, opt.SeriesList, 3)
assert.Equal(t, ChartTypeFunnel, opt.SeriesList[0].getType())
assert.Equal(t, defaultPadding, opt.Padding)
p := NewPainter(PainterOptions{})
assert.NoError(t, p.FunnelChart(opt))
}
func TestFunnelChart(t *testing.T) {
t.Parallel()
tests := []struct {
name string
themed bool
makeOptions func() FunnelChartOption
result string
}{
{
name: "basic_themed",
themed: true,
makeOptions: makeBasicFunnelChartOption,
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 400\nL 0 400\nL 0 0\" style=\"stroke:none;fill:rgb(40,40,40)\"/><path d=\"M 86 3\nL 116 3\nL 116 16\nL 86 16\nL 86 3\" style=\"stroke:none;fill:rgb(255,100,100)\"/><text x=\"118\" y=\"15\" style=\"stroke:none;fill:rgb(238,238,238);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Show</text><path d=\"M 176 3\nL 206 3\nL 206 16\nL 176 16\nL 176 3\" style=\"stroke:none;fill:rgb(255,210,100)\"/><text x=\"208\" y=\"15\" style=\"stroke:none;fill:rgb(238,238,238);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Click</text><path d=\"M 262 3\nL 292 3\nL 292 16\nL 262 16\nL 262 3\" style=\"stroke:none;fill:rgb(100,180,210)\"/><text x=\"294\" y=\"15\" style=\"stroke:none;fill:rgb(238,238,238);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Visit</text><path d=\"M 345 3\nL 375 3\nL 375 16\nL 345 16\nL 345 3\" style=\"stroke:none;fill:rgb(64,160,110)\"/><text x=\"377\" y=\"15\" style=\"stroke:none;fill:rgb(238,238,238);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Inquiry</text><path d=\"M 444 3\nL 474 3\nL 474 16\nL 444 16\nL 444 3\" style=\"stroke:none;fill:rgb(154,100,180)\"/><text x=\"476\" y=\"15\" style=\"stroke:none;fill:rgb(238,238,238);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Order</text><text x=\"0\" y=\"16\" style=\"stroke:none;fill:rgb(238,238,238);font-size:15.3px;font-family:'Roboto Medium',sans-serif\">Funnel</text><path d=\"M 0 36\nL 600 36\nL 540 107\nL 60 107\nL 0 36\" style=\"stroke:none;fill:rgb(255,100,100)\"/><text x=\"264\" y=\"71\" style=\"stroke:none;fill:rgb(238,238,238);font-size:12.8px;font-family:'Roboto Medium',sans-serif\">Show(100%)</text><path d=\"M 60 109\nL 540 109\nL 480 180\nL 120 180\nL 60 109\" style=\"stroke:none;fill:rgb(255,210,100)\"/><text x=\"269\" y=\"144\" style=\"stroke:none;fill:rgb(238,238,238);font-size:12.8px;font-family:'Roboto Medium',sans-serif\">Click(80%)</text><path d=\"M 120 182\nL 480 182\nL 420 253\nL 180 253\nL 120 182\" style=\"stroke:none;fill:rgb(100,180,210)\"/><text x=\"271\" y=\"217\" style=\"stroke:none;fill:rgb(238,238,238);font-size:12.8px;font-family:'Roboto Medium',sans-serif\">Visit(60%)</text><path d=\"M 180 255\nL 420 255\nL 360 326\nL 240 326\nL 180 255\" style=\"stroke:none;fill:rgb(64,160,110)\"/><text x=\"264\" y=\"290\" style=\"stroke:none;fill:rgb(238,238,238);font-size:12.8px;font-family:'Roboto Medium',sans-serif\">Inquiry(40%)</text><path d=\"M 240 328\nL 360 328\nL 300 399\nL 300 399\nL 240 328\" style=\"stroke:none;fill:rgb(154,100,180)\"/><text x=\"268\" y=\"363\" style=\"stroke:none;fill:rgb(238,238,238);font-size:12.8px;font-family:'Roboto Medium',sans-serif\">Order(20%)</text></svg>",
},
{
name: "custom_legend",
makeOptions: func() FunnelChartOption {
opt := makeBasicFunnelChartOption()
opt.Legend.Symbol = SymbolDot
opt.Legend.FontStyle = NewFontStyleWithSize(4.0)
opt.Legend.Vertical = Ptr(true)
opt.Legend.Offset = OffsetStr{
Left: PositionRight,
Top: PositionBottom,
}
opt.Title.Show = Ptr(false)
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 400\nL 0 400\nL 0 0\" style=\"stroke:none;fill:white\"/><path d=\"M 552 304\nL 582 304\" style=\"stroke-width:3;stroke:rgb(84,112,198);fill:none\"/><circle cx=\"567\" cy=\"304\" r=\"5\" style=\"stroke-width:3;stroke:rgb(84,112,198);fill:rgb(84,112,198)\"/><text x=\"584\" y=\"310\" style=\"stroke:none;fill:rgb(70,70,70);font-size:5.1px;font-family:'Roboto Medium',sans-serif\">Show</text><path d=\"M 552 324\nL 582 324\" style=\"stroke-width:3;stroke:rgb(145,204,117);fill:none\"/><circle cx=\"567\" cy=\"324\" r=\"5\" style=\"stroke-width:3;stroke:rgb(145,204,117);fill:rgb(145,204,117)\"/><text x=\"584\" y=\"330\" style=\"stroke:none;fill:rgb(70,70,70);font-size:5.1px;font-family:'Roboto Medium',sans-serif\">Click</text><path d=\"M 552 344\nL 582 344\" style=\"stroke-width:3;stroke:rgb(250,200,88);fill:none\"/><circle cx=\"567\" cy=\"344\" r=\"5\" style=\"stroke-width:3;stroke:rgb(250,200,88);fill:rgb(250,200,88)\"/><text x=\"584\" y=\"350\" style=\"stroke:none;fill:rgb(70,70,70);font-size:5.1px;font-family:'Roboto Medium',sans-serif\">Visit</text><path d=\"M 552 364\nL 582 364\" style=\"stroke-width:3;stroke:rgb(238,102,102);fill:none\"/><circle cx=\"567\" cy=\"364\" r=\"5\" style=\"stroke-width:3;stroke:rgb(238,102,102);fill:rgb(238,102,102)\"/><text x=\"584\" y=\"370\" style=\"stroke:none;fill:rgb(70,70,70);font-size:5.1px;font-family:'Roboto Medium',sans-serif\">Inquiry</text><path d=\"M 552 384\nL 582 384\" style=\"stroke-width:3;stroke:rgb(115,192,222);fill:none\"/><circle cx=\"567\" cy=\"384\" r=\"5\" style=\"stroke-width:3;stroke:rgb(115,192,222);fill:rgb(115,192,222)\"/><text x=\"584\" y=\"390\" style=\"stroke:none;fill:rgb(70,70,70);font-size:5.1px;font-family:'Roboto Medium',sans-serif\">Order</text><path d=\"M 0 0\nL 600 0\nL 540 78\nL 60 78\nL 0 0\" style=\"stroke:none;fill:rgb(84,112,198)\"/><text x=\"264\" y=\"39\" style=\"stroke:none;fill:rgb(70,70,70);font-size:12.8px;font-family:'Roboto Medium',sans-serif\">Show(100%)</text><path d=\"M 60 80\nL 540 80\nL 480 158\nL 120 158\nL 60 80\" style=\"stroke:none;fill:rgb(145,204,117)\"/><text x=\"269\" y=\"119\" style=\"stroke:none;fill:rgb(70,70,70);font-size:12.8px;font-family:'Roboto Medium',sans-serif\">Click(80%)</text><path d=\"M 120 160\nL 480 160\nL 420 238\nL 180 238\nL 120 160\" style=\"stroke:none;fill:rgb(250,200,88)\"/><text x=\"271\" y=\"199\" style=\"stroke:none;fill:rgb(70,70,70);font-size:12.8px;font-family:'Roboto Medium',sans-serif\">Visit(60%)</text><path d=\"M 180 240\nL 420 240\nL 360 318\nL 240 318\nL 180 240\" style=\"stroke:none;fill:rgb(238,102,102)\"/><text x=\"264\" y=\"279\" style=\"stroke:none;fill:rgb(70,70,70);font-size:12.8px;font-family:'Roboto Medium',sans-serif\">Inquiry(40%)</text><path d=\"M 240 320\nL 360 320\nL 300 398\nL 300 398\nL 240 320\" style=\"stroke:none;fill:rgb(115,192,222)\"/><text x=\"268\" y=\"359\" style=\"stroke:none;fill:rgb(70,70,70);font-size:12.8px;font-family:'Roboto Medium',sans-serif\">Order(20%)</text></svg>",
},
}
for i, tt := range tests {
painterOptions := PainterOptions{
OutputFormat: ChartOutputSVG,
Width: 600,
Height: 400,
}
if tt.themed {
t.Run(strconv.Itoa(i)+"-"+tt.name+"-painter", func(t *testing.T) {
p := NewPainter(painterOptions, PainterThemeOption(GetTheme(ThemeVividDark)))
validateFunnelChartRender(t, p, tt.makeOptions(), tt.result)
})
t.Run(strconv.Itoa(i)+"-"+tt.name+"-options", func(t *testing.T) {
p := NewPainter(painterOptions)
opt := tt.makeOptions()
opt.Theme = GetTheme(ThemeVividDark)
validateFunnelChartRender(t, p, opt, tt.result)
})
} else {
t.Run(strconv.Itoa(i)+"-"+tt.name, func(t *testing.T) {
p := NewPainter(painterOptions)
validateFunnelChartRender(t, p, tt.makeOptions(), tt.result)
})
}
}
}
func validateFunnelChartRender(t *testing.T, p *Painter, opt FunnelChartOption, expectedResult string) {
t.Helper()
err := p.FunnelChart(opt)
require.NoError(t, err)
data, err := p.Bytes()
require.NoError(t, err)
assertEqualSVG(t, expectedResult, data)
}
func TestFunnelChartError(t *testing.T) {
t.Parallel()
tests := []struct {
name string
makeOptions func() FunnelChartOption
errorMsgContains string
}{
{
name: "empty_series",
makeOptions: func() FunnelChartOption {
return NewFunnelChartOptionWithData([]float64{})
},
errorMsgContains: "empty series list",
},
}
for i, tt := range tests {
t.Run(strconv.Itoa(i)+"-"+tt.name, func(t *testing.T) {
p := NewPainter(PainterOptions{
OutputFormat: ChartOutputSVG,
Width: 600,
Height: 400,
})
err := p.FunnelChart(tt.makeOptions())
require.Error(t, err)
require.ErrorContains(t, err, tt.errorMsgContains)
})
}
}