def setUp(self):
self.pt = PartitionTable(512)
class TestParititionTable(unittest.TestCase):
def setUp(self):
self.pt = PartitionTable(512)
def test_when_init_then_partition_table_has_no_partition_entries(self):
self.assertEqual(len(self.pt.partition_entries()), 0)
def test_when_create_partition_then_it_has_one_partition(self):
size = mb_to_sectors(10, 512)
self.pt.create_partition(0, size)
self.assertEqual(len(self.pt.partition_entries()), 1)
def test_when_delete_partition_then_partition_no_longer_exists(self):
size = mb_to_sectors(10, 512)
self.pt.create_partition(0, size)
self.pt.delete_partition(0)
self.assertEqual(len(self.pt.partition_entries()), 0)
def test_when_create_partition_that_overlaps_another_then_raises_InvalidSectorRangeException(self):
size = mb_to_sectors(10, 512)
self.pt.create_partition(0, size)
with self.assertRaises(InvalidSectorRangeException):
self.pt.create_partition(10, size)