move compression tests to another test module

black format

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

Author: desultory

tests/test_compression.py

@@ -0,0 +1,25 @@
+from unittest import TestCase, main
+
+from pycpio.errors import UnavailableCompression
+from ugrd.initramfs_generator import InitramfsGenerator
+from zenlib.logging import loggify
+
+
+@loggify
+class TestCompression(TestCase):
+    def test_xz(self):
+        """Text XZ compression for initramfs."""
+        generator = InitramfsGenerator(logger=self.logger, config="tests/fullauto.toml", cpio_compression="xz")
+        generator.build()
+
+    def test_zstd(self):
+        """Test ZSTD compression for initramfs."""
+        generator = InitramfsGenerator(logger=self.logger, config="tests/fullauto.toml", cpio_compression="zstd")
+        try:
+            generator.build()
+        except UnavailableCompression as e:
+            self.skipTest(f"ZSTD compression is not available: {e}")
+
+
+if __name__ == "__main__":
+    main()

tests/test_filesystems.py

@@ -1,5 +1,6 @@
 from unittest import TestCase, main
 
+from ugrd.exceptions import AutodetectError
 from ugrd.initramfs_generator import InitramfsGenerator
 from ugrd.kmod import MissingModuleError
 from zenlib.logging import loggify
@@ -8,37 +9,38 @@
 @loggify
 class TestFilesystems(TestCase):
     def test_ext4(self):
-        """ Test ext4 root filesystem functionality. """
+        """Test ext4 root filesystem functionality."""
         generator = InitramfsGenerator(logger=self.logger, config="tests/fs/ext4.toml")
         generator.build()
 
     def test_btrfs(self):
-        """ Test btrfs root filesystem functionality. """
+        """Test btrfs root filesystem functionality."""
         generator = InitramfsGenerator(logger=self.logger, config="tests/fs/btrfs.toml")
         generator.build()
 
     def test_xfs(self):
-        """ Test xfs root filesystem functionality. """
+        """Test xfs root filesystem functionality."""
         generator = InitramfsGenerator(logger=self.logger, config="tests/fs/xfs.toml")
         generator.build()
 
     def test_f2fs(self):
-        """ Test f2fs root filesystem functionality. """
+        """Test f2fs root filesystem functionality."""
         generator = InitramfsGenerator(logger=self.logger, config="tests/fs/f2fs.toml")
         try:
             generator.build()
-        except MissingModuleError:
-            generator.logger.critical("F2FS is not supported on this system, skipping test.")
+        except (MissingModuleError, AutodetectError) as e:
+            self.skipTest(f"F2FS is not supported on this system: {e}")
 
     def test_overlayfs(self):
-        """ Test overlayfs/tmpfs overlay over root creation. """
+        """Test overlayfs/tmpfs overlay over root creation."""
         generator = InitramfsGenerator(logger=self.logger, config="tests/fs/overlayfs.toml")
         generator.build()
 
     def test_squashfs(self):
-        """ Test squashfs/overlayfs/tmpfs for live cd systems. """
+        """Test squashfs/overlayfs/tmpfs for live cd systems."""
         generator = InitramfsGenerator(logger=self.logger, config="tests/fs/squashfs.toml")
         generator.build()
 
+
 if __name__ == "__main__":
     main()

tests/test_ugrd.py

@@ -1,7 +1,6 @@
 from unittest import TestCase, main
 
 from ugrd.initramfs_generator import InitramfsGenerator
-from pycpio.errors import UnavailableCompression
 from zenlib.logging import loggify
 
 
@@ -12,19 +11,6 @@ def test_fullauto(self):
         generator = InitramfsGenerator(logger=self.logger, config="tests/fullauto.toml")
         generator.build()
 
-    def test_xz(self):
-        """Text XZ compression for initramfs."""
-        generator = InitramfsGenerator(logger=self.logger, config="tests/fullauto.toml", cpio_compression="xz")
-        generator.build()
-
-    def test_zstd(self):
-        """Test ZSTD compression for initramfs."""
-        generator = InitramfsGenerator(logger=self.logger, config="tests/fullauto.toml", cpio_compression="zstd")
-        try:
-            generator.build()
-        except UnavailableCompression as e:
-            self.skipTest(f"ZSTD compression is not available: {e}")
-
     def test_bad_config(self):
         """Test that a bad config file which should raise an error."""
         with self.assertRaises(ValueError):