fix test

Author: ndyakov

hitless/pool_hook_test.go

@@ -267,7 +267,8 @@ func TestConnectionHook(t *testing.T) {
 			EndpointType:      EndpointTypeAuto,
 			MaxWorkers:        2,
 			HandoffQueueSize:  10,
-			MaxHandoffRetries: 3, // Explicit queue size to avoid 0-size queue
+			MaxHandoffRetries: 3,
+			HandoffTimeout:    1 * time.Second, // Shorter timeout for faster test
 			LogLevel:          2,
 		}
 		processor := NewPoolHook(failingDialer, "tcp", config, nil)
@@ -293,7 +294,8 @@ func TestConnectionHook(t *testing.T) {
 		}
 
 		// Wait for handoff to complete and fail with proper timeout and polling
-		timeout := time.After(2 * time.Second)
+		// Use longer timeout to account for handoff timeout + processing time
+		timeout := time.After(5 * time.Second)
 		ticker := time.NewTicker(10 * time.Millisecond)
 		defer ticker.Stop()
 
@@ -305,14 +307,14 @@ func TestConnectionHook(t *testing.T) {
 			case <-timeout:
 				t.Fatal("Timeout waiting for failed handoff to complete")
 			case <-ticker.C:
-				if _, pending := processor.pending.Load(conn); !pending {
+				if _, pending := processor.pending.Load(conn.GetID()); !pending {
 					handoffCompleted = true
 				}
 			}
 		}
 
 		// Connection should be removed from pending map after failed handoff
-		if _, pending := processor.pending.Load(conn); pending {
+		if _, pending := processor.pending.Load(conn.GetID()); pending {
 			t.Error("Connection should be removed from pending map after failed handoff")
 		}