diff options
| -rw-r--r-- | robusta_krr/main.py | 5 | ||||
| -rw-r--r-- | tests/test_krr.py | 7 |
2 files changed, 8 insertions, 4 deletions
diff --git a/robusta_krr/main.py b/robusta_krr/main.py index 0412974..fdd654d 100644 --- a/robusta_krr/main.py +++ b/robusta_krr/main.py @@ -36,7 +36,7 @@ def __process_type(_T: type) -> str: return "str" # It the type is unknown, just use str and let pydantic handle it -def run() -> None: +def load_commands() -> None: for strategy_name, strategy_type in BaseStrategy.get_all().items(): # type: ignore FUNC_TEMPLATE = textwrap.dedent( """ @@ -133,6 +133,9 @@ def run() -> None: locals(), ) + +def run() -> None: + load_commands() app() diff --git a/tests/test_krr.py b/tests/test_krr.py index 30704ba..23dab01 100644 --- a/tests/test_krr.py +++ b/tests/test_krr.py @@ -9,9 +9,10 @@ import pytest import yaml from typer.testing import CliRunner -from robusta_krr import app +from robusta_krr.main import app, load_commands runner = CliRunner() +load_commands() STRATEGY_NAME = "simple" @@ -28,7 +29,7 @@ def test_help(): def test_run(log_flag: str): result = runner.invoke(app, [STRATEGY_NAME, log_flag, "--namespace", "default"]) try: - assert result.exit_code == 0 + assert result.exit_code == 0, result.stdout except AssertionError as e: raise e from result.exception @@ -37,7 +38,7 @@ def test_run(log_flag: str): def test_output_formats(format: str): result = runner.invoke(app, [STRATEGY_NAME, "-q", "-f", format, "--namespace", "default"]) try: - assert result.exit_code == 0 + assert result.exit_code == 0, result.exc_info except AssertionError as e: raise e from result.exception |