fix flaky session test

This commit is contained in:
Marten Seemann
2018-06-06 12:01:16 +02:00
parent e53b611961
commit 4616f06996

View File

@@ -640,8 +640,8 @@ var _ = Describe("Session", func() {
Expect(err).To(MatchError(testErr))
close(done)
}()
sess.handlePacket(&receivedPacket{header: hdr})
sessionRunner.EXPECT().removeConnectionID(gomock.Any())
sess.handlePacket(&receivedPacket{header: hdr})
Eventually(done).Should(BeClosed())
})