def check_completeness_test(self):
"""Test CheckCompleteness."""
dev1 = StorageDevice("dev1",
fmt=get_format("ext4"),
size=Size("10 GiB"),
exists=True)
dev2 = MDRaidArrayDevice(name="dev2",
size=Size("500 MiB"),
level=1,
member_devices=2,
total_devices=2,
exists=True)
dev3 = LVMVolumeGroupDevice("dev3", pv_count=2, exists=True)
self._add_device(dev1)
self._add_device(dev2)
self._add_device(dev3)
self._check_report(self.interface.CheckCompleteness("dev1"))
self._check_report(
self.interface.CheckCompleteness("dev2"),
"This Software RAID array is missing 2 of 2 member partitions. "
"You can remove it or select a different device.")
self._check_report(
self.interface.CheckCompleteness("dev3"),
"This LVM Volume Group is missing 2 of 2 physical volumes. "
"You can remove it or select a different device.")
dev1.complete = False
self._check_report(
self.interface.CheckCompleteness("dev1"),
"This blivet device is missing member devices. "
"You can remove it or select a different device.")
def is_device_test(self):
"""Test IsDevice."""
dev1 = StorageDevice("dev1",
fmt=get_format("ext4"),
size=Size("10 GiB"),
exists=True)
self.assertEqual(self.interface.IsDevice("dev1"), False)
self._add_device(dev1)
self.assertEqual(self.interface.IsDevice("dev1"), True)
dev1.complete = False
self.assertEqual(self.interface.IsDevice("dev1"), True)
self.storage.devicetree.hide(dev1)
self.assertEqual(self.interface.IsDevice("dev1"), True)
def test_is_device(self):
"""Test IsDevice."""
dev1 = StorageDevice("dev1",
fmt=get_format("ext4"),
size=Size("10 GiB"),
exists=True)
assert self.interface.IsDevice("dev1") is False
self._add_device(dev1)
assert self.interface.IsDevice("dev1") is True
dev1.complete = False
assert self.interface.IsDevice("dev1") is True
self.storage.devicetree.hide(dev1)
assert self.interface.IsDevice("dev1") is True
def test_get_device_by_name(self):
dt = DeviceTree()
dev1 = StorageDevice("dev1", exists=False, parents=[])
dev2 = StorageDevice("dev2", exists=False, parents=[dev1])
dt._add_device(dev1)
dt._add_device(dev2)
self.assertIsNone(dt.get_device_by_name("dev3"))
self.assertEqual(dt.get_device_by_name("dev2"), dev2)
self.assertEqual(dt.get_device_by_name("dev1"), dev1)
dev2.complete = False
self.assertEqual(dt.get_device_by_name("dev2"), None)
self.assertEqual(dt.get_device_by_name("dev2", incomplete=True), dev2)
dev3 = StorageDevice("dev3", exists=True, parents=[])
dt._add_device(dev3)
dt.hide(dev3)
self.assertIsNone(dt.get_device_by_name("dev3"))
self.assertEqual(dt.get_device_by_name("dev3", hidden=True), dev3)
def test_get_device_by_name(self):
dt = DeviceTree()
dev1 = StorageDevice("dev1", exists=False, parents=[])
dev2 = StorageDevice("dev2", exists=False, parents=[dev1])
dt._add_device(dev1)
dt._add_device(dev2)
self.assertIsNone(dt.get_device_by_name("dev3"))
self.assertEqual(dt.get_device_by_name("dev2"), dev2)
self.assertEqual(dt.get_device_by_name("dev1"), dev1)
dev2.complete = False
self.assertEqual(dt.get_device_by_name("dev2"), None)
self.assertEqual(dt.get_device_by_name("dev2", incomplete=True), dev2)
dev3 = StorageDevice("dev3", exists=True, parents=[])
dt._add_device(dev3)
dt.hide(dev3)
self.assertIsNone(dt.get_device_by_name("dev3"))
self.assertEqual(dt.get_device_by_name("dev3", hidden=True), dev3)
