add some retry logic with a delay to check the VM's state

virt-s1 · Sep 14, 2024 · 7f2f72c · 7f2f72c
1 parent aa00e17
commit 7f2f72c
Showing 1 changed file with 5 additions and 13 deletions.
diff --git a/os_tests/tests/test_lifecycle.py b/os_tests/tests/test_lifecycle.py
@@ -1003,19 +1003,11 @@ def test_reboot_simultaneous(self):
     def _start_vm_and_check(self):
         self.vm.start(wait=True)
         time.sleep(60)
-        for _ in range(10):
-            try:
-                utils_lib.init_connection(self, timeout=self.ssh_timeout)
-                output = utils_lib.run_cmd(self, 'whoami').strip()
-                self.assertEqual(self.vm.vm_username,
-                                 output,
-                                 "Start VM error: output of cmd `who` unexpected -> %s" % output)
-                break
-            except Exception as e:
-                self.log.error("SSH connection failed, retrying... Error: {}".format(e))
-                time.sleep(60)
-        else:
-            self.fail("Failed to establish SSH connection after VM restart.")
+        utils_lib.init_connection(self, timeout=self.ssh_timeout*10)
+        output = utils_lib.run_cmd(self, 'whoami').strip()
+        self.assertEqual(self.vm.vm_username,
+            output,
+            "Start VM error: output of cmd `who` unexpected -> %s" % output)
 
     def test_stop_start_vm(self):
         """