From 4616f06996739bfd0aa97e407e61c343f5f0bfe7 Mon Sep 17 00:00:00 2001 From: Marten Seemann Date: Wed, 6 Jun 2018 12:01:16 +0200 Subject: [PATCH] fix flaky session test --- session_test.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/session_test.go b/session_test.go index 38aec5261..f422001f1 100644 --- a/session_test.go +++ b/session_test.go @@ -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()) })