Wait for exact number of units after scale down

Signed-off-by: Marcelo Henrique Neppel <[email protected]>
canonical · Aug 2, 2024 · d9607e6 · d9607e6
1 parent a870470
commit d9607e6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/integration/ha_tests/test_async_replication.py b/tests/integration/ha_tests/test_async_replication.py
@@ -476,7 +476,11 @@ async def test_async_replication_failover_in_main_cluster(
     async with ops_test.fast_forward(FAST_INTERVAL), fast_forward(second_model, FAST_INTERVAL):
         await gather(
             first_model.wait_for_idle(
-                apps=[DATABASE_APP_NAME], status="active", idle_period=IDLE_PERIOD, timeout=TIMEOUT
+                apps=[DATABASE_APP_NAME],
+                status="active",
+                idle_period=IDLE_PERIOD,
+                timeout=TIMEOUT,
+                wait_for_exact_units=(CLUSTER_SIZE - 1),
             ),
             second_model.wait_for_idle(
                 apps=[DATABASE_APP_NAME], status="active", idle_period=IDLE_PERIOD, timeout=TIMEOUT