def test_info(self):
"""Test info messages. """
checker = StorageChecker()
def error_check(storage, constraints, report_error, report_warning):
report_error("error")
def warning_check(storage, constraints, report_error, report_warning):
report_warning("warning")
def skipped_check(storage, constraints, report_error, report_warning):
report_warning("skipped")
checker.add_constraint("x", None)
checker.add_check(error_check)
checker.add_check(warning_check)
checker.add_check(skipped_check)
report = checker.check(None, skip=(skipped_check,))
self.assertListEqual(report.info, [
"Storage check started with constraints {'x': None}.",
"Run sanity check error_check.",
"Found sanity error: error",
"Run sanity check warning_check.",
"Found sanity warning: warning",
"Skipped sanity check skipped_check.",
"Storage check finished with failure(s)."
])
def test_simple_info(self):
"""Test simple info messages. """
checker = StorageChecker()
report = checker.check(None)
self.assertListEqual(report.info, [
"Storage check started with constraints {}.",
"Storage check finished with success."
])
def test_nocheck(self):
"""Test a check with no checks."""
checker = StorageChecker()
report = checker.check(None)
assert report.success is True
assert report.failure is False
assert report.errors == []
assert report.warnings == []
def test_nocheck(self):
"""Test a check with no checks."""
checker = StorageChecker()
report = checker.check(None)
self.assertEqual(report.success, True)
self.assertEqual(report.failure, False)
self.assertListEqual(report.errors, [])
self.assertListEqual(report.warnings, [])
def test_simple_constraints(self):
"""Test simple constraint adding."""
checker = StorageChecker()
# Try to add a new constraint with a wrong method.
self.assertRaises(KeyError, checker.set_constraint, "x", None)
# Try to add a new constraint two times.
checker.add_constraint("x", None)
self.assertRaises(KeyError, checker.add_constraint, "x", None)
def default_settings_test(self):
"""Check the default storage checker."""
checker = StorageChecker()
checker.set_default_constraints()
checker.set_default_checks()
self.assertEqual(
checker.constraints, {
checks.STORAGE_MIN_RAM: Size("320 MiB"),
checks.STORAGE_ROOT_DEVICE_TYPES: set(),
checks.STORAGE_MIN_PARTITION_SIZES: {
'/': Size("250 MiB"),
'/usr': Size("250 MiB"),
'/tmp': Size("50 MiB"),
'/var': Size("384 MiB"),
'/home': Size("100 MiB"),
'/boot': Size("200 MiB")
},
checks.STORAGE_REQ_PARTITION_SIZES: dict(),
checks.STORAGE_MUST_BE_ON_LINUXFS: {
'/', '/var', '/tmp', '/usr', '/home', '/usr/share',
'/usr/lib'
},
checks.STORAGE_MUST_BE_ON_ROOT: {
'/bin', '/dev', '/sbin', '/etc', '/lib', '/root', '/mnt',
'lost+found', '/proc'
},
checks.STORAGE_MUST_NOT_BE_ON_ROOT: set(),
checks.STORAGE_REFORMAT_ALLOWLIST:
{'/boot', '/var', '/tmp', '/usr'},
checks.STORAGE_REFORMAT_BLOCKLIST:
{'/home', '/usr/local', '/opt', '/var/www'},
checks.STORAGE_SWAP_IS_RECOMMENDED: False,
checks.STORAGE_LUKS2_MIN_RAM: Size("128 MiB"),
})
self.assertEqual(checker.checks, [
checks.verify_root,
checks.verify_s390_constraints,
checks.verify_partition_formatting,
checks.verify_partition_sizes,
checks.verify_partition_format_sizes,
checks.verify_bootloader,
checks.verify_gpt_biosboot,
checks.verify_swap,
checks.verify_swap_uuid,
checks.verify_mountpoints_on_linuxfs,
checks.verify_mountpoints_on_root,
checks.verify_mountpoints_not_on_root,
checks.verify_unlocked_devices_have_key,
checks.verify_luks_devices_have_key,
checks.verify_luks2_memory_requirements,
checks.verify_mounted_partitions,
checks.verify_lvm_destruction,
])
def test_dictionary_constraints(self):
"""Test the dictionary constraints."""
checker = StorageChecker()
checker.add_constraint("x", {"a": 1, "b": 2, "c": 3})
self.assertIn("x", checker.constraints)
self.assertEqual(checker.constraints["x"], {"a": 1, "b": 2, "c": 3})
checker.set_constraint("x", {"e": 4, "f": 5})
self.assertIn("x", checker.constraints)
self.assertEqual(checker.constraints["x"], {"e": 4, "f": 5})
def test_dictionary_constraints(self):
"""Test the dictionary constraints."""
checker = StorageChecker()
checker.add_constraint("x", {"a": 1, "b": 2, "c": 3})
assert "x" in checker.constraints
assert checker.constraints["x"] == {"a": 1, "b": 2, "c": 3}
checker.set_constraint("x", {"e": 4, "f": 5})
assert "x" in checker.constraints
assert checker.constraints["x"] == {"e": 4, "f": 5}
def test_simple_warning(self):
"""Test an simple warning report."""
checker = StorageChecker()
def check(storage, constraints, report_error, report_warning):
report_warning("warning")
checker.add_check(check)
report = checker.check(None)
self.assertEqual(report.success, False)
self.assertListEqual(report.errors, [])
self.assertListEqual(report.warnings, ["warning"])
def test_simple_warning(self):
"""Test an simple warning report."""
checker = StorageChecker()
def check(storage, constraints, report_error, report_warning):
report_warning("warning")
checker.add_check(check)
report = checker.check(None)
assert report.success is False
assert report.errors == []
assert report.warnings == ["warning"]
def test_check_constraints(self):
"""Test constraints checking."""
checker = StorageChecker()
def check(storage, constraints, report_error, report_warning):
report_warning("%s" % constraints)
checker.add_check(check)
report = checker.check(None)
self.assertListEqual(report.warnings, ["{}"])
checker.add_constraint("x", 1)
report = checker.check(None)
self.assertListEqual(report.warnings, ["{'x': 1}"])
checker.set_constraint("x", 0)
report = checker.check(None)
self.assertListEqual(report.warnings, ["{'x': 0}"])
def test_complicated(self):
"""Run a complicated check."""
checker = StorageChecker()
# Set the checks,
def check_x(storage, constraints, report_error, report_warning):
if constraints["x"] != 1:
report_error("x is not equal to 1")
def check_y(storage, constraints, report_error, report_warning):
if constraints["y"] != 2:
report_error("y is not equal to 2")
def check_z(storage, constraints, report_error, report_warning):
if constraints["z"] != 3:
report_error("z is not equal to 3")
checker.add_check(check_x)
checker.add_check(check_y)
checker.add_check(check_z)
# Set the constraints.
checker.add_constraint("x", 1)
checker.add_constraint("y", 2)
checker.add_constraint("z", 3)
# Run the checker. OK
report = checker.check(None)
self.assertEqual(report.success, True)
self.assertListEqual(report.errors, [])
self.assertListEqual(report.warnings, [])
# Set constraints to different values.
checker.set_constraint("x", 0)
checker.set_constraint("y", 1)
checker.set_constraint("z", 2)
# Run the checker. FAIL
report = checker.check(None)
self.assertEqual(report.success, False)
self.assertListEqual(report.errors, [
"x is not equal to 1",
"y is not equal to 2",
"z is not equal to 3"
])
self.assertListEqual(report.warnings, [])
# Run the checker. Test SKIP.
report = checker.check(None, skip=(check_y,))
self.assertEqual(report.success, False)
self.assertListEqual(report.errors, [
"x is not equal to 1",
"z is not equal to 3"
])
self.assertListEqual(report.warnings, [])
# Run the checker. Test CONSTRAINTS.
constraints = {"x": 1, "y": 2, "z": 3}
report = checker.check(None, constraints=constraints)
self.assertEqual(report.success, True)
self.assertListEqual(report.errors, [])
self.assertListEqual(report.warnings, [])
# Remove checks.
checker.remove_check(check_x)
checker.remove_check(check_y)
checker.remove_check(check_z)
report = checker.check(None)
self.assertEqual(report.success, True)
self.assertListEqual(report.errors, [])
self.assertListEqual(report.warnings, [])