Update io_test.go

9 years ago · a2d8d55ba1
parent 2f47074c98
commit a2d8d55ba1
1 changed files with 2 additions and 0 deletions
--- a/proxy/vmess/io/io_test.go
+++ b/proxy/vmess/io/io_test.go
@ -32,6 +32,7 @@ func TestSingleIO(t *testing.T) {

 	writer := NewAuthChunkWriter(v2io.NewAdaptiveWriter(content))
 	writer.Write(alloc.NewBuffer().Clear().AppendString("abcd"))
+	writer.Write(alloc.NewBuffer().Clear())
 	writer.Release()

 	reader := NewAuthChunkReader(content)
@ -90,6 +91,7 @@ func TestLargeIO(t *testing.T) {
 			break
 		}
 	}
+	writer.Write(alloc.NewBuffer().Clear())
 	writer.Release()

 	actualContent := make([]byte, 0, len(content))