def test_primary(self):
expected_partitions = [
objects.Partition('name', 'count', 'device', 'begin', 'end',
'primary')
]
self.prtd.partitions.extend(expected_partitions)
self.assertEqual(expected_partitions, self.prtd.primary)
def test_next_type_no_extended(self):
self.prtd.label = 'msdos'
self.assertEqual('primary', self.prtd.next_type())
self.prtd.partitions.extend(
3 * [objects.Partition('name', 'count', 'device', 'begin',
'end', 'primary')])
self.assertEqual('extended', self.prtd.next_type())
def test_conversion(self):
prt = objects.Partition(name='name',
count='count',
device='device',
begin='begin',
end='end',
partition_type='primary',
keep_data=True)
self.prtd.partitions.append(prt)
serialized = self.prtd.to_dict()
assert serialized == {
'label': 'label',
'name': 'name',
'partitions': [
prt.to_dict(),
],
'install_bootloader': False,
}
new_prtd = objects.Parted.from_dict(serialized)
assert serialized == new_prtd.to_dict()
def setUp(self):
super(TestPartition, self).setUp()
self.pt = objects.Partition('name', 'count', 'device', 'begin', 'end',
'partition_type')
def test_next_type_has_extended(self):
self.prtd.label = 'msdos'
self.prtd.partitions.append(
objects.Partition('name', 'count', 'device', 'begin', 'end',
'extended'))
self.assertEqual('logical', self.prtd.next_type())
def test_next_count_has_logical(self):
self.prtd.partitions.append(
objects.Partition('name', 'count', 'device', 'begin', 'end',
'logical'))
self.assertEqual(6, self.prtd.next_count('logical'))
def test_next_begin_no_last_extended_partition(self):
self.prtd.partitions.append(
objects.Partition('name', 'count', 'device', 'begin', 'end',
'primary'))
self.assertEqual('end', self.prtd.next_begin())
def test_next_begin_last_extended_partition(self):
self.prtd.partitions.append(
objects.Partition('name', 'count', 'device', 1, 100, 'extended'))
self.assertEqual(2, self.prtd.next_begin())
