fixup! Add version guards and http.Status constants to shard routing tests

Signed-off-by: Sean Chittenden <sean.chittenden@crowdstrike.com>

Author: sean-

opensearchtransport/integration_test_helpers_test.go

@@ -52,7 +52,7 @@ func waitForAllNodesActive(t *testing.T, transport *Client, ctx context.Context,
 			return false
 		}
 
-		conns := extractActiveConnsFromPolicy(routerPolicy)
+		conns := collectActiveConns(routerPolicy)
 		if len(conns) < expectedNodes {
 			return false
 		}
@@ -66,6 +66,23 @@ func waitForAllNodesActive(t *testing.T, transport *Client, ctx context.Context,
 		"expected %d active nodes with names, got fewer after repeated DiscoverNodes", expectedNodes)
 }
 
+// collectActiveConns walks the policy tree depth-first and returns the first
+// non-empty set of active connections found. Unlike extractActiveConnsFromPolicy,
+// this handles poolRouter by recursing into its inner policy via policyTreeWalker.
+func collectActiveConns(p Policy) []*Connection {
+	if conns := extractActiveConnsFromPolicy(p); len(conns) > 0 {
+		return conns
+	}
+	if walker, ok := p.(policyTreeWalker); ok {
+		for _, child := range walker.childPolicies() {
+			if conns := collectActiveConns(child); len(conns) > 0 {
+				return conns
+			}
+		}
+	}
+	return nil
+}
+
 // clusterNodeCount queries /_nodes via the transport and returns the total
 // number of nodes reported by the cluster. The transport must already be
 // warmed up (able to serve requests) before calling this.