From 170b2c5557bba64ad82adf4596d7429677966ff7 Mon Sep 17 00:00:00 2001
From: Kyle Carberry <kyle@carberry.com>
Date: Thu, 20 May 2021 20:31:11 +0000
Subject: [PATCH] fix: Test race when accessing session

---
 wsnet/wsnet_test.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/wsnet/wsnet_test.go b/wsnet/wsnet_test.go
index 6b7fa6f1..8452015d 100644
--- a/wsnet/wsnet_test.go
+++ b/wsnet/wsnet_test.go
@@ -58,13 +58,13 @@ func createDumbBroker(t *testing.T) (connectAddr string, listenAddr string) {
 			t.Error(err)
 			return
 		}
+		nc := websocket.NetConn(context.Background(), c, websocket.MessageBinary)
+		mut.Lock()
+		defer mut.Unlock()
 		if sess == nil {
 			t.Error("listen not called")
 			return
 		}
-		nc := websocket.NetConn(context.Background(), c, websocket.MessageBinary)
-		mut.Lock()
-		defer mut.Unlock()
 		oc, err := sess.Open()
 		if err != nil {
 			t.Error(err)