From 828276084757a1f1bd24659e1712dfe4dcb8d159 Mon Sep 17 00:00:00 2001
From: Marten Seemann <martenseemann@gmail.com>
Date: Fri, 24 Feb 2017 10:14:19 +0700
Subject: [PATCH] fix concurrent map access in server test

---
 server_test.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/server_test.go b/server_test.go
index c7dee0f05..da2f114fc 100644
--- a/server_test.go
+++ b/server_test.go
@@ -182,7 +182,12 @@ var _ = Describe("Server", func() {
 			Expect(serv.sessions).To(HaveLen(1))
 			serv.closeCallback(connID)
 			Expect(serv.sessions).To(HaveKey(connID))
-			Eventually(func() map[protocol.ConnectionID]packetHandler { return serv.sessions }).ShouldNot(HaveKey(connID))
+			Eventually(func() bool {
+				serv.sessionsMutex.Lock()
+				_, ok := serv.sessions[connID]
+				serv.sessionsMutex.Unlock()
+				return ok
+			}).Should(BeFalse())
 		})
 
 		It("closes sessions and the connection when Close is called", func() {