mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-04 04:37:36 +03:00
Code Issues Projects Releases Packages Wiki Activity

implement a context for the stream

Browse source 
The context is cancelled when the write-side of the stream is closed.
This commit is contained in:
Marten Seemann 2017-07-29 09:18:33 +07:00
parent e02f5d5fbe
commit 8ef69143ba
5 changed files with 55 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

31
stream_test.go
View file

@ -435,6 +435,12 @@ var _ = Describe("Stream", func() {
Expect(n).To(BeZero())
Expect(err).To(MatchError(io.EOF))
})
It("doesn't cancel the context", func() {
mockFcm.EXPECT().UpdateHighestReceived(streamID, protocol.ByteCount(0))
str.CloseRemote(0)
Expect(str.Context().Done()).ToNot(BeClosed())
})
})
})
@ -463,6 +469,12 @@ var _ = Describe("Stream", func() {
Expect(n).To(BeZero())
Expect(err).To(MatchError(testErr))
})
It("cancels the context", func() {
Expect(str.Context().Done()).ToNot(BeClosed())
str.Cancel(testErr)
Expect(str.Context().Done()).To(BeClosed())
})
})
})
@ -713,6 +725,12 @@ var _ = Describe("Stream", func() {
str.Reset(testErr)
Expect(resetCalled).To(BeFalse())
})
It("cancels the context", func() {
Expect(str.Context().Done()).ToNot(BeClosed())
str.Reset(testErr)
Expect(str.Context().Done()).To(BeClosed())
})
})
})
@ -958,6 +976,13 @@ var _ = Describe("Stream", func() {
Expect(str.finished()).To(BeFalse())
})
It("cancels the context after it is closed", func() {
Expect(str.Context().Done()).ToNot(BeClosed())
str.Close()
str.sentFin()
Expect(str.Context().Done()).To(BeClosed())
})
It("is not finished if it is only closed for reading", func() {
mockFcm.EXPECT().UpdateHighestReceived(streamID, protocol.ByteCount(0))
mockFcm.EXPECT().AddBytesRead(streamID, protocol.ByteCount(0))
@ -972,6 +997,12 @@ var _ = Describe("Stream", func() {
Expect(str.finished()).To(BeTrue())
})
It("cancels the context after receiving a RST", func() {
Expect(str.Context().Done()).ToNot(BeClosed())
str.RegisterRemoteError(testErr)
Expect(str.Context().Done()).To(BeClosed())
})
It("is finished after being locally reset and receiving a RST in response", func() {
str.Reset(testErr)
Expect(str.finished()).To(BeFalse())