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

fix flaky server unit test

Browse source
This commit is contained in:
Marten Seemann 2019-04-03 01:27:31 +09:00
parent 907071221c
commit 7b569ba071
1 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
server_test.go
View file

@ -507,7 +507,8 @@ var _ = Describe("Server", func() {
It("never blocks when calling the onHandshakeComplete callback", func() {
const num = 50
done := make(chan struct{}, num)
runs := make(chan struct{}, num)
contexts := make(chan struct{}, num)
serv.newSession = func(
_ connection,
runner sessionRunner,
@ -521,10 +522,9 @@ var _ = Describe("Server", func() {
_ protocol.VersionNumber,
) (quicSession, error) {
sess := NewMockQuicSession(mockCtrl)
sess.EXPECT().run().Do(func() {})
sess.EXPECT().Context().Return(context.Background())
sess.EXPECT().run().Do(func() { runs <- struct{}{} })
sess.EXPECT().Context().Do(func() { contexts <- struct{}{} }).Return(context.Background())
runner.OnHandshakeComplete(sess)
done <- struct{}{}
return sess, nil
}
@ -534,7 +534,8 @@ var _ = Describe("Server", func() {
Expect(err).ToNot(HaveOccurred())
}
}()
Eventually(done).Should(HaveLen(num))
Eventually(runs).Should(HaveLen(num))
Eventually(contexts).Should(HaveLen(num))
})
})
})