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