speed up removing application by scaling down

canonical · Nov 1, 2023 · 53fa13b · 53fa13b
1 parent 3ff082e
commit 53fa13b
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/integration/sharding_tests/test_sharding.py b/tests/integration/sharding_tests/test_sharding.py
@@ -304,6 +304,16 @@ async def test_unconventual_shard_removal(ops_test: OpsTest):
         f"{CONFIG_SERVER_APP_NAME}:{CONFIG_SERVER_REL_NAME}",
     )
 
+    await ops_test.model.applications[SHARD_TWO_APP_NAME].destroy_units(f"{SHARD_TWO_APP_NAME}/0")
+    async with ops_test.fast_forward():
+        await ops_test.model.wait_for_idle(
+            apps=[SHARD_TWO_APP_NAME],
+            idle_period=20,
+            status="active",
+            timeout=TIMEOUT,
+            raise_on_error=False,
+        )
+
     await ops_test.model.remove_application(SHARD_TWO_APP_NAME, block_until_done=True)
 
     async with ops_test.fast_forward():