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):
"""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.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)
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)
