gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

use a MockPacketHandlerManager in all server tests

Browse Source
This commit is contained in:
Marten Seemann
2020-01-13 14:39:34 +07:00
parent 54b38cac0f
commit cf71bdd05a
1 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
server_test.go
View File

@@ -598,12 +598,17 @@ var _ = Describe("Server", func() {
}) })
Context("server accepting sessions that haven't completed the handshake", func() { Context("server accepting sessions that haven't completed the handshake", func() {
var serv *earlyServer var (
serv *earlyServer
phm *MockPacketHandlerManager
)
BeforeEach(func() { BeforeEach(func() {
ln, err := ListenEarly(conn, tlsConf, nil) ln, err := ListenEarly(conn, tlsConf, nil)
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
serv = ln.(*earlyServer) serv = ln.(*earlyServer)
phm = NewMockPacketHandlerManager(mockCtrl)
serv.sessionHandler = phm
}) })
It("accepts new sessions when they become ready", func() { It("accepts new sessions when they become ready", func() {
@@ -640,6 +645,9 @@ var _ = Describe("Server", func() {
sess.EXPECT().Context().Return(context.Background()) sess.EXPECT().Context().Return(context.Background())
return sess return sess
} }
phm.EXPECT().GetStatelessResetToken(gomock.Any())
phm.EXPECT().AddIfNotTaken(gomock.Any(), sess).Return(true)
phm.EXPECT().Add(gomock.Any(), sess)
serv.createNewSession(&net.UDPAddr{}, nil, nil, nil, nil, protocol.VersionWhatever) serv.createNewSession(&net.UDPAddr{}, nil, nil, nil, nil, protocol.VersionWhatever)
Consistently(done).ShouldNot(BeClosed()) Consistently(done).ShouldNot(BeClosed())
close(ready) close(ready)
@@ -681,6 +689,9 @@ var _ = Describe("Server", func() {
go func() { go func() {
defer GinkgoRecover() defer GinkgoRecover()
defer wg.Done() defer wg.Done()
phm.EXPECT().GetStatelessResetToken(gomock.Any())
phm.EXPECT().AddIfNotTaken(gomock.Any(), gomock.Any()).Return(true)
phm.EXPECT().Add(gomock.Any(), gomock.Any())
serv.handlePacket(getInitialWithRandomDestConnID()) serv.handlePacket(getInitialWithRandomDestConnID())
Consistently(conn.dataWritten).ShouldNot(Receive()) Consistently(conn.dataWritten).ShouldNot(Receive())
}() }()
@@ -730,6 +741,9 @@ var _ = Describe("Server", func() {
return sess return sess
} }
phm.EXPECT().GetStatelessResetToken(gomock.Any())
phm.EXPECT().AddIfNotTaken(gomock.Any(), sess).Return(true)
phm.EXPECT().Add(gomock.Any(), sess)
serv.handlePacket(p) serv.handlePacket(p)
Consistently(conn.dataWritten).ShouldNot(Receive()) Consistently(conn.dataWritten).ShouldNot(Receive())
Eventually(sessionCreated).Should(BeClosed()) Eventually(sessionCreated).Should(BeClosed())
@@ -745,6 +759,7 @@ var _ = Describe("Server", func() {
Consistently(done).ShouldNot(BeClosed()) Consistently(done).ShouldNot(BeClosed())
// make the go routine return // make the go routine return
phm.EXPECT().CloseServer()
sess.EXPECT().getPerspective().MaxTimes(2) // once for every conn ID sess.EXPECT().getPerspective().MaxTimes(2) // once for every conn ID
Expect(serv.Close()).To(Succeed()) Expect(serv.Close()).To(Succeed())
Eventually(done).Should(BeClosed()) Eventually(done).Should(BeClosed())