refactor io package

common/bufio/buffered_reader_test.go

@@ -0,0 +1,37 @@
+package bufio_test
+
+import (
+	"crypto/rand"
+	"testing"
+
+	"v2ray.com/core/common/buf"
+	. "v2ray.com/core/common/bufio"
+	"v2ray.com/core/testing/assert"
+)
+
+func TestBufferedReader(t *testing.T) {
+	assert := assert.On(t)
+
+	content := buf.New()
+	content.AppendSupplier(buf.ReadFrom(rand.Reader))
+
+	len := content.Len()
+
+	reader := NewReader(content)
+	assert.Bool(reader.Cached()).IsTrue()
+
+	payload := make([]byte, 16)
+
+	nBytes, err := reader.Read(payload)
+	assert.Int(nBytes).Equals(16)
+	assert.Error(err).IsNil()
+
+	len2 := content.Len()
+	assert.Int(len - len2).GreaterThan(16)
+
+	nBytes, err = reader.Read(payload)
+	assert.Int(nBytes).Equals(16)
+	assert.Error(err).IsNil()
+
+	assert.Int(content.Len()).Equals(len2)
+}