solove data race

pkg/network/netpoll/transport_test.go

@@ -22,6 +22,7 @@ import (
 	"context"
 	"net"
 	"reflect"
+	"sync"
 	"sync/atomic"
 	"syscall"
 	"testing"
@@ -73,12 +74,15 @@ func TestTransport(t *testing.T) {
 	t.Run("TestSenseClientDisconnection", func(t *testing.T) {
 		var onConnFlag int32
 		var ctxVal context.Context
+		var mutex sync.Mutex
 		transporter := NewTransporter(&config.Options{
 			Addr:                     addr,
 			Network:                  nw,
 			SenseClientDisconnection: true,
 			OnConnect: func(ctx context.Context, conn network.Conn) context.Context {
 				atomic.StoreInt32(&onConnFlag, 1)
+				mutex.Lock()
+				defer mutex.Unlock()
 				ctxVal = ctx
 				return ctx
 			},
@@ -97,6 +101,8 @@ func TestTransport(t *testing.T) {
 		time.Sleep(100 * time.Millisecond)
 
 		assert.Assert(t, atomic.LoadInt32(&onConnFlag) == 1)
+		mutex.Lock()
+		defer mutex.Unlock()
 		assert.DeepEqual(t, "*context.cancelCtx", reflect.TypeOf(ctxVal).String())
 	})