vmupdate: wait only for PCI-assigned VM shutdown

Author: Jayant-kernel

vmupdate/qube_connection.py

@@ -96,10 +96,19 @@ def __exit__(self, exc_type, exc_val, exc_tb):
         if self.qube.is_running() and not self._initially_running:
             self.logger.info('Shutdown %s', self.qube.name)
             self.qube.shutdown()
-            asyncio.run(wait_for_domain_shutdown([self.qube]))
+            if self._has_assigned_pci_devices(self.qube):
+                asyncio.run(wait_for_domain_shutdown([self.qube]))
 
         self.__connected = False
 
+    @staticmethod
+    def _has_assigned_pci_devices(vm) -> bool:
+        """Return True when VM has assigned PCI devices."""
+        try:
+            return any(vm.devices['pci'].get_assigned_devices())
+        except Exception:  # best effort check only
+            return False
+
     def transfer_agent(self, src_dir: str) -> ProcessResult:
         """
         Copy a directory content to the workdir in the qube.

vmupdate/tests/test_qube_connection.py

@@ -20,6 +20,8 @@ def test_wait_for_shutdown_when_vm_started_by_update(_arun, wait_shutdown):
     vm = Mock()
     vm.name = "hvm1"
     vm.is_running.side_effect = [False, True]
+    vm.devices = {'pci': Mock()}
+    vm.devices['pci'].get_assigned_devices.return_value = ["00_1f.2"]
     status_notifier = Mock()
     logger = Mock()
 
@@ -35,10 +37,33 @@ def test_wait_for_shutdown_when_vm_started_by_update(_arun, wait_shutdown):
 
 @patch("vmupdate.qube_connection.wait_for_domain_shutdown")
 @patch("vmupdate.qube_connection.asyncio.run")
-def test_do_not_shutdown_if_vm_was_already_running(_arun, wait_shutdown):
+def test_do_not_wait_for_shutdown_without_assigned_pci(_arun, wait_shutdown):
     vm = Mock()
     vm.name = "hvm2"
+    vm.is_running.side_effect = [False, True]
+    vm.devices = {'pci': Mock()}
+    vm.devices['pci'].get_assigned_devices.return_value = []
+    status_notifier = Mock()
+    logger = Mock()
+
+    with QubeConnection(
+            vm, "/tmp/qubes-update", cleanup=False, logger=logger,
+            show_progress=False, status_notifier=status_notifier):
+        pass
+
+    vm.shutdown.assert_called_once_with()
+    wait_shutdown.assert_not_called()
+    _arun.assert_not_called()
+
+
+@patch("vmupdate.qube_connection.wait_for_domain_shutdown")
+@patch("vmupdate.qube_connection.asyncio.run")
+def test_do_not_shutdown_if_vm_was_already_running(_arun, wait_shutdown):
+    vm = Mock()
+    vm.name = "hvm3"
     vm.is_running.return_value = True
+    vm.devices = {'pci': Mock()}
+    vm.devices['pci'].get_assigned_devices.return_value = ["00_1f.2"]
     status_notifier = Mock()
     logger = Mock()