Merge "Sync the make_partitions function with ironic"

This commit is contained in:
Jenkins 2015-10-05 18:29:27 +00:00 committed by Gerrit Code Review
commit 1022292f5a
2 changed files with 4 additions and 67 deletions

@ -66,8 +66,6 @@ LOG = logging.getLogger(__name__)
_PARTED_PRINT_RE = re.compile(r"^(\d+):([\d\.]+)MiB:"
"([\d\.]+)MiB:([\d\.]+)MiB:(\w*)::(\w*)")
_ISCSI_RE = re.compile(r"^ip-[\d+.]*:\w+-iscsi-[\w+.]*-lun-\d+")
def list_partitions(device):
"""Get partitions information from given device.
@ -100,12 +98,6 @@ def list_partitions(device):
return result
def is_iscsi_device(dev):
"""check whether the device path belongs to an iscsi device. """
basename = os.path.basename(dev)
return bool(_ISCSI_RE.match(basename))
def get_disk_identifier(dev):
"""Get the disk identifier from the disk being exposed by the ramdisk.
@ -155,12 +147,7 @@ def make_partitions(dev, root_mb, swap_mb, ephemeral_mb,
LOG.debug("Starting to partition the disk device: %(dev)s "
"for node %(node)s",
{'dev': dev, 'node': node_uuid})
if is_iscsi_device(dev):
part_template = dev + '-part%d'
else:
part_template = dev + '%d'
part_template = dev + '-part%d'
part_dict = {}
# For uefi localboot, switch partition table to gpt and create the efi

@ -259,9 +259,9 @@ class MakePartitionsTestCase(test_base.BaseTestCase):
'mkpart', 'primary', 'linux-swap', '2049', '2561',
'mkpart', 'primary', '', '2561', '3585']
self.dev = 'fake-dev'
expected_result = {'ephemeral': 'fake-dev1',
'swap': 'fake-dev2',
'root': 'fake-dev3'}
expected_result = {'ephemeral': 'fake-dev-part1',
'swap': 'fake-dev-part2',
'root': 'fake-dev-part3'}
cmd = self._get_parted_cmd(self.dev) + expected_mkpart
mock_exc.return_value = (None, None)
result = disk_utils.make_partitions(
@ -272,48 +272,6 @@ class MakePartitionsTestCase(test_base.BaseTestCase):
mock_exc.assert_has_calls([parted_call])
self.assertEqual(expected_result, result)
def test_make_partitions_with_iscsi_device(self, mock_exc):
self.ephemeral_mb = 2048
expected_mkpart = ['mkpart', 'primary', '', '1', '2049',
'mkpart', 'primary', 'linux-swap', '2049', '2561',
'mkpart', 'primary', '', '2561', '3585']
self.dev = 'ip-10.10.1.10:abc-iscsi-xyz-lun-123'
expected_result = {'ephemeral': self.dev + '-part1',
'swap': self.dev + '-part2',
'root': self.dev + '-part3'}
cmd = self._get_parted_cmd(self.dev) + expected_mkpart
mock_exc.return_value = (None, None)
result = disk_utils.make_partitions(
self.dev, self.root_mb, self.swap_mb, self.ephemeral_mb,
self.configdrive_mb, self.node_uuid)
parted_call = mock.call(*cmd, run_as_root=True, check_exit_code=[0])
mock_exc.assert_has_calls([parted_call])
self.assertEqual(expected_result, result)
def test_make_partitions_with_iscsi_device_uefi_local(self, mock_exc):
self.ephemeral_mb = 2048
expected_mkpart = ['mkpart', 'primary', 'fat32', '1', '201',
'set', '1', 'boot', 'on',
'mkpart', 'primary', '', '201', '2249',
'mkpart', 'primary', 'linux-swap', '2249', '2761',
'mkpart', 'primary', '', '2761', '3785']
self.dev = 'ip-10.10.1.10:abc-iscsi-xyz-lun-123'
expected_result = {'efi system partition': self.dev + '-part1',
'ephemeral': self.dev + '-part2',
'swap': self.dev + '-part3',
'root': self.dev + '-part4'}
cmd = self._get_parted_cmd_uefi(self.dev) + expected_mkpart
mock_exc.return_value = (None, None)
result = disk_utils.make_partitions(
self.dev, self.root_mb, self.swap_mb, self.ephemeral_mb,
self.configdrive_mb, self.node_uuid, boot_option='local',
boot_mode='uefi')
parted_call = mock.call(*cmd, run_as_root=True, check_exit_code=[0])
mock_exc.assert_has_calls([parted_call])
self.assertEqual(expected_result, result)
@mock.patch.object(disk_utils, 'get_dev_block_size')
@mock.patch.object(utils, 'execute')
@ -581,11 +539,3 @@ class OtherFunctionTestCase(test_base.BaseTestCase):
return_value=mb + 1)
self.assertEqual(2, disk_utils.get_image_mb('x', False))
self.assertEqual(2, disk_utils.get_image_mb('x', True))
def test_is_iscsi_device_True(self):
device = '/dev/disk/by-path/ip-1.2.3.4:5678-iscsi-iqn.fake-lun-9'
self.assertTrue(disk_utils.is_iscsi_device(device))
def test_is_iscsi_device_False(self):
device = '/dev/disk/by-path/fake-dev'
self.assertFalse(disk_utils.is_iscsi_device(device))