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

implement closing the quic client with an error

Browse source
This commit is contained in:
Marten Seemann 2016-12-17 10:11:31 +07:00
parent c42262c2b3
commit 08c267431b
No known key found for this signature in database
GPG key ID: 3603F40B121FCDEA
5 changed files with 16 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

15
client_test.go
View file

@ -3,7 +3,9 @@ package quic
import (
"bytes"
"encoding/binary"
"errors"
"net"
"runtime"
"github.com/lucas-clemente/quic-go/protocol"
"github.com/lucas-clemente/quic-go/qerr"
@ -59,6 +61,7 @@ var _ = Describe("Client", func() {
})
It("properly closes the client", func(done Done) {
numGoRoutines := runtime.NumGoroutine()
startUDPConn()
var stoppedListening bool
go func() {
@ -67,11 +70,13 @@ var _ = Describe("Client", func() {
stoppedListening = true
}()
err := client.Close()
testErr := errors.New("test error")
err := client.Close(testErr)
Expect(err).ToNot(HaveOccurred())
Eventually(session.closed).Should(BeTrue())
Expect(session.closeReason).To(BeNil())
Expect(session.closeReason).To(MatchError(testErr))
Eventually(func() bool { return stoppedListening }).Should(BeTrue())
Eventually(runtime.NumGoroutine()).Should(Equal(numGoRoutines))
close(done)
})
@ -85,7 +90,7 @@ var _ = Describe("Client", func() {
Expect(client.session.(*Session).connectionID).To(Equal(client.connectionID))
Expect(client.session.(*Session).version).To(Equal(client.version))
err = client.Close()
err = client.Close(nil)
Expect(err).ToNot(HaveOccurred())
})
@ -130,7 +135,7 @@ var _ = Describe("Client", func() {
Expect(session.closed).To(BeFalse())
Eventually(func() bool { return stoppedListening }).Should(BeFalse())
err = client.Close()
err = client.Close(nil)
Expect(err).ToNot(HaveOccurred())
close(done)
})
@ -202,7 +207,7 @@ var _ = Describe("Client", func() {
// it didn't pass the version negoation packet to the session (since it has no payload)
Expect(session.packetCount).To(BeZero())
err = client.Close()
err = client.Close(nil)
Expect(err).ToNot(HaveOccurred())
})