race in test

Author: ndyakov

hitless/pool_hook_test.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"net"
 	"sync"
+	"sync/atomic"
 	"testing"
 	"time"
 
@@ -919,7 +920,7 @@ func TestConnectionHook(t *testing.T) {
 		}
 
 		// Set a dialer that will check the context timeout
-		timeoutVerified := false
+		var timeoutVerified int32 // Use atomic for thread safety
 		conn.SetInitConnFunc(func(ctx context.Context, cn *pool.Conn) error {
 			// Check that the context has the expected timeout
 			deadline, ok := ctx.Deadline()
@@ -935,7 +936,7 @@ func TestConnectionHook(t *testing.T) {
 				t.Errorf("Context deadline not as expected. Expected around %v, got %v (diff: %v)",
 					expectedDeadline, deadline, timeDiff)
 			} else {
-				timeoutVerified = true
+				atomic.StoreInt32(&timeoutVerified, 1)
 			}
 
 			return nil // Successful handoff
@@ -955,7 +956,7 @@ func TestConnectionHook(t *testing.T) {
 		// Wait for handoff to complete
 		time.Sleep(500 * time.Millisecond)
 
-		if !timeoutVerified {
+		if atomic.LoadInt32(&timeoutVerified) == 0 {
 			t.Error("HandoffTimeout was not properly applied to context")
 		}