66fd7cf41c
Add the ability to print custom messages in override scripts. Adds the functions - `print` - `print_if_true` - `print_if_false` Will be used to make under-the-hood preset things more verbose.
58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
import logging
|
|
from typing import Any
|
|
from typing import Optional
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
from unit.script.conftest import single_variable_output
|
|
|
|
|
|
class TestPrintFunctions:
|
|
@pytest.mark.parametrize(
|
|
"function_str, expected_print, expected_output",
|
|
[
|
|
# print
|
|
("{%print('hi mom', True)}", "hi mom", True),
|
|
("{%print('this is great', [1, 2, 3])}", "this is great", [1, 2, 3]),
|
|
("{%print([1, 2], [3, 4])}", "[1, 2]", [3, 4]),
|
|
# print_if_true
|
|
("{%print_if_true('hi mom', True)}", "hi mom", True),
|
|
("{%print_if_true('hi mom', False)}", None, False),
|
|
# print_if_false
|
|
("{%print_if_false('hi mom', True)}", None, True),
|
|
("{%print_if_false('hi mom', False)}", "hi mom", False),
|
|
],
|
|
)
|
|
def test_print_functions(
|
|
self, function_str: str, expected_print: Optional[str], expected_output: Any
|
|
):
|
|
with patch.object(logging.Logger, "info") as mock_logger:
|
|
output = single_variable_output(function_str)
|
|
assert output == expected_output
|
|
|
|
if expected_print is not None:
|
|
assert mock_logger.call_count == 1
|
|
assert mock_logger.call_args.args[0] == expected_print
|
|
else:
|
|
assert mock_logger.call_count == 0
|
|
|
|
@pytest.mark.parametrize(
|
|
"function_str, expected_print, expected_output",
|
|
[
|
|
# print
|
|
("{%print('hi mom', True, LEVEL)}", "hi mom", True),
|
|
# print_if_true
|
|
("{%print_if_true('hi mom', True, LEVEL)}", "hi mom", True),
|
|
# print_if_false
|
|
("{%print_if_false('hi mom', False, LEVEL)}", "hi mom", False),
|
|
],
|
|
)
|
|
@pytest.mark.parametrize("level", [-1, 0, 1, 2])
|
|
def test_levels(self, function_str: str, expected_print: str, expected_output: str, level: int):
|
|
level_mapping = {-1: "debug", 0: "info", 1: "warning", 2: "error"}
|
|
with patch.object(logging.Logger, level_mapping[level]) as mock_logger:
|
|
output = single_variable_output(function_str.replace("LEVEL", str(level)))
|
|
assert output == expected_output
|
|
|
|
assert mock_logger.call_count == 1
|
|
assert mock_logger.call_args.args[0] == expected_print
|