tests: Add a libparted test for ped_partition_set_system on msdos

Test the libparted API to make sure the flag is not cleared by calling
ped_partition_set_system.

1 files changed, 35 insertions, 0 deletions

diff --git a/libparted/tests/flags.c b/libparted/tests/flags.c
index c83a361..c4b290b 100644
--- a/libparted/tests/flags.c
+++ b/libparted/tests/flags.c
@@ -54,6 +54,34 @@ START_TEST (test_gpt_flag)
 }
 END_TEST
 
+/* TEST: Test partition type flag on msdos disklabel */
+START_TEST (test_msdos_flag)
+{
+        PedDevice* dev = ped_device_get (temporary_disk);
+        if (dev == NULL)
+                return;
+
+        PedDisk* disk = ped_disk_new_fresh (dev, ped_disk_type_get ("msdos"));
+        PedConstraint *constraint = ped_constraint_any (dev);
+        PedPartition *part = ped_partition_new (disk, PED_PARTITION_NORMAL,
+            ped_file_system_type_get("ext4"), 2048, 4096);
+        ped_partition_set_flag(part, PED_PARTITION_BLS_BOOT, 1);
+        // Type should remain set to BIOS_GRUB
+        ped_partition_set_system(part, ped_file_system_type_get("ext4"));
+
+        ped_disk_add_partition (disk, part, constraint);
+        ped_disk_commit (disk);
+        ped_constraint_destroy (constraint);
+
+        // Check flag to confirm it is still set
+        part = ped_disk_get_partition (disk, 1);
+        fail_if (ped_partition_get_flag(part, PED_PARTITION_BLS_BOOT) != 1, "BLS_BOOT flag not set");
+
+        ped_disk_destroy (disk);
+        ped_device_destroy (dev);
+}
+END_TEST
+
 int
 main (int argc, char **argv)
 {
@@ -61,6 +89,7 @@ main (int argc, char **argv)
         int number_failed;
         Suite* suite = suite_create ("Partition Flags");
         TCase* tcase_gpt = tcase_create ("GPT");
+        TCase* tcase_msdos = tcase_create ("MSDOS");
 
         /* Fail when an exception is raised */
         ped_exception_set_handler (_test_exception_handler);
@@ -71,6 +100,12 @@ main (int argc, char **argv)
         tcase_set_timeout (tcase_gpt, 0);
         suite_add_tcase (suite, tcase_gpt);
 
+        tcase_add_checked_fixture (tcase_msdos, create_disk, destroy_disk);
+        tcase_add_test (tcase_msdos, test_msdos_flag);
+        /* Disable timeout for this test */
+        tcase_set_timeout (tcase_msdos, 0);
+        suite_add_tcase (suite, tcase_msdos);
+
         SRunner* srunner = srunner_create (suite);
         srunner_run_all (srunner, CK_VERBOSE);