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
|
package strings
import (
"strings"
"testing"
"unicode"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func FuzzIndent(f *testing.F) {
f.Add(0, " ", "foo\n")
f.Add(1, " ", "bar\nbaz\nqux\n")
f.Add(1, " ", "\n")
f.Add(3, " ", "quux\n")
f.Add(15, "\n0", "\n0")
f.Fuzz(func(t *testing.T, width int, indent, s string) {
out, err := Indent(width, indent, s)
if width <= 0 {
require.Error(t, err)
return
}
if strings.Contains(indent, "\n") {
require.Error(t, err)
return
}
require.NoError(t, err)
// out should be equal to s when both have the indent character
// completely removed.
assert.Equal(t,
strings.ReplaceAll(s, indent, ""),
strings.ReplaceAll(out, indent, ""),
)
})
}
func FuzzTrunc(f *testing.F) {
f.Add(0, "foo")
f.Fuzz(func(t *testing.T, length int, s string) {
out := Trunc(length, s)
assert.LessOrEqual(t, len(out), len(s))
if length >= 0 {
assert.LessOrEqual(t, len(out), length)
}
assert.Equal(t, s[0:len(out)], out)
})
}
func FuzzWordWrap(f *testing.F) {
out := `There shouldn't be any wrapping of long words or URLs because that would break
things very badly. To wit:
https://example.com/a/super-long/url/that-shouldnt-be?wrapped=for+fear+of#the-breaking-of-functionality
should appear on its own line, regardless of the desired word-wrapping width
that has been set.`
f.Add(out, "", uint32(0))
f.Add(out, "\n", uint32(80))
f.Add(out, "\v", uint32(10))
f.Fuzz(func(t *testing.T, in, lbSeq string, width uint32) {
for _, r := range lbSeq {
if !unicode.IsSpace(r) {
t.Skip("ignore non-whitespace sequences")
}
}
out := WordWrap(in, WordWrapOpts{
LBSeq: lbSeq,
Width: width,
})
if lbSeq == "" {
lbSeq = "\n"
}
// compare by stripping both the line-break sequence and spaces
assert.Equal(t,
strings.ReplaceAll(strings.ReplaceAll(in, lbSeq, ""), " ", ""),
strings.ReplaceAll(strings.ReplaceAll(out, lbSeq, ""), " ", ""),
)
})
}
|
