Fix pytest

1 files changed, 4 insertions, 3 deletions

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