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"
"math/rand"
"net"
@@ -33,7 +34,7 @@ var _ = Describe("Stateless Resets", func() {
go func() {
defer GinkgoRecover()
sess, err := ln.Accept()
sess, err := ln.Accept(context.Background())
Expect(err).ToNot(HaveOccurred())
str, err := sess.OpenStream()
Expect(err).ToNot(HaveOccurred())
@@ -86,7 +87,7 @@ var _ = Describe("Stateless Resets", func() {
acceptStopped := make(chan struct{})
go func() {
defer GinkgoRecover()
_, err := ln2.Accept()
_, err := ln2.Accept(context.Background())
Expect(err).To(HaveOccurred())
close(acceptStopped)
}()