Try to retry and wait for message

Author: dragomirp

tests/integration/spaces/test_spaced_database.py

@@ -7,6 +7,7 @@
 
 import jubilant
 import pytest
+from tenacity import Retrying, stop_after_attempt, wait_fixed
 
 PG_NAME = "postgresql"
 APP_NAME = "postgresql-test-app"
@@ -88,8 +89,11 @@ def test_integrate_with_isolated_space(juju: jubilant.Juju):
     juju.integrate(PG_NAME, f"{ISOLATED_APP_NAME}:database")
     sleep(SLEEP_TIME)
     # Wait for the relation to be established
+    msg = "received database credentials of the first database"
     juju.wait(
-        lambda status: jubilant.all_active(status, PG_NAME, ISOLATED_APP_NAME), delay=SLEEP_TIME
+        lambda status: jubilant.all_active(status, PG_NAME, ISOLATED_APP_NAME)
+        and status.apps[ISOLATED_APP_NAME].app_status.message == msg,
+        delay=SLEEP_TIME,
     )
 
     status = juju.status()
@@ -98,5 +102,6 @@ def test_integrate_with_isolated_space(juju: jubilant.Juju):
     logger.info("Flush default routes on client")
     juju.exec("sudo ip route flush default", unit=unit)
 
-    with pytest.raises(jubilant.TaskError):
-        juju.run(unit, "start-continuous-writes")
+    for attempt in Retrying(stop=stop_after_attempt(3), wait=wait_fixed(5)):
+        with attempt, pytest.raises(jubilant.TaskError):
+            juju.run(unit, "start-continuous-writes")