diff options
Diffstat (limited to 'tests/integration_tests/instances.py')
| -rw-r--r-- | tests/integration_tests/instances.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tests/integration_tests/instances.py b/tests/integration_tests/instances.py index c68ee753..f8f98e42 100644 --- a/tests/integration_tests/instances.py +++ b/tests/integration_tests/instances.py @@ -35,7 +35,7 @@ class IntegrationInstance: def destroy(self): self.instance.delete() - def restart(self): + def restart(self, raise_on_cloudinit_failure=False): """Restart this instance (via cloud mechanism) and wait for boot. This wraps pycloudlib's `BaseInstance.restart` to pass @@ -44,7 +44,9 @@ class IntegrationInstance: """ self.instance.restart(wait=False) log.info("Instance restarted; waiting for boot") - self.instance.wait(raise_on_cloudinit_failure=False) + self.instance.wait( + raise_on_cloudinit_failure=raise_on_cloudinit_failure + ) def execute(self, command, *, use_sudo=True) -> Result: if self.instance.username == 'root' and use_sudo is False: |