SERVER-54131 zone_changes_range.js test has a race condition in runBalancer()

1 files changed, 8 insertions, 5 deletions

diff --git a/jstests/sharding/libs/zone_changes_util.js b/jstests/sharding/libs/zone_changes_util.js
index 7d28f482cae..b6426aba10e 100644
--- a/jstests/sharding/libs/zone_changes_util.js
+++ b/jstests/sharding/libs/zone_changes_util.js
@@ -68,14 +68,17 @@ function moveZoneToShard(st, zoneName, fromShard, toShard) {
 }
 
 /**
- * Starts the balancer, lets it run for the given number of rounds, then stops the
- * balancer.
+ * Starts the balancer, lets it run for at least the given number of rounds,
+ * then stops the balancer.
  */
-function runBalancer(st, numRounds) {
+function runBalancer(st, minNumRounds) {
     st.startBalancer();
-    for (let i = 0; i < numRounds; i++) {
+
+    // We add 1 to the number of rounds to avoid a race condition
+    // where the first round is a no-op round
+    for (let i = 0; i < minNumRounds + 1; ++i)
         st.awaitBalancerRound();
-    }
+
     st.stopBalancer();
 }