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

pass an error code to Session.CloseWithError

Browse source
This commit is contained in:
Marten Seemann 2018-07-04 17:47:10 +07:00
parent 8b2992a243
commit 6d3d6dc47d
7 changed files with 27 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

8
session_test.go
View file

@ -535,9 +535,9 @@ var _ = Describe("Session", func() {
It("closes streams with proper error", func() {
testErr := errors.New("test error")
streamManager.EXPECT().CloseWithError(qerr.Error(qerr.InternalError, testErr.Error()))
streamManager.EXPECT().CloseWithError(qerr.Error(0x1337, testErr.Error()))
sessionRunner.EXPECT().removeConnectionID(gomock.Any())
sess.CloseWithError(testErr)
sess.CloseWithError(0x1337, testErr)
Eventually(areSessionsRunning).Should(BeFalse())
Expect(sess.Context().Done()).To(BeClosed())
})
@ -1476,12 +1476,12 @@ var _ = Describe("Session", func() {
go func() {
defer GinkgoRecover()
err := sess.run()
Expect(err).To(MatchError(testErr))
Expect(err).To(MatchError(qerr.Error(0x1337, testErr.Error())))
close(done)
}()
streamManager.EXPECT().CloseWithError(gomock.Any())
sessionRunner.EXPECT().removeConnectionID(gomock.Any())
Expect(sess.CloseWithError(testErr)).To(Succeed())
Expect(sess.CloseWithError(0x1337, testErr)).To(Succeed())
Eventually(done).Should(BeClosed())
})