fix: Test race when accessing session (#355)

kylecarbs · web-flow · commit d6b2afd11e37 · 2021-05-20T20:40:11.000Z
diff --git 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)
