From 8d8eb0f35ac0de85fc80b69aaffc89684b24d074 Mon Sep 17 00:00:00 2001
From: Darien Raymond <admin@v2ray.com>
Date: Mon, 19 Nov 2018 16:29:27 +0100
Subject: [PATCH] update pipe test

---
 transport/pipe/pipe_test.go | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/transport/pipe/pipe_test.go b/transport/pipe/pipe_test.go
index 3a16041d..06521d37 100644
--- a/transport/pipe/pipe_test.go
+++ b/transport/pipe/pipe_test.go
@@ -76,17 +76,16 @@ func TestPipeLimitZero(t *testing.T) {
 	}, func() error {
 		time.Sleep(time.Second)
 
-		rb, err := pReader.ReadMultiBuffer()
-		if err != nil {
+		var container buf.MultiBufferContainer
+		if err := buf.Copy(pReader, &container); err != nil {
 			return err
 		}
-		assert(rb.String(), Equals, "ab")
 
-		rb, err = pReader.ReadMultiBuffer()
-		if err != nil {
-			return err
-		}
-		assert(rb.String(), Equals, "cd")
+		assert(container.String(), Equals, "abcd")
+		return nil
+	}, func() error {
+		time.Sleep(time.Second * 2)
+		pWriter.Close()
 		return nil
 	}))()