Browse Source

fix broken test

Darien Raymond 7 years ago
parent
commit
50e77cbb19
1 changed files with 3 additions and 2 deletions
  1. 3 2
      transport/pipe/pipe_test.go

+ 3 - 2
transport/pipe/pipe_test.go

@@ -1,6 +1,7 @@
 package pipe_test
 package pipe_test
 
 
 import (
 import (
+	"context"
 	"io"
 	"io"
 	"sync"
 	"sync"
 	"testing"
 	"testing"
@@ -73,7 +74,7 @@ func TestPipeLimitZero(t *testing.T) {
 	bb.Write([]byte{'a', 'b'})
 	bb.Write([]byte{'a', 'b'})
 	assert(pWriter.WriteMultiBuffer(buf.MultiBuffer{bb}), IsNil)
 	assert(pWriter.WriteMultiBuffer(buf.MultiBuffer{bb}), IsNil)
 
 
-	err := task.Run(task.Parallel(func() error {
+	err := task.Run(context.Background(), func() error {
 		b := buf.New()
 		b := buf.New()
 		b.Write([]byte{'c', 'd'})
 		b.Write([]byte{'c', 'd'})
 		return pWriter.WriteMultiBuffer(buf.MultiBuffer{b})
 		return pWriter.WriteMultiBuffer(buf.MultiBuffer{b})
@@ -91,7 +92,7 @@ func TestPipeLimitZero(t *testing.T) {
 		time.Sleep(time.Second * 2)
 		time.Sleep(time.Second * 2)
 		pWriter.Close()
 		pWriter.Close()
 		return nil
 		return nil
-	}))()
+	})
 
 
 	assert(err, IsNil)
 	assert(err, IsNil)
 }
 }