mirror of
https://github.com/chipsec/chipsec
synced 2026-06-08 13:31:00 +00:00
Add _logger.disable() to tests to reduce prints
Signed-off-by: Nathaniel Mitchell <nathaniel.p.mitchell@intel.com>
This commit is contained in:
@@ -36,11 +36,16 @@ def run_chipsec_module(csm: ChipsecMain, module_replay_file: str) -> int:
|
||||
ret = csm.run_loaded_modules()
|
||||
return ret
|
||||
|
||||
def setup_run_destroy_module(init_replay_file: str, module_str: str, module_args: str = "", module_replay_file: str = "", mock_logger = True) -> int:
|
||||
if mock_logger:
|
||||
chipsec.logger._logger = Mock()
|
||||
chipsec.logger._logger.VERBOSE = False
|
||||
chipsec.logger._logger.DEBUG = False
|
||||
def setup_run_destroy_module_with_mock_logger(init_replay_file: str, module_str: str, module_args: str = "", module_replay_file: str = "") -> int:
|
||||
chipsec.logger._logger.disable()
|
||||
chipsec.logger._logger = Mock()
|
||||
chipsec.logger._logger.VERBOSE = False
|
||||
chipsec.logger._logger.DEBUG = False
|
||||
retval = setup_run_destroy_module(init_replay_file, module_str, module_args, module_replay_file)
|
||||
chipsec.logger._logger = chipsec.logger.Logger()
|
||||
return retval
|
||||
|
||||
def setup_run_destroy_module(init_replay_file: str, module_str: str, module_args: str = "", module_replay_file: str = "") -> int:
|
||||
arg_str = f" {module_args}" if module_args else ""
|
||||
cli_cmds = f"-m {module_str}{arg_str}".strip().split(' ')
|
||||
cs._chipset = None
|
||||
@@ -48,7 +53,4 @@ def setup_run_destroy_module(init_replay_file: str, module_str: str, module_args
|
||||
csm = ChipsecMain(par, cli_cmds)
|
||||
replayHelper = rph.ReplayHelper(init_replay_file)
|
||||
csm._helper = replayHelper
|
||||
retval = run_chipsec_module(csm, module_replay_file)
|
||||
if mock_logger:
|
||||
chipsec.logger._logger = chipsec.logger.Logger()
|
||||
return retval
|
||||
return run_chipsec_module(csm, module_replay_file)
|
||||
@@ -25,7 +25,7 @@ import os
|
||||
|
||||
from chipsec.file import get_main_dir
|
||||
from chipsec.testcase import ExitCode
|
||||
from tests.modules.run_chipsec_module import setup_run_destroy_module
|
||||
from tests.modules.run_chipsec_module import setup_run_destroy_module_with_mock_logger
|
||||
|
||||
class TestTglModules(unittest.TestCase):
|
||||
def setUp(self) -> None:
|
||||
@@ -35,10 +35,10 @@ class TestTglModules(unittest.TestCase):
|
||||
def derive_filename(self, module_name:str) -> str:
|
||||
return f"{module_name.replace('.', '-')}_test.json"
|
||||
|
||||
def run_and_test_module(self, module_name:str, expected_returncode:int, mock_logger = True) -> None:
|
||||
def run_and_test_module(self, module_name:str, expected_returncode:int) -> None:
|
||||
test_recording = self.derive_filename(module_name)
|
||||
replay_file = os.path.join(self.folder_path, test_recording)
|
||||
retval = setup_run_destroy_module(self.init_replay_file, module_name, module_replay_file=replay_file, mock_logger=mock_logger)
|
||||
retval = setup_run_destroy_module_with_mock_logger(self.init_replay_file, module_name, module_replay_file=replay_file)
|
||||
self.assertEqual(retval, expected_returncode, f"Expected: {expected_returncode} but got: {retval}")
|
||||
|
||||
def test_tgl_module_bios_smi(self):
|
||||
|
||||
@@ -26,7 +26,7 @@ from chipsec_util import ChipsecUtil, parse_args
|
||||
import chipsec.logger
|
||||
|
||||
|
||||
def run_chipsec_util(csu: ChipsecUtil, util_replay_file: str) -> bool:
|
||||
def run_chipsec_util(csu: ChipsecUtil, util_replay_file: str) -> int:
|
||||
csu._cs.init(csu._platform, csu._pch, csu._helper, not csu._no_driver, csu._load_config, csu._ignore_platform)
|
||||
if util_replay_file:
|
||||
csu._helper.config_file = util_replay_file
|
||||
@@ -38,7 +38,8 @@ def run_chipsec_util(csu: ChipsecUtil, util_replay_file: str) -> bool:
|
||||
comm.tear_down()
|
||||
return comm.ExitCode
|
||||
|
||||
def setup_run_destroy_util(init_replay_file: str, util_name: str, util_args: str = "", util_replay_file: str = "") -> bool:
|
||||
def setup_run_destroy_util(init_replay_file: str, util_name: str, util_args: str = "", util_replay_file: str = "") -> int:
|
||||
chipsec.logger._logger.disable()
|
||||
chipsec.logger._logger = Mock()
|
||||
arg_str = f" {util_args}" if util_args else ""
|
||||
cli_cmds = f"{util_name}{arg_str}".strip().split(' ')
|
||||
|
||||
Reference in New Issue
Block a user