def test_ParsePartitionTableMBR(self):
self.expectedCommands.append(('parted -s -m /dev/sda unit MB print',
self._getMBRPartitionTableInMachineFormat(diskSizeGB=self.diskSizeGB)))
tested = PartitionTable("/dev/sda", layoutScheme="MBR")
parsed = tested.parsePartitionTable()
self.assertEquals(len(parsed), 2)
self.assertEquals(parsed[0]['device'], '/dev/sda1')
self.assertEquals(parsed[0]['sizeMB'], self.BOOT_PARTITION_SIZE)
self.assertEquals(parsed[0]['fs'], "ext4")
self.assertEquals(parsed[0]['flags'], "boot")
MbrLvmPartition = "/dev/sda2"
self.assertEquals(parsed[1]['device'], MbrLvmPartition)
self.assertEquals(parsed[1]['sizeMB'],
self.diskSizeGB * self._NR_MBS_IN_GB - self.BOOT_PARTITION_SIZE)
self.assertEquals(parsed[1]['fs'], "")
self.assertEquals(parsed[1]['flags'], "lvm")
self.assertEquals(len(self.expectedCommands), 0)
def test_ParsePartitionTable(self):
example = "\n".join([
"# partition table of /dev/sda", "unit: sectors", "",
"/dev/sda1 : start= 2048, size= 16023552, Id=82",
"/dev/sda2 : start= 16025600, size=484091904, Id=83, bootable",
"/dev/sda3 : start= 0, size= 0, Id= 0",
"/dev/sda4 : start= 0, size= 0, Id= 0", ""
])
self.expectedCommands.append(('sfdisk --dump /dev/sda', example))
tested = PartitionTable("/dev/sda")
parsed = tested.parsePartitionTable()
self.assertEquals(len(parsed), 2)
self.assertEquals(parsed[0]['device'], '/dev/sda1')
self.assertEquals(parsed[0]['sizeMB'], 16023552 / 2 / 1024)
self.assertEquals(parsed[0]['id'], 0x82)
self.assertEquals(parsed[1]['device'], '/dev/sda2')
self.assertEquals(parsed[1]['sizeMB'], 484091904 / 2 / 1024)
self.assertEquals(parsed[1]['id'], 0x83)
self.assertEquals(len(self.expectedCommands), 0)
def test_ParsePartitionTableMBR(self):
self.expectedCommands.append(
('parted -s -m /dev/sda unit MB print',
self._getMBRPartitionTableInMachineFormat(
diskSizeGB=self.diskSizeGB)))
tested = PartitionTable("/dev/sda", layoutScheme="MBR")
parsed = tested.parsePartitionTable()
self.assertEquals(len(parsed), 2)
self.assertEquals(parsed[0]['device'], '/dev/sda1')
self.assertEquals(parsed[0]['sizeMB'], self.BOOT_PARTITION_SIZE)
self.assertEquals(parsed[0]['fs'], "ext4")
self.assertEquals(parsed[0]['flags'], "boot")
MbrLvmPartition = "/dev/sda2"
self.assertEquals(parsed[1]['device'], MbrLvmPartition)
self.assertEquals(
parsed[1]['sizeMB'],
self.diskSizeGB * self._NR_MBS_IN_GB - self.BOOT_PARTITION_SIZE)
self.assertEquals(parsed[1]['fs'], "")
self.assertEquals(parsed[1]['flags'], "lvm")
self.assertEquals(len(self.expectedCommands), 0)
def test_ParsePartitionTableGPT(self):
self.expectedCommands.append(('parted -s -m /dev/sda unit MB print',
self._getPartitionTableInMachineFormat(diskSizeGB=self.diskSizeGB)))
tested = PartitionTable("/dev/sda")
parsed = tested.parsePartitionTable()
self.assertEquals(len(parsed), 3)
self.assertEquals(parsed[0]['device'], '/dev/sda1')
self.assertEquals(parsed[0]['sizeMB'], 2)
self.assertEquals(parsed[0]['fs'], "")
self.assertEquals(parsed[0]['flags'], "bios_grub")
self.assertEquals(parsed[1]['device'], '/dev/sda2')
self.assertEquals(parsed[1]['sizeMB'], self.BOOT_PARTITION_SIZE)
self.assertEquals(parsed[1]['fs'], "ext4")
self.assertEquals(parsed[1]['flags'], "boot")
self.assertEquals(parsed[2]['device'], self.GPT_LVM_PARTITION)
self.assertEquals(parsed[2]['sizeMB'],
self.diskSizeGB * self._NR_MBS_IN_GB -
(self.BIOS_BOOT_PARTITION_SIZE + self.BOOT_PARTITION_SIZE))
self.assertEquals(parsed[2]['fs'], "")
self.assertEquals(parsed[2]['flags'], "lvm")
self.assertEquals(len(self.expectedCommands), 0)
def test_ParsePartitionTableGPT(self):
self.expectedCommands.append(('parted -s -m /dev/sda unit MB print',
self._getPartitionTableInMachineFormat(
diskSizeGB=self.diskSizeGB)))
tested = PartitionTable("/dev/sda")
parsed = tested.parsePartitionTable()
self.assertEquals(len(parsed), 3)
self.assertEquals(parsed[0]['device'], '/dev/sda1')
self.assertEquals(parsed[0]['sizeMB'], 2)
self.assertEquals(parsed[0]['fs'], "")
self.assertEquals(parsed[0]['flags'], "bios_grub")
self.assertEquals(parsed[1]['device'], '/dev/sda2')
self.assertEquals(parsed[1]['sizeMB'], self.BOOT_PARTITION_SIZE)
self.assertEquals(parsed[1]['fs'], "ext4")
self.assertEquals(parsed[1]['flags'], "boot")
self.assertEquals(parsed[2]['device'], self.GPT_LVM_PARTITION)
self.assertEquals(
parsed[2]['sizeMB'], self.diskSizeGB * self._NR_MBS_IN_GB -
(self.BIOS_BOOT_PARTITION_SIZE + self.BOOT_PARTITION_SIZE))
self.assertEquals(parsed[2]['fs'], "")
self.assertEquals(parsed[2]['flags'], "lvm")
self.assertEquals(len(self.expectedCommands), 0)
def test_ParsePartitionTable(self):
example = "\n".join(
[
"# partition table of /dev/sda",
"unit: sectors",
"",
"/dev/sda1 : start= 2048, size= 16023552, Id=82",
"/dev/sda2 : start= 16025600, size=484091904, Id=83, bootable",
"/dev/sda3 : start= 0, size= 0, Id= 0",
"/dev/sda4 : start= 0, size= 0, Id= 0",
"",
]
)
self.expectedCommands.append(("sfdisk --dump /dev/sda", example))
tested = PartitionTable("/dev/sda")
parsed = tested.parsePartitionTable()
self.assertEquals(len(parsed), 2)
self.assertEquals(parsed[0]["device"], "/dev/sda1")
self.assertEquals(parsed[0]["sizeMB"], 16023552 / 2 / 1024)
self.assertEquals(parsed[0]["id"], 0x82)
self.assertEquals(parsed[1]["device"], "/dev/sda2")
self.assertEquals(parsed[1]["sizeMB"], 484091904 / 2 / 1024)
self.assertEquals(parsed[1]["id"], 0x83)
self.assertEquals(len(self.expectedCommands), 0)
