def test__get_sysfs_partitions(self, _populate_disks_entries_mock, extract_size_mock, glob_mock):
glob_mock.return_value = ["/dev/sda1", "/dev/sda2", "/dev/sda3"]
fake_size = 1024
extract_size_mock.return_value = fake_size
dq = LinuxDiskDeviceQueries()
expected_results = [
DiskEntry("sda1", "/dev/sda1", fake_size, (250, 1), (1, 1, 1, 1)),
DiskEntry("sda2", "/dev/sda2", fake_size, (250, 2), (2, 2, 2, 2)),
DiskEntry("sda3", "/dev/sda3", fake_size, (250, 3), (3, 3, 3, 3)),
]
# TODO: This is kinda ugly, better would be to mock _map_hctl_to_disk_device_names
dq.hctl_map = {"sda1": (1, 1, 1, 1), "sda2": (2, 2, 2, 2), "sda3": (3, 3, 3, 3)}
actual_results = dq._get_sysfs_partitions("sda")
self.assertEqual(len(expected_results), len(actual_results))
for expected_result, actual_result in map(None, expected_results, actual_results):
self.assertEqual(expected_result.name, actual_result.name)
self.assertEqual(expected_result.filepath, actual_result.filepath)
self.assertEqual(expected_result.size, actual_result.size)
self.assertEqual(expected_result.major_minor, actual_result.major_minor)
self.assertEqual(expected_result.hctl, actual_result.hctl)
