Fix pytest

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