gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

add a context to Session.Accept{Uni}Stream

Browse Source
This commit is contained in:
Marten Seemann
2019-05-28 17:32:18 +01:00
parent f74082b2fb
commit 5550ba2c3b
28 changed files with 140 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
integrationtests/self/timeout_test.go
View File

@@ -95,7 +95,7 @@ var _ = Describe("Timeout tests", func() {
&quic.Config{IdleTimeout: idleTimeout},
)
Expect(err).ToNot(HaveOccurred())
strIn, err := sess.AcceptStream()
strIn, err := sess.AcceptStream(context.Background())
Expect(err).ToNot(HaveOccurred())
strOut, err := sess.OpenStream()
Expect(err).ToNot(HaveOccurred())
@@ -116,9 +116,9 @@ var _ = Describe("Timeout tests", func() {
checkTimeoutError(err)
_, err = sess.OpenUniStream()
checkTimeoutError(err)
_, err = sess.AcceptStream()
_, err = sess.AcceptStream(context.Background())
checkTimeoutError(err)
_, err = sess.AcceptUniStream()
_, err = sess.AcceptUniStream(context.Background())
checkTimeoutError(err)
})
@@ -148,7 +148,7 @@ var _ = Describe("Timeout tests", func() {
defer GinkgoRecover()
sess, err := server.Accept(context.Background())
Expect(err).ToNot(HaveOccurred())
sess.AcceptStream() // blocks until the session is closed
sess.AcceptStream(context.Background()) // blocks until the session is closed
close(serverSessionClosed)
}()
@@ -162,7 +162,7 @@ var _ = Describe("Timeout tests", func() {
done := make(chan struct{})
go func() {
defer GinkgoRecover()
_, err := sess.AcceptStream()
_, err := sess.AcceptStream(context.Background())
checkTimeoutError(err)
close(done)
}()
@@ -189,7 +189,7 @@ var _ = Describe("Timeout tests", func() {
defer GinkgoRecover()
sess, err := server.Accept(context.Background())
Expect(err).ToNot(HaveOccurred())
sess.AcceptStream() // blocks until the session is closed
sess.AcceptStream(context.Background()) // blocks until the session is closed
close(serverSessionClosed)
}()
@@ -212,7 +212,7 @@ var _ = Describe("Timeout tests", func() {
done := make(chan struct{})
go func() {
defer GinkgoRecover()
_, err := sess.AcceptStream()
_, err := sess.AcceptStream(context.Background())
checkTimeoutError(err)
close(done)
}()