actually fix tests (maybe)

Signed-off-by: Zen <[email protected]>

Author: desultory

src/ugrd/base/test.py

@@ -26,8 +26,30 @@ def init_test_vars(self):
         self['test_flag'] = uuid4()
 
 
+def _get_qemu_cmd_args(self, test_image):
+    """ Gets the qemu command from the configuration """
+    test_initrd = self._archive_out_path
+    test_rootfs = test_image['_archive_out_path']
+    qemu_args = {'-m': self['test_memory'],
+                 '-cpu': self['test_cpu'],
+                 '-kernel': self['test_kernel'],
+                 '-initrd': test_initrd,
+                 '-serial': 'mon:stdio',
+                 '-append': self['test_cmdline'],
+                 '-drive': 'file=%s,format=raw' % test_rootfs}
+
+    qemu_bools = [f'-{item}' for item in self['qemu_bool_args']]
+
+    arglist = [f"qemu-system-{self['test_arch']}"] + qemu_bools
+    for key, value in qemu_args.items():
+        arglist.append(key)
+        arglist.append(value)
+
+    return arglist
+
+
 def make_test_image(self):
-    """ Creates a new initramfs generator to creaate the test image """
+    """ Creates a new initramfs generator to create the test image """
     from ugrd.initramfs_generator import InitramfsGenerator
 
     kwargs = {'logger': self.logger,
@@ -46,37 +68,17 @@ def make_test_image(self):
         self.logger.error("Test image configuration:\n%s", target_fs)
         raise RuntimeError("Failed to build test rootfs: %s" % e)
 
-    self['_test_rootfs'] = target_fs
-
-
-def _get_qemu_cmd_args(self):
-    """ Gets the qemu command from the configuration """
-    test_initrd = self._archive_out_path
-    test_rootfs = self['_test_rootfs']['_archive_out_path']
-    qemu_args = {'-m': self['test_memory'],
-                 '-cpu': self['test_cpu'],
-                 '-kernel': self['test_kernel'],
-                 '-initrd': test_initrd,
-                 '-serial': 'mon:stdio',
-                 '-append': self['test_cmdline'],
-                 '-drive': 'file=%s,format=raw' % test_rootfs}
-
-    qemu_bools = [f'-{item}' for item in self['qemu_bool_args']]
-
-    arglist = [f"qemu-system-{self['test_arch']}"] + qemu_bools
-    for key, value in qemu_args.items():
-        arglist.append(key)
-        arglist.append(value)
-
-    return arglist
+    return target_fs
 
 
 def test_image(self):
-    qemu_cmd = _get_qemu_cmd_args(self)
+    image = make_test_image(self)
+    qemu_cmd = _get_qemu_cmd_args(self, image)
+
     self.logger.info("Testing initramfs image: %s", self['_archive_out_path'])
-    self.logger.debug("Test config:\n%s", self._test_rootfs)
+    self.logger.debug("Test config:\n%s", image)
     self.logger.info("Test kernel: %s", self['test_kernel'])
-    self.logger.info("Test rootfs: %s", self['_test_rootfs']['_archive_out_path'])
+    self.logger.info("Test rootfs: %s", image['_archive_out_path'])
     self.logger.info("Test flag: %s", self['test_flag'])
     self.logger.info("QEMU command: %s", ' '.join([str(arg) for arg in qemu_cmd]))
 

src/ugrd/base/test.toml

@@ -12,9 +12,6 @@ qemu_bool_args = ['nographic', 'no-reboot', 'enable-kvm']
 [imports.build_pre]
 "ugrd.base.test" = [ "init_test_vars" ]
 
-[imports.build_tasks]
-"ugrd.base.test" = [ "make_test_image" ]
-
 [imports.tests]
 "ugrd.base.test" = [ "test_image" ]
 
@@ -30,5 +27,4 @@ test_flag = "str"  # Define the success flag for the test
 test_rootfs_name = "str"  # Define the name of the rootfs image
 test_rootfs_build_dir = "Path"  # Define the build directory for the rootfs image
 qemu_bool_args = "NoDupFlatList"  # Define the qemu boolean arguments
-_test_rootfs = "InitramfsGenerator"  # Define the target rootfs generator
 _qemu_cmd = "str"  # Define the qemu command to use for the test