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

fix go routine leaks in client tests

Browse source
This commit is contained in:
Marten Seemann 2017-08-31 17:55:48 +07:00
parent e7d1d225dc
commit fa2ecefa31
1 changed files with 23 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

27
client_test.go
View file

@ -135,8 +135,15 @@ var _ = Describe("Client", func() {
remoteAddrChan <- conn.RemoteAddr().String()
return sess, nil, nil
}
go DialAddr("localhost:17890", nil, &Config{})
dialed := make(chan struct{})
go func() {
defer GinkgoRecover()
DialAddr("localhost:17890", nil, &Config{HandshakeTimeout: time.Millisecond})
close(dialed)
}()
Eventually(remoteAddrChan).Should(Receive(Equal("127.0.0.1:17890")))
sess.Close(errors.New("peer doesn't reply"))
Eventually(dialed).Should(BeClosed())
close(done)
})
@ -154,8 +161,15 @@ var _ = Describe("Client", func() {
hostnameChan <- h
return sess, nil, nil
}
go DialAddr("localhost:17890", &tls.Config{ServerName: "foobar"}, nil)
dialed := make(chan struct{})
go func() {
defer GinkgoRecover()
DialAddr("localhost:17890", &tls.Config{ServerName: "foobar"}, nil)
close(dialed)
}()
Eventually(hostnameChan).Should(Receive(Equal("foobar")))
sess.Close(errors.New("peer doesn't reply"))
Eventually(dialed).Should(BeClosed())
close(done)
})
@ -163,6 +177,7 @@ var _ = Describe("Client", func() {
testErr := errors.New("early handshake error")
var dialErr error
go func() {
defer GinkgoRecover()
_, dialErr = Dial(packetConn, addr, "quic.clemente.io:1337", nil, config)
Expect(dialErr).To(MatchError(testErr))
close(done)
@ -338,15 +353,19 @@ var _ = Describe("Client", func() {
close(c)
return sess, nil, nil
}
dialed := make(chan struct{})
go func() {
_, err := Dial(packetConn, addr, "quic.clemente.io:1337", nil, config)
Expect(err).ToNot(HaveOccurred())
defer GinkgoRecover()
Dial(packetConn, addr, "quic.clemente.io:1337", nil, config)
close(dialed)
}()
Eventually(c).Should(BeClosed())
Expect(cconn.(*conn).pconn).To(Equal(packetConn))
Expect(hostname).To(Equal("quic.clemente.io"))
Expect(version).To(Equal(cl.version))
Expect(conf.Versions).To(Equal(config.Versions))
sess.Close(errors.New("peer doesn't reply"))
Eventually(dialed).Should(BeClosed())
close(done)
})