/
truncate_test.go
93 lines (82 loc) · 1.9 KB
/
truncate_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
package stringfy_test
import (
"testing"
"github.com/hgsigner/stringfy"
)
func Test_Truncate(t *testing.T) {
tests := []struct {
in string
length int
addLength bool
omission string
addOmission bool
separator string
addSeparator bool
out string
}{
{
in: "Lorem ipsum.",
out: "Lorem ipsum.",
},
{
in: "Lor",
addLength: true,
length: 3,
out: "Lor",
},
{
in: "Lorem ipsum dolor sit amet, consectetur adipiscing elit.",
out: "Lorem ipsum dolor sit amet,...",
},
{
in: "Lorem ipsum dolor sit amet, consectetur adipiscing elit.",
addOmission: true,
omission: "... (continued)",
out: "Lorem ipsum dol... (continued)",
},
{
in: "Lorem ipsum dolor sit amet, consectetur adipiscing elit.",
addLength: true,
length: 13,
addSeparator: true,
separator: " ",
out: "Lorem ipsum...",
},
{
in: "Lorem ipsum dolor sit amet, consectetur adipiscing elit.",
addLength: true,
length: 13,
addSeparator: true,
separator: " ",
addOmission: true,
omission: "... (continued) ...",
out: "Lorem ipsum... (continued) ...",
},
{
in: "Lorem ipsum dolor sit amet, c",
addLength: true,
length: 13,
addSeparator: true,
separator: " ",
addOmission: true,
omission: "... (continued) ...",
out: "Lorem ipsum dolor sit amet, c",
},
}
for _, test := range tests {
tr := stringfy.NewTruncate()
if test.addLength {
tr.Options(stringfy.AddLength(test.length))
}
if test.addOmission {
tr.Options(stringfy.AddOmission(test.omission))
}
if test.addSeparator {
tr.Options(stringfy.AddSeparator(test.separator))
}
trf := tr.Perform(test.in)
if trf != test.out {
t.Errorf("\nExpected: %s\nGot: %s", test.out, trf)
}
}
}