diff --git a/testing/cli_tests/console.py b/testing/cli_tests/console.py index 14a6a7f7f..2f295a90d 100644 --- a/testing/cli_tests/console.py +++ b/testing/cli_tests/console.py @@ -4,7 +4,14 @@ from rich.theme import Theme from rich.style import Style -custom_theme = Theme({"info": "bold blue", "error": "bold red", "debug": "bold blue"}) +custom_theme = Theme( + { + "info": "bold blue", + "error": "bold red", + "debug": "bold blue", + "test": "bold green", + } +) console = Console(theme=custom_theme, force_terminal=True) @@ -87,3 +94,29 @@ def debug( log_locals=log_locals, _stack_offset=_stack_offset + 1, ) + +def test( + *objects: Any, + sep: str = " ", + end: str = "\n", + style: Optional[Union[str, Style]] = None, + justify: Optional[JustifyMethod] = None, + emoji: Optional[bool] = None, + markup: Optional[bool] = None, + highlight: Optional[bool] = None, + log_locals: bool = False, + _stack_offset: int = 1, +): + console.log( + "[test]TEST[/test]", + *objects, + sep=sep, + end=end, + style=style, + justify=justify, + emoji=emoji, + markup=markup, + highlight=highlight, + log_locals=log_locals, + _stack_offset=_stack_offset + 1, + ) \ No newline at end of file diff --git a/testing/cli_tests/test_limbo_cli.py b/testing/cli_tests/test_limbo_cli.py index ddd59695e..626d7defe 100755 --- a/testing/cli_tests/test_limbo_cli.py +++ b/testing/cli_tests/test_limbo_cli.py @@ -132,7 +132,7 @@ INSERT INTO t VALUES (zeroblob(1024 - 1), zeroblob(1024 - 2), zeroblob(1024 - 3) self.shell.quit() def run_test(self, name: str, sql: str, expected: str) -> None: - console.info(f"Running test: {name}", _stack_offset=2) + console.test(f"Running test: {name}", _stack_offset=2) actual = self.shell.execute(sql) assert actual == expected, ( f"Test failed: {name}\n" @@ -152,7 +152,7 @@ INSERT INTO t VALUES (zeroblob(1024 - 1), zeroblob(1024 - 2), zeroblob(1024 - 3) # Print the test that is executing before executing the sql command # Printing later confuses the user of the code what test has actually failed if desc: - console.info(f"Testing: {desc}", _stack_offset=2) + console.test(f"Testing: {desc}", _stack_offset=2) actual = self.shell.execute(sql) assert validate(actual), f"Test failed\nSQL: {sql}\nActual:\n{repr(actual)}"