Fix test_create_linode_with_kernel_and_boot_size_then_add_disk_and_rebuild

Author: psnoch-akamai

tests/integration/linodes/helpers.py

@@ -312,19 +312,25 @@ def wait_for_disk_status(
     must_end = time.time() + timeout
     while time.time() < must_end:
         time.sleep(period)
-        result = exec_test_command(
-            [
-                "linode-cli",
-                "linodes",
-                "disk-view",
-                linode_id,
-                disk_id,
-                "--format",
-                "status",
-                "--text",
-                "--no-headers",
-            ]
-        )
+        try:
+            result = exec_test_command(
+                [
+                    "linode-cli",
+                    "linodes",
+                    "disk-view",
+                    linode_id,
+                    disk_id,
+                    "--format",
+                    "status",
+                    "--text",
+                    "--no-headers",
+                ]
+            )
+        except RuntimeError as response_error:
+            if "Not found" in str(response_error):
+                continue
+            else:
+                raise RuntimeError(response_error)
         if status == result:
             return True
     return False

tests/integration/linodes/test_linodes.py

@@ -121,7 +121,7 @@ def test_create_linode_with_kernel_and_boot_size_then_add_disk_and_rebuild(
     assert wait_for_disk_status(
         linode_id=result_create[0],
         disk_id=response_create_disk[0],
-        timeout=180,
+        timeout=90,
         status="ready",
     ), "linode failed to change disk status to ready after disk creation.."