ssh connection retry on restarts (#3001)

Azure · Dec 27, 2023 · 847bb08 · 847bb08
1 parent e74677b
commit 847bb08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests_e2e/tests/lib/virtual_machine_client.py b/tests_e2e/tests/lib/virtual_machine_client.py
@@ -174,7 +174,7 @@ def restart(
                         return
                     log.info("The VM has not rebooted yet. Restart time: %s. Boot time: %s", before_restart, boot_time)
                 except CommandError as e:
-                    if (e.exit_code == 255 and "Connection refused" in str(e)) or "Unprivileged users are not permitted to log in yet" in str(e):
+                    if (e.exit_code == 255 and ("Connection refused" in str(e) or "Connection timed out" in str(e))) or "Unprivileged users are not permitted to log in yet" in str(e):
                         log.info("VM %s is not yet accepting SSH connections", self)
                     else:
                         raise