| #!/usr/bin/env python3 |
| # Copyright (c) 2023 Intel Corporation |
| # |
| # SPDX-License-Identifier: Apache-2.0 |
| """ |
| Tests for log_helper.py functions |
| """ |
| |
| import logging |
| import mock |
| import pytest |
| |
| from importlib import reload |
| |
| import twisterlib.log_helper |
| |
| |
| TESTDATA = [ |
| ('Windows', 'dummy message: [\'dummy\', \'command\', \'-flag\']'), |
| ('Linux', 'dummy message: dummy command -flag'), |
| ] |
| |
| @pytest.mark.parametrize( |
| 'system, expected_log', |
| TESTDATA, |
| ids=['Windows', 'Linux'] |
| ) |
| def test_log_command(caplog, system, expected_log): |
| caplog.set_level(logging.DEBUG) |
| |
| logger = logging.getLogger('dummy') |
| message = 'dummy message' |
| args = ['dummy', 'command', '-flag'] |
| |
| with mock.patch('platform.system', return_value=system): |
| reload(twisterlib.log_helper) |
| twisterlib.log_helper.log_command(logger, message, args) |
| |
| reload(twisterlib.log_helper) |
| |
| assert expected_log in caplog.text |