Fix race condition found in TestRejoinOnReconnect

gempir · Sep 12, 2021 · a45147b · a45147b
1 parent cb9ca8d
commit a45147b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/client_test.go b/client_test.go
@@ -1993,7 +1993,7 @@ func TestRejoinOnReconnect(t *testing.T) {
 
 	client.Join("gempiR")
 
-	go client.Connect()
+	clientDisconnected := connectAndEnsureGoodDisconnect(t, client)
 
 	// wait for server to receive message
 	select {
@@ -2010,10 +2010,13 @@ func TestRejoinOnReconnect(t *testing.T) {
 	// Manually disconnect
 	client.Disconnect()
 
+	<-clientDisconnected
+
+	waitEnd = make(chan struct{})
+
 	// Manually reconnect
 	go client.Connect()
 
-	waitEnd = make(chan struct{})
 	select {
 	case <-waitEnd:
 	case <-time.After(time.Second * 3):