Browse Source

update pipe test

Darien Raymond 7 years ago
parent
commit
8d8eb0f35a
1 changed files with 7 additions and 8 deletions
  1. 7 8
      transport/pipe/pipe_test.go

+ 7 - 8
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
 	}))()