Fix test data race

olahol · Mar 6, 2024 · 87e736d · 87e736d
1 parent f3ccd79
commit 87e736d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/melody_test.go b/melody_test.go
@@ -10,6 +10,7 @@ import (
 	"strconv"
 	"strings"
 	"sync"
+	"sync/atomic"
 	"testing"
 	"testing/quick"
 	"time"
@@ -790,9 +791,9 @@ func TestConcurrentMessageHandling(t *testing.T) {
 		ws.m.Config.ConcurrentMessageHandling = cmh
 		ws.m.Config.PongWait = base
 
-		var errorSet bool
+		var errorSet atomic.Bool
 		ws.m.HandleError(func(s *Session, err error) {
-			errorSet = true
+			errorSet.Store(true)
 			done <- struct{}{}
 		})
 
@@ -811,7 +812,7 @@ func TestConcurrentMessageHandling(t *testing.T) {
 
 		<-done
 
-		return errorSet
+		return errorSet.Load()
 	}
 
 	t.Run("text should error", func(t *testing.T) {