Unmute SearchReplicaSelectionIT#testNodeSelection test (#132567)

Author: drempapis

muted-tests.yml

@@ -503,9 +503,6 @@ tests:
 - class: org.elasticsearch.xpack.esql.action.CrossClusterAsyncQueryIT
   method: testBadAsyncId
   issue: https://github.com/elastic/elasticsearch/issues/132353
-- class: org.elasticsearch.search.routing.SearchReplicaSelectionIT
-  method: testNodeSelection
-  issue: https://github.com/elastic/elasticsearch/issues/132354
 - class: org.elasticsearch.xpack.logsdb.qa.LogsDbVersusReindexedLogsDbChallengeRestIT
   method: testRandomQueries
   issue: https://github.com/elastic/elasticsearch/issues/132376

server/src/internalClusterTest/java/org/elasticsearch/search/routing/SearchReplicaSelectionIT.java

@@ -39,7 +39,7 @@ public Settings nodeSettings(int nodeOrdinal, Settings otherSettings) {
             .build();
     }
 
-    public void testNodeSelection() {
+    public void testNodeSelection() throws Exception {
         // We grab a client directly to avoid using a randomizing client that might set a search preference.
         Client client = internalCluster().coordOnlyNodeClient();
 
@@ -76,14 +76,16 @@ public void testNodeSelection() {
         assertNotNull(nodeStats);
         assertEquals(3, nodeStats.getAdaptiveSelectionStats().getComputedStats().size());
 
-        assertResponse(client.prepareSearch().setQuery(matchAllQuery()), response -> {
-            String selectedNodeId = response.getHits().getAt(0).getShard().getNodeId();
-            double selectedRank = nodeStats.getAdaptiveSelectionStats().getRanks().get(selectedNodeId);
+        assertBusy(() -> {
+            assertResponse(client.prepareSearch().setQuery(matchAllQuery()), response -> {
+                String selectedNodeId = response.getHits().getAt(0).getShard().getNodeId();
+                double selectedRank = nodeStats.getAdaptiveSelectionStats().getRanks().get(selectedNodeId);
 
-            for (Map.Entry<String, Double> entry : nodeStats.getAdaptiveSelectionStats().getRanks().entrySet()) {
-                double rank = entry.getValue();
-                assertThat(rank, greaterThanOrEqualTo(selectedRank));
-            }
+                for (Map.Entry<String, Double> entry : nodeStats.getAdaptiveSelectionStats().getRanks().entrySet()) {
+                    double rank = entry.getValue();
+                    assertThat(rank, greaterThanOrEqualTo(selectedRank));
+                }
+            });
         });
     }
 }