test(util): fix wait_until_cloud_init exit code 2 (#5058)

Make wait_until_cloud_init aware of the new success-sh exit code 2.
canonical · Jun 5, 2024 · a061e81 · a061e81
1 parent 7e4d293
commit a061e81
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/integration_tests/util.py b/tests/integration_tests/util.py
@@ -304,10 +304,8 @@ def wait_for_cloud_init(client: "IntegrationInstance", num_retries: int = 30):
     for _ in range(num_retries):
         try:
             result = client.execute("cloud-init status")
-            if (
-                result
-                and result.ok
-                and ("running" not in result or "not started" not in result)
+            if result.return_code in (0, 2) and (
+                "running" not in result or "not started" not in result
             ):
                 return result
         except Exception as e: