X-Git-Url: https://jxself.org/git/?a=blobdiff_plain;ds=inline;f=kconfig_hardened_check%2Ftest_engine.py;h=531dfeead6763e9f5cb8e520d202123146c217ff;hb=a161c6b03331da0711018c688549a0bae306e389;hp=433e5844e596f6c5122618116857f32cd107aca8;hpb=d6caae5328a051d33e43ffec040cae03d8f6a07f;p=kconfig-hardened-check.git diff --git a/kconfig_hardened_check/test_engine.py b/kconfig_hardened_check/test_engine.py index 433e584..531dfee 100644 --- a/kconfig_hardened_check/test_engine.py +++ b/kconfig_hardened_check/test_engine.py @@ -15,7 +15,7 @@ import io import sys from collections import OrderedDict import json -from .engine import KconfigCheck, CmdlineCheck, VersionCheck, OR, AND, populate_with_data, perform_checks, override_expected_value +from .engine import KconfigCheck, CmdlineCheck, SysctlCheck, VersionCheck, OR, AND, populate_with_data, perform_checks, override_expected_value class TestEngine(unittest.TestCase): @@ -26,6 +26,7 @@ class TestEngine(unittest.TestCase): config_checklist = [] config_checklist += [KconfigCheck('reason_1', 'decision_1', 'KCONFIG_NAME', 'expected_1')] config_checklist += [CmdlineCheck('reason_2', 'decision_2', 'cmdline_name', 'expected_2')] + config_checklist += [SysctlCheck('reason_3', 'decision_3', 'sysctl_name', 'expected_3')] # 2. prepare the parsed kconfig options parsed_kconfig_options = OrderedDict() @@ -35,25 +36,31 @@ class TestEngine(unittest.TestCase): parsed_cmdline_options = OrderedDict() parsed_cmdline_options['cmdline_name'] = 'expected_2' - # 4. prepare the kernel version + # 4. prepare the parsed sysctl options + parsed_sysctl_options = OrderedDict() + parsed_sysctl_options['sysctl_name'] = 'expected_3' + + # 5. prepare the kernel version kernel_version = (42, 43) - # 5. run the engine - self.run_engine(config_checklist, parsed_kconfig_options, parsed_cmdline_options, kernel_version) + # 6. run the engine + self.run_engine(config_checklist, parsed_kconfig_options, parsed_cmdline_options, parsed_sysctl_options, kernel_version) - # 6. check that the results are correct + # 7. check that the results are correct result = [] self.get_engine_result(config_checklist, result, 'json') self.assertEqual(... """ @staticmethod - def run_engine(checklist, parsed_kconfig_options, parsed_cmdline_options, kernel_version): + def run_engine(checklist, parsed_kconfig_options, parsed_cmdline_options, parsed_sysctl_options, kernel_version): # populate the checklist with data if parsed_kconfig_options: populate_with_data(checklist, parsed_kconfig_options, 'kconfig') if parsed_cmdline_options: populate_with_data(checklist, parsed_cmdline_options, 'cmdline') + if parsed_sysctl_options: + populate_with_data(checklist, parsed_sysctl_options, 'sysctl') if kernel_version: populate_with_data(checklist, kernel_version, 'version') @@ -120,7 +127,7 @@ class TestEngine(unittest.TestCase): parsed_kconfig_options['CONFIG_NAME_9'] = '0' # 3. run the engine - self.run_engine(config_checklist, parsed_kconfig_options, None, None) + self.run_engine(config_checklist, parsed_kconfig_options, None, None, None) # 4. check that the results are correct result = [] @@ -163,7 +170,7 @@ class TestEngine(unittest.TestCase): parsed_cmdline_options['name_9'] = '0' # 3. run the engine - self.run_engine(config_checklist, None, parsed_cmdline_options, None) + self.run_engine(config_checklist, None, parsed_cmdline_options, None, None) # 4. check that the results are correct result = [] @@ -182,6 +189,49 @@ class TestEngine(unittest.TestCase): ["name_10", "cmdline", "is not off", "decision_10", "reason_10", "FAIL: is off, not found"]] ) + def test_simple_sysctl(self): + # 1. prepare the checklist + config_checklist = [] + config_checklist += [SysctlCheck('reason_1', 'decision_1', 'name_1', 'expected_1')] + config_checklist += [SysctlCheck('reason_2', 'decision_2', 'name_2', 'expected_2')] + config_checklist += [SysctlCheck('reason_3', 'decision_3', 'name_3', 'expected_3')] + config_checklist += [SysctlCheck('reason_4', 'decision_4', 'name_4', 'is not set')] + config_checklist += [SysctlCheck('reason_5', 'decision_5', 'name_5', 'is present')] + config_checklist += [SysctlCheck('reason_6', 'decision_6', 'name_6', 'is present')] + config_checklist += [SysctlCheck('reason_7', 'decision_7', 'name_7', 'is not off')] + config_checklist += [SysctlCheck('reason_8', 'decision_8', 'name_8', 'is not off')] + config_checklist += [SysctlCheck('reason_9', 'decision_9', 'name_9', 'is not off')] + config_checklist += [SysctlCheck('reason_10', 'decision_10', 'name_10', 'is not off')] + + # 2. prepare the parsed sysctl options + parsed_sysctl_options = OrderedDict() + parsed_sysctl_options['name_1'] = 'expected_1' + parsed_sysctl_options['name_2'] = 'UNexpected_2' + parsed_sysctl_options['name_5'] = '' + parsed_sysctl_options['name_7'] = '' + parsed_sysctl_options['name_8'] = 'off' + parsed_sysctl_options['name_9'] = '0' + + # 3. run the engine + self.run_engine(config_checklist, None, None, parsed_sysctl_options, None) + + # 4. check that the results are correct + result = [] + self.get_engine_result(config_checklist, result, 'json') + self.assertEqual( + result, + [["name_1", "sysctl", "expected_1", "decision_1", "reason_1", "OK"], + ["name_2", "sysctl", "expected_2", "decision_2", "reason_2", "FAIL: \"UNexpected_2\""], + ["name_3", "sysctl", "expected_3", "decision_3", "reason_3", "FAIL: is not found"], + ["name_4", "sysctl", "is not set", "decision_4", "reason_4", "OK: is not found"], + ["name_5", "sysctl", "is present", "decision_5", "reason_5", "OK: is present"], + ["name_6", "sysctl", "is present", "decision_6", "reason_6", "FAIL: is not present"], + ["name_7", "sysctl", "is not off", "decision_7", "reason_7", "OK: is not off, \"\""], + ["name_8", "sysctl", "is not off", "decision_8", "reason_8", "FAIL: is off"], + ["name_9", "sysctl", "is not off", "decision_9", "reason_9", "FAIL: is off, \"0\""], + ["name_10", "sysctl", "is not off", "decision_10", "reason_10", "FAIL: is off, not found"]] + ) + def test_complex_or(self): # 1. prepare the checklist config_checklist = [] @@ -210,7 +260,7 @@ class TestEngine(unittest.TestCase): parsed_kconfig_options['CONFIG_NAME_11'] = 'really_not_off' # 3. run the engine - self.run_engine(config_checklist, parsed_kconfig_options, None, None) + self.run_engine(config_checklist, parsed_kconfig_options, None, None, None) # 4. check that the results are correct result = [] @@ -255,7 +305,7 @@ class TestEngine(unittest.TestCase): parsed_kconfig_options['CONFIG_NAME_12'] = 'expected_12' # 3. run the engine - self.run_engine(config_checklist, parsed_kconfig_options, None, None) + self.run_engine(config_checklist, parsed_kconfig_options, None, None, None) # 4. check that the results are correct result = [] @@ -291,7 +341,7 @@ class TestEngine(unittest.TestCase): kernel_version = (42, 43) # 4. run the engine - self.run_engine(config_checklist, parsed_kconfig_options, None, kernel_version) + self.run_engine(config_checklist, parsed_kconfig_options, None, None, kernel_version) # 5. check that the results are correct result = [] @@ -320,7 +370,7 @@ class TestEngine(unittest.TestCase): parsed_cmdline_options['name_6'] = 'UNexpected_6' # 3. run the engine - self.run_engine(config_checklist, None, parsed_cmdline_options, None) + self.run_engine(config_checklist, None, parsed_cmdline_options, None, None) # 4. check that the results are correct json_result = [] @@ -378,7 +428,7 @@ name_6 |cmdline| expected_6 |decision_6| re parsed_cmdline_options['name_2'] = 'expected_2_new' # 4. run the engine - self.run_engine(config_checklist, parsed_kconfig_options, parsed_cmdline_options, None) + self.run_engine(config_checklist, parsed_kconfig_options, parsed_cmdline_options, None, None) # 5. check that the results are correct result = []