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
|
// Copyright 2021-2023 The Connect Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package connect
import (
"bytes"
"io"
"testing"
"connectrpc.com/connect/internal/assert"
)
func TestEnvelope(t *testing.T) {
t.Parallel()
payload := []byte(`{"number": 42}`)
head := makeEnvelopePrefix(0, len(payload))
buf := &bytes.Buffer{}
buf.Write(head[:])
buf.Write(payload)
t.Run("read", func(t *testing.T) {
t.Parallel()
t.Run("full", func(t *testing.T) {
t.Parallel()
env := &envelope{Data: &bytes.Buffer{}}
rdr := envelopeReader{
reader: bytes.NewReader(buf.Bytes()),
}
assert.Nil(t, rdr.Read(env))
assert.Equal(t, payload, env.Data.Bytes())
})
t.Run("byteByByte", func(t *testing.T) {
t.Parallel()
env := &envelope{Data: &bytes.Buffer{}}
rdr := envelopeReader{
reader: byteByByteReader{
reader: bytes.NewReader(buf.Bytes()),
},
}
assert.Nil(t, rdr.Read(env))
assert.Equal(t, payload, env.Data.Bytes())
})
})
t.Run("write", func(t *testing.T) {
t.Parallel()
t.Run("full", func(t *testing.T) {
t.Parallel()
dst := &bytes.Buffer{}
wtr := envelopeWriter{
sender: writeSender{writer: dst},
}
env := &envelope{Data: bytes.NewBuffer(payload)}
err := wtr.Write(env)
assert.Nil(t, err)
assert.Equal(t, buf.Bytes(), dst.Bytes())
})
t.Run("partial", func(t *testing.T) {
t.Parallel()
dst := &bytes.Buffer{}
env := &envelope{Data: bytes.NewBuffer(payload)}
_, err := io.CopyN(dst, env, 2)
assert.Nil(t, err)
_, err = env.WriteTo(dst)
assert.Nil(t, err)
assert.Equal(t, buf.Bytes(), dst.Bytes())
})
})
t.Run("seek", func(t *testing.T) {
t.Parallel()
t.Run("start", func(t *testing.T) {
t.Parallel()
dst1 := &bytes.Buffer{}
dst2 := &bytes.Buffer{}
env := &envelope{Data: bytes.NewBuffer(payload)}
_, err := io.CopyN(dst1, env, 2)
assert.Nil(t, err)
assert.Equal(t, env.Len(), len(payload)+3)
_, err = env.Seek(0, io.SeekStart)
assert.Nil(t, err)
assert.Equal(t, env.Len(), len(payload)+5)
_, err = io.CopyN(dst2, env, 2)
assert.Nil(t, err)
assert.Equal(t, dst1.Bytes(), dst2.Bytes())
_, err = env.WriteTo(dst2)
assert.Nil(t, err)
assert.Equal(t, dst2.Bytes(), buf.Bytes())
assert.Equal(t, env.Len(), 0)
})
})
}
// byteByByteReader is test reader that reads a single byte at a time.
type byteByByteReader struct {
reader io.ByteReader
}
func (b byteByByteReader) Read(data []byte) (int, error) {
if len(data) == 0 {
return 0, nil
}
next, err := b.reader.ReadByte()
if err != nil {
return 0, err
}
data[0] = next
return 1, nil
}
|
