def test_get_helper(self, *args): """Test get_format_helper for disklabels.""" device_get_disklabel_type = args[0] device = Mock() device.is_disk = True data = Mock() self.assertEqual(get_format_helper(data, device), self.helper_class) # no disklabel type reported by udev/blkid -> get_format_helper does not return # disklabel helper device_get_disklabel_type.return_value = None self.assertNotEqual(get_format_helper(data, device), self.helper_class) device_get_disklabel_type.return_value = "dos"
def test_get_helper(self, *args): if self.udev_type is None: return data = dict() device = Mock() with patch("blivet.udev.device_get_format", return_value=self.udev_type): self.assertEqual(get_format_helper(data, device), self.helper_class, msg="get_format_helper failed for %s" % self.udev_type)
def test_get_helper(self, *args): if self.helper_class is None: return partition = PartitionDevice("testpartitiondev") data = dict() fmt_class = get_device_format_class(self.helper_class._type_specifier) if fmt_class is None: self.skipTest("failed to look up format class for %s" % self.helper_class._type_specifier) data["ID_FS_TYPE"] = self.helper_class._base_type_specifier data["DEVTYPE"] = "partition" partition._bootable = self.helper_class._bootable partition._size = fmt_class._min_size self.assertEqual(get_format_helper(data, partition), self.helper_class)