Prefer refreshAndWaitForReplication in case of seg rep.

finnegancarroll · finnegancarroll · commit eabfc3632f10 · 2025-07-29T15:55:50.000-07:00
Signed-off-by: Finn Carroll &lt;carrofin@amazon.com&gt;
diff --git a/server/src/internalClusterTest/java/org/opensearch/cluster/routing/WeightedRoutingIT.java b/server/src/internalClusterTest/java/org/opensearch/cluster/routing/WeightedRoutingIT.java
@@ -898,7 +898,7 @@ public void testCustomPreferenceShardIdCombination() {
         for (int i = 0; i < 30; i++) {
             client().prepareIndex("test_index").setId(String.valueOf(i)).setSource("field", "value" + i).get();
         }
-        refresh("test_index");
+        refreshAndWaitForReplication("test_index");
 
         /*
         Execute the same match all query with custom string preference.

Original file line number	Diff line number	Diff line change
`@@ -898,7 +898,7 @@ public void testCustomPreferenceShardIdCombination() {`
`898`	`898`	`for (int i = 0; i < 30; i++) {`
`899`	`899`	`client().prepareIndex("test_index").setId(String.valueOf(i)).setSource("field", "value" + i).get();`
`900`	`900`	`}`
`901`		`- refresh("test_index");`
	`901`	`+ refreshAndWaitForReplication("test_index");`
`902`	`902`
`903`	`903`	`/*`
`904`	`904`	`Execute the same match all query with custom string preference.`