From aabdb172ffe75daf2ad960c541855f3289fd4dcb Mon Sep 17 00:00:00 2001
From: Marten Seemann <martenseemann@gmail.com>
Date: Mon, 4 May 2020 19:23:48 +0700
Subject: [PATCH] fix race condition in the client session tests

---
 session_test.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/session_test.go b/session_test.go
index edaf044c..26e7acfb 100644
--- a/session_test.go
+++ b/session_test.go
@@ -2126,6 +2126,7 @@ var _ = Describe("Client Session", func() {
 				defer GinkgoRecover()
 				cryptoSetup.EXPECT().RunHandshake().MaxTimes(1)
 				errChan <- sess.run()
+				close(errChan)
 			}()
 		})
 
@@ -2147,6 +2148,7 @@ var _ = Describe("Client Session", func() {
 			expectClose()
 			sess.shutdown()
 			Eventually(sess.Context().Done()).Should(BeClosed())
+			Eventually(errChan).Should(BeClosed())
 		})
 
 		It("uses the preferred_address connection ID", func() {