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
|
package term
import (
"reflect"
"testing"
"src.elv.sh/pkg/ui"
)
var bufferBuilderWritesTests = []struct {
bb *BufferBuilder
text string
style string
want *Buffer
}{
// Writing nothing.
{NewBufferBuilder(10), "", "", &Buffer{Width: 10, Lines: Lines{Line{}}}},
// Writing a single rune.
{NewBufferBuilder(10), "a", "1",
&Buffer{Width: 10, Lines: Lines{Line{Cell{"a", "1"}}}}},
// Writing control character.
{NewBufferBuilder(10), "\033", "",
&Buffer{Width: 10, Lines: Lines{Line{Cell{"^[", "7"}}}}},
// Writing styled control character.
{NewBufferBuilder(10), "a\033b", "1",
&Buffer{Width: 10, Lines: Lines{Line{
Cell{"a", "1"},
Cell{"^[", "1;7"},
Cell{"b", "1"}}}}},
// Writing text containing a newline.
{NewBufferBuilder(10), "a\nb", "1",
&Buffer{Width: 10, Lines: Lines{
Line{Cell{"a", "1"}}, Line{Cell{"b", "1"}}}}},
// Writing text containing a newline when there is indent.
{NewBufferBuilder(10).SetIndent(2), "a\nb", "1",
&Buffer{Width: 10, Lines: Lines{
Line{Cell{"a", "1"}},
Line{Cell{" ", ""}, Cell{" ", ""}, Cell{"b", "1"}},
}}},
// Writing long text that triggers wrapping.
{NewBufferBuilder(4), "aaaab", "1",
&Buffer{Width: 4, Lines: Lines{
Line{Cell{"a", "1"}, Cell{"a", "1"}, Cell{"a", "1"}, Cell{"a", "1"}},
Line{Cell{"b", "1"}}}}},
// Writing long text that triggers wrapping when there is indent.
{NewBufferBuilder(4).SetIndent(2), "aaaab", "1",
&Buffer{Width: 4, Lines: Lines{
Line{Cell{"a", "1"}, Cell{"a", "1"}, Cell{"a", "1"}, Cell{"a", "1"}},
Line{Cell{" ", ""}, Cell{" ", ""}, Cell{"b", "1"}}}}},
// Writing long text that triggers eager wrapping.
{NewBufferBuilder(4).SetIndent(2).SetEagerWrap(true), "aaaa", "1",
&Buffer{Width: 4, Lines: Lines{
Line{Cell{"a", "1"}, Cell{"a", "1"}, Cell{"a", "1"}, Cell{"a", "1"}},
Line{Cell{" ", ""}, Cell{" ", ""}}}}},
}
// TestBufferBuilderWrites tests BufferBuilder.Writes by calling Writes on a
// BufferBuilder and see if the built Buffer matches what is expected.
func TestBufferBuilderWrites(t *testing.T) {
for _, test := range bufferBuilderWritesTests {
bb := cloneBufferBuilder(test.bb)
bb.WriteStringSGR(test.text, test.style)
buf := bb.Buffer()
if !reflect.DeepEqual(buf, test.want) {
t.Errorf("buf.writes(%q, %q) makes it %v, want %v",
test.text, test.style, buf, test.want)
}
}
}
var styles = ui.RuneStylesheet{
'-': ui.Underlined,
}
var bufferBuilderTests = []struct {
name string
builder *BufferBuilder
wantBuf *Buffer
}{
{
"MarkLines",
NewBufferBuilder(10).MarkLines(
"foo ", styles,
"-- ", DotHere, "\n",
"",
"bar",
),
&Buffer{Width: 10, Dot: Pos{0, 4}, Lines: Lines{
Line{Cell{"f", "4"}, Cell{"o", "4"}, Cell{"o", ""}, Cell{" ", ""}},
Line{Cell{"b", ""}, Cell{"a", ""}, Cell{"r", ""}},
}},
},
}
func TestBufferBuilder(t *testing.T) {
for _, test := range bufferBuilderTests {
t.Run(test.name, func(t *testing.T) {
buf := test.builder.Buffer()
if !reflect.DeepEqual(buf, test.wantBuf) {
t.Errorf("Got buf %v, want %v", buf, test.wantBuf)
}
})
}
}
func cloneBufferBuilder(bb *BufferBuilder) *BufferBuilder {
return &BufferBuilder{
bb.Width, bb.Col, bb.Indent,
bb.EagerWrap, cloneLines(bb.Lines), bb.Dot}
}
|
