add a context to Listener.Accept

This commit is contained in:
Marten Seemann
2019-05-28 16:03:38 +02:00
parent 8dbe1684be
commit 12bce1caaa
21 changed files with 74 additions and 40 deletions

View File

@@ -1,6 +1,7 @@
package self_test
import (
"context"
"fmt"
"io/ioutil"
"net"
@@ -92,7 +93,7 @@ var _ = Describe("Bidirectional streams", func() {
go func() {
defer GinkgoRecover()
var err error
sess, err = server.Accept()
sess, err = server.Accept(context.Background())
Expect(err).ToNot(HaveOccurred())
runReceivingPeer(sess)
}()
@@ -109,7 +110,7 @@ var _ = Describe("Bidirectional streams", func() {
It(fmt.Sprintf("server opening %d streams to a client", numStreams), func() {
go func() {
defer GinkgoRecover()
sess, err := server.Accept()
sess, err := server.Accept(context.Background())
Expect(err).ToNot(HaveOccurred())
runSendingPeer(sess)
sess.Close()
@@ -129,7 +130,7 @@ var _ = Describe("Bidirectional streams", func() {
done1 := make(chan struct{})
go func() {
defer GinkgoRecover()
sess, err := server.Accept()
sess, err := server.Accept(context.Background())
Expect(err).ToNot(HaveOccurred())
done := make(chan struct{})
go func() {