def test__extract_size_from_fdisk_good_output_multipath_device(self, _populate_disks_entries_mock, run_cmd_mock):
run_cmd_mock.return_value = FAKE_FDISK_GOOD_OUTPUT_MULTIPATH_DEVICE
dq = LinuxDiskDeviceQueries()
fake_device_filepath = "/dev/mapper/mpathc"
actual_size = dq._extract_size_from_fdisk(fake_device_filepath)
run_cmd_mock.assert_called_once_with(SIZE_FROM_FDISK + [fake_device_filepath])
self.assertEqual(17208180736, actual_size)
def test__extract_size_from_fdisk_good_output_partition(self, _populate_disks_entries_mock, run_cmd_mock):
run_cmd_mock.return_value = FAKE_FDISK_GOOD_OUTPUT_PARTITION
dq = LinuxDiskDeviceQueries()
fake_device_filepath = "/dev/sda6"
actual_size = dq._extract_size_from_fdisk(fake_device_filepath)
run_cmd_mock.assert_called_once_with(SIZE_FROM_FDISK + [fake_device_filepath])
self.assertEqual(125998989312, actual_size)
def test__extract_size_from_fdisk_good_output(self, _populate_disks_entries_mock, run_cmd_mock):
run_cmd_mock.return_value = FAKE_FDISK_GOOD_OUTPUT
dq = LinuxDiskDeviceQueries()
fake_device_filepath = "/dev/sda"
actual_size = dq._extract_size_from_fdisk(fake_device_filepath)
run_cmd_mock.assert_called_with(SIZE_FROM_FDISK + [fake_device_filepath])
self.assertEqual(8185184256, actual_size)
def test__extract_size_from_fdisk_bad_output(self, _populate_disks_entries_mock, run_cmd_mock):
run_cmd_mock.return_value = FAKE_FDISK_BAD_OUTPUT
dq = LinuxDiskDeviceQueries()
self.assertEqual(None, dq._extract_size_from_fdisk("/dev/sda"))