uitests: top off test coverage

Signed-off-by: Cole Robinson <crobinso@redhat.com>

2 files changed, 14 insertions, 3 deletions

diff --git a/virtManager/lib/statsmanager.py b/virtManager/lib/statsmanager.py
index d19129f3..28495495 100644
--- a/virtManager/lib/statsmanager.py
+++ b/virtManager/lib/statsmanager.py
@@ -128,7 +128,6 @@ class vmmStatsManager(vmmGObject):
         self._disk_stats_lxc_supported = True
         self._mem_stats_supported = True
 
-
     def _cleanup(self):
         for statslist in self._vm_stats.values():
             statslist.cleanup()
@@ -390,7 +389,9 @@ class vmmStatsManager(vmmGObject):
     ####################
 
     def _get_all_stats(self, conn):
-        if not self._all_stats_supported:
+        # test conn supports allstats as of 2021, but for test coverage
+        # purposes lets still use the old stats code for the test driver
+        if not self._all_stats_supported or conn.is_test():
             return {}
 
         statflags = 0
diff --git a/virtManager/lib/testmock.py b/virtManager/lib/testmock.py
index 3234fcae..1f3e91ac 100644
--- a/virtManager/lib/testmock.py
+++ b/virtManager/lib/testmock.py
@@ -107,9 +107,19 @@ def fake_openauth(conn, cb, data):
 
 
 class fakeVirtBootstrap:
+    @staticmethod
     def bootstrap(**kwargs):
         import time
-        time.sleep(1)
+        import logging
+        log = logging.getLogger("virtBootstrap")
+        log.info("mock virtBootstrap msg1")
+        kwargs["progress_cb"]({"status": "msg1"})
+        time.sleep(.5)
+        log.info("mock virtBootstrap msg2")
+        kwargs["progress_cb"]({"status": "msg2"})
+        time.sleep(.5)
+        log.info("mock virtBootstrap msg3")
+        kwargs["progress_cb"]({"status": "msg3"})
         if "username" in kwargs:
             raise RuntimeError("fakeVirtBootstrap mock auth failure!")