integration_tests: introduce IntegrationInstance.restart (#708)

This wraps pycloudlib's `BaseInstance.restart` and `BaseInstance.wait`
to pass the same parameters as on launch, to avoid cloud-init failures
on the _reboot_ raising an exception.

1 files changed, 11 insertions, 0 deletions

diff --git a/tests/integration_tests/instances.py b/tests/integration_tests/instances.py
index 8f6573cd..c68ee753 100644
--- a/tests/integration_tests/instances.py
+++ b/tests/integration_tests/instances.py
@@ -35,6 +35,17 @@ class IntegrationInstance:
     def destroy(self):
         self.instance.delete()
 
+    def restart(self):
+        """Restart this instance (via cloud mechanism) and wait for boot.
+
+        This wraps pycloudlib's `BaseInstance.restart` to pass
+        `raise_on_cloudinit_failure=False` to `BaseInstance.wait`, mirroring
+        our launch behaviour.
+        """
+        self.instance.restart(wait=False)
+        log.info("Instance restarted; waiting for boot")
+        self.instance.wait(raise_on_cloudinit_failure=False)
+
     def execute(self, command, *, use_sudo=True) -> Result:
         if self.instance.username == 'root' and use_sudo is False:
             raise Exception('Root user cannot run unprivileged')