1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2026-01-29 20:35:31 -05:00

cleanup buffer usage

This commit is contained in:
Darien Raymond
2017-11-09 22:33:15 +01:00
parent 6e61538b36
commit 594ec15c09
41 changed files with 358 additions and 529 deletions

View File

@@ -19,26 +19,26 @@ func TestChunkStreamIO(t *testing.T) {
b := buf.New()
b.AppendBytes('a', 'b', 'c', 'd')
assert(writer.Write(buf.NewMultiBufferValue(b)), IsNil)
assert(writer.WriteMultiBuffer(buf.NewMultiBufferValue(b)), IsNil)
b = buf.New()
b.AppendBytes('e', 'f', 'g')
assert(writer.Write(buf.NewMultiBufferValue(b)), IsNil)
assert(writer.WriteMultiBuffer(buf.NewMultiBufferValue(b)), IsNil)
assert(writer.Write(buf.MultiBuffer{}), IsNil)
assert(writer.WriteMultiBuffer(buf.MultiBuffer{}), IsNil)
assert(cache.Len(), Equals, 13)
mb, err := reader.Read()
mb, err := reader.ReadMultiBuffer()
assert(err, IsNil)
assert(mb.Len(), Equals, 4)
assert(mb[0].Bytes(), Equals, []byte("abcd"))
mb, err = reader.Read()
mb, err = reader.ReadMultiBuffer()
assert(err, IsNil)
assert(mb.Len(), Equals, 3)
assert(mb[0].Bytes(), Equals, []byte("efg"))
_, err = reader.Read()
_, err = reader.ReadMultiBuffer()
assert(err, Equals, io.EOF)
}