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