def test_primary(self): expected_partitions = [ partition.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 * [partition.Partition('name', 'count', 'device', 'begin', 'end', 'primary')]) self.assertEqual('extended', self.prtd.next_type())
def setUp(self): super(TestPartition, self).setUp() self.pt = partition.Partition('name', 'count', 'device', 'begin', 'end', 'partition_type')
def test_next_type_has_extended(self): self.prtd.label = 'msdos' self.prtd.partitions.append( partition.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( partition.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( partition.Partition('name', 'count', 'device', 'begin', 'end', 'primary')) self.assertEqual('end', self.prtd.next_begin())