Fix backups integration test

Signed-off-by: Marcelo Henrique Neppel <[email protected]>
canonical · Jun 11, 2024 · 936b115 · 936b115
1 parent 0e5877c
commit 936b115
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/integration/helpers.py b/tests/integration/helpers.py
@@ -1036,10 +1036,10 @@ async def wait_for_idle_on_blocked(
     unit = ops_test.model.units.get(f"{database_app_name}/{unit_number}")
     await asyncio.gather(
         ops_test.model.wait_for_idle(apps=[other_app_name], status="active"),
-        ops_test.model.wait_for_idle(
-            apps=[database_app_name], status="blocked", raise_on_blocked=False
+        ops_test.model.block_until(
+            lambda: unit.workload_status == "blocked"
+            and unit.workload_status_message == status_message
         ),
-        ops_test.model.block_until(lambda: unit.workload_status_message == status_message),
     )