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
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
|
package sideband
import (
"bytes"
"errors"
"io"
"testing"
"github.com/go-git/go-git/v5/plumbing/format/pktline"
. "gopkg.in/check.v1"
)
func Test(t *testing.T) { TestingT(t) }
type SidebandSuite struct{}
var _ = Suite(&SidebandSuite{})
func (s *SidebandSuite) TestDecode(c *C) {
expected := []byte("abcdefghijklmnopqrstuvwxyz")
buf := bytes.NewBuffer(nil)
e := pktline.NewEncoder(buf)
e.Encode(PackData.WithPayload(expected[0:8]))
e.Encode(ProgressMessage.WithPayload([]byte{'F', 'O', 'O', '\n'}))
e.Encode(PackData.WithPayload(expected[8:16]))
e.Encode(PackData.WithPayload(expected[16:26]))
content := make([]byte, 26)
d := NewDemuxer(Sideband64k, buf)
n, err := io.ReadFull(d, content)
c.Assert(err, IsNil)
c.Assert(n, Equals, 26)
c.Assert(content, DeepEquals, expected)
}
func (s *SidebandSuite) TestDecodeMoreThanContain(c *C) {
expected := []byte("abcdefghijklmnopqrstuvwxyz")
buf := bytes.NewBuffer(nil)
e := pktline.NewEncoder(buf)
e.Encode(PackData.WithPayload(expected))
content := make([]byte, 42)
d := NewDemuxer(Sideband64k, buf)
n, err := io.ReadFull(d, content)
c.Assert(err, Equals, io.ErrUnexpectedEOF)
c.Assert(n, Equals, 26)
c.Assert(content[0:26], DeepEquals, expected)
}
func (s *SidebandSuite) TestDecodeWithError(c *C) {
expected := []byte("abcdefghijklmnopqrstuvwxyz")
buf := bytes.NewBuffer(nil)
e := pktline.NewEncoder(buf)
e.Encode(PackData.WithPayload(expected[0:8]))
e.Encode(ErrorMessage.WithPayload([]byte{'F', 'O', 'O', '\n'}))
e.Encode(PackData.WithPayload(expected[8:16]))
e.Encode(PackData.WithPayload(expected[16:26]))
content := make([]byte, 26)
d := NewDemuxer(Sideband64k, buf)
n, err := io.ReadFull(d, content)
c.Assert(err, ErrorMatches, "unexpected error: FOO\n")
c.Assert(n, Equals, 8)
c.Assert(content[0:8], DeepEquals, expected[0:8])
}
type mockReader struct{}
func (r *mockReader) Read([]byte) (int, error) { return 0, errors.New("foo") }
func (s *SidebandSuite) TestDecodeFromFailingReader(c *C) {
content := make([]byte, 26)
d := NewDemuxer(Sideband64k, &mockReader{})
n, err := io.ReadFull(d, content)
c.Assert(err, ErrorMatches, "foo")
c.Assert(n, Equals, 0)
}
func (s *SidebandSuite) TestDecodeWithProgress(c *C) {
expected := []byte("abcdefghijklmnopqrstuvwxyz")
input := bytes.NewBuffer(nil)
e := pktline.NewEncoder(input)
e.Encode(PackData.WithPayload(expected[0:8]))
e.Encode(ProgressMessage.WithPayload([]byte{'F', 'O', 'O', '\n'}))
e.Encode(PackData.WithPayload(expected[8:16]))
e.Encode(PackData.WithPayload(expected[16:26]))
output := bytes.NewBuffer(nil)
content := make([]byte, 26)
d := NewDemuxer(Sideband64k, input)
d.Progress = output
n, err := io.ReadFull(d, content)
c.Assert(err, IsNil)
c.Assert(n, Equals, 26)
c.Assert(content, DeepEquals, expected)
progress, err := io.ReadAll(output)
c.Assert(err, IsNil)
c.Assert(progress, DeepEquals, []byte{'F', 'O', 'O', '\n'})
}
func (s *SidebandSuite) TestDecodeFlushEOF(c *C) {
expected := []byte("abcdefghijklmnopqrstuvwxyz")
input := bytes.NewBuffer(nil)
e := pktline.NewEncoder(input)
e.Encode(PackData.WithPayload(expected[0:8]))
e.Encode(ProgressMessage.WithPayload([]byte{'F', 'O', 'O', '\n'}))
e.Encode(PackData.WithPayload(expected[8:16]))
e.Encode(PackData.WithPayload(expected[16:26]))
e.Flush()
e.Encode(PackData.WithPayload([]byte("bar\n")))
output := bytes.NewBuffer(nil)
content := bytes.NewBuffer(nil)
d := NewDemuxer(Sideband64k, input)
d.Progress = output
n, err := content.ReadFrom(d)
c.Assert(err, IsNil)
c.Assert(n, Equals, int64(26))
c.Assert(content.Bytes(), DeepEquals, expected)
progress, err := io.ReadAll(output)
c.Assert(err, IsNil)
c.Assert(progress, DeepEquals, []byte{'F', 'O', 'O', '\n'})
}
func (s *SidebandSuite) TestDecodeWithUnknownChannel(c *C) {
buf := bytes.NewBuffer(nil)
e := pktline.NewEncoder(buf)
e.Encode([]byte{'4', 'F', 'O', 'O', '\n'})
content := make([]byte, 26)
d := NewDemuxer(Sideband64k, buf)
n, err := io.ReadFull(d, content)
c.Assert(err, ErrorMatches, "unknown channel 4FOO\n")
c.Assert(n, Equals, 0)
}
func (s *SidebandSuite) TestDecodeWithPending(c *C) {
expected := []byte("abcdefghijklmnopqrstuvwxyz")
buf := bytes.NewBuffer(nil)
e := pktline.NewEncoder(buf)
e.Encode(PackData.WithPayload(expected[0:8]))
e.Encode(PackData.WithPayload(expected[8:16]))
e.Encode(PackData.WithPayload(expected[16:26]))
content := make([]byte, 13)
d := NewDemuxer(Sideband64k, buf)
n, err := io.ReadFull(d, content)
c.Assert(err, IsNil)
c.Assert(n, Equals, 13)
c.Assert(content, DeepEquals, expected[0:13])
n, err = d.Read(content)
c.Assert(err, IsNil)
c.Assert(n, Equals, 13)
c.Assert(content, DeepEquals, expected[13:26])
}
func (s *SidebandSuite) TestDecodeErrMaxPacked(c *C) {
buf := bytes.NewBuffer(nil)
e := pktline.NewEncoder(buf)
e.Encode(PackData.WithPayload(bytes.Repeat([]byte{'0'}, MaxPackedSize+1)))
content := make([]byte, 13)
d := NewDemuxer(Sideband, buf)
n, err := io.ReadFull(d, content)
c.Assert(err, Equals, ErrMaxPackedExceeded)
c.Assert(n, Equals, 0)
}
|
