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
|
package wordwrap
import (
"strings"
"testing"
)
func TestWrapString(t *testing.T) {
cases := []struct {
Input, Output string
Lim uint
}{
// A simple word passes through.
{
"foo",
"foo",
4,
},
// A single word that is too long passes through.
// We do not break words.
{
"foobarbaz",
"foobarbaz",
4,
},
// Lines are broken at whitespace.
{
"foo bar baz",
"foo\nbar\nbaz",
4,
},
// Lines are broken at whitespace, even if words
// are too long. We do not break words.
{
"foo bars bazzes",
"foo\nbars\nbazzes",
4,
},
// A word that would run beyond the width is wrapped.
{
"fo sop",
"fo\nsop",
4,
},
// Do not break on non-breaking space.
{
"foo bar\u00A0baz",
"foo\nbar\u00A0baz",
10,
},
// Whitespace that trails a line and fits the width
// passes through, as does whitespace prefixing an
// explicit line break. A tab counts as one character.
{
"foo\nb\t r\n baz",
"foo\nb\t r\n baz",
4,
},
// Trailing whitespace is removed if it doesn't fit the width.
// Runs of whitespace on which a line is broken are removed.
{
"foo \nb ar ",
"foo\nb\nar",
4,
},
// An explicit line break at the end of the input is preserved.
{
"foo bar baz\n",
"foo\nbar\nbaz\n",
4,
},
// Explicit break are always preserved.
{
"\nfoo bar\n\n\nbaz\n",
"\nfoo\nbar\n\n\nbaz\n",
4,
},
// Complete example:
{
" This is a list: \n\n\t* foo\n\t* bar\n\n\n\t* baz \nBAM ",
" This\nis a\nlist: \n\n\t* foo\n\t* bar\n\n\n\t* baz\nBAM",
6,
},
// Multi-byte characters
{
strings.Repeat("\u2584 ", 4),
"\u2584 \u2584" + "\n" +
strings.Repeat("\u2584 ", 2),
4,
},
}
for i, tc := range cases {
actual := WrapString(tc.Input, tc.Lim)
if actual != tc.Output {
t.Fatalf("Case %d Input:\n\n`%s`\n\nExpected Output:\n\n`%s`\n\nActual Output:\n\n`%s`", i, tc.Input, tc.Output, actual)
}
}
}
|
