def test_match(self, *args):
"""Test matching for disklabel format populator."""
device_get_disklabel_type = args[0]
device_get_format = args[1]
device_is_biosraid_member = args[2]
is_mpath_member = args[3]
device = Mock()
device.is_disk = True
data = Mock()
self.assertTrue(self.helper_class.match(data, device))
# ID_PART_TABLE_TYPE is required in udev data
device_get_disklabel_type.return_value = None
self.assertFalse(self.helper_class.match(data, device))
device_get_disklabel_type.return_value = "dos"
# no match for whole-disk iso9660 filesystems (isohybrid media)
device_get_format.return_value = "iso9660"
self.assertFalse(self.helper_class.match(data, device))
device_get_format.return_value = None
# no match for biosraid members
device_is_biosraid_member.return_value = True
self.assertFalse(self.helper_class.match(data, device))
device_is_biosraid_member.return_value = False
# no match for multipath members
is_mpath_member.return_value = True
self.assertFalse(self.helper_class.match(data, device))
is_mpath_member.return_value = False
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"
