From 29d32140511005c6989c7fe6844435287adb2d2a Mon Sep 17 00:00:00 2001 From: Ryan Harper Date: Tue, 6 Oct 2015 16:47:47 -0500 Subject: [PATCH] unittest: filesystemmodel - get_disk with partition name - get_all_disks - get_available_disks Signed-off-by: Ryan Harper --- subiquity/tests/test_models_filesystems.py | 34 ++++++++++++++++++++++ 1 file changed, 34 insertions(+) diff --git a/subiquity/tests/test_models_filesystems.py b/subiquity/tests/test_models_filesystems.py index beea8732..42388410 100644 --- a/subiquity/tests/test_models_filesystems.py +++ b/subiquity/tests/test_models_filesystems.py @@ -76,6 +76,40 @@ class TestFilesystemModel(testtools.TestCase): print(test_disk) self.assertEqual(test_disk, disk) + def test_filesystemmodel_get_disk_from_partition(self): + self.fsm.probe_storage() + diskname = random.choice(list(self.fsm.info.keys())) + disk = self.fsm.get_disk(diskname) + disk.add_partition(1, disk.freespace, None, None, flag='raid') + + partpath = '{}{}'.format(disk.path, 1) + print(partpath) + self.assertTrue(partpath[-1], 1) + test_disk = self.fsm.get_disk(partpath) + print(disk) + print(test_disk) + self.assertEqual(test_disk, disk) + + def test_filesystemmodel_get_all_disks(self): + self.fsm.probe_storage() + all_disks = self.fsm.get_all_disks() + for disk in all_disks: + self.assertTrue(disk in self.fsm.devices.values()) + + def test_filesystemmodel_get_available_disks(self): + ''' occupy one of the probed disks and ensure + that it's not included in the available disks + result since it's not actually avaialable + ''' + self.fsm.probe_storage() + diskname = random.choice(list(self.fsm.info.keys())) + disk = self.fsm.get_disk(diskname) + disk.add_partition(1, disk.freespace, None, None, flag='raid') + + avail_disks = self.fsm.get_available_disks() + self.assertLess(len(avail_disks), len(self.fsm.devices.values())) + self.assertTrue(disk not in avail_disks) + class TestBlockdev(testtools.TestCase): def setUp(self):