summaryrefslogtreecommitdiff
path: root/tests/test_krr.py
blob: fe441769c3a40c33003393bf82060f5380bd7d8d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import pytest
from typer.testing import CliRunner

from robusta_krr.main import app, load_commands

runner = CliRunner(mix_stderr=False)
load_commands()

STRATEGY_NAME = "simple"


def test_help():
    result = runner.invoke(app, [STRATEGY_NAME, "--help"])
    try:
        assert result.exit_code == 0
    except AssertionError as e:
        raise e from result.exception


@pytest.mark.parametrize("log_flag", ["-v", "-q"])
def test_run(log_flag: str):
    result = runner.invoke(app, [STRATEGY_NAME, log_flag, "--namespace", "default"])
    try:
        assert result.exit_code == 0, result.stdout
    except AssertionError as e:
        raise e from result.exception


@pytest.mark.parametrize("format", ["json", "yaml", "table", "pprint"])
@pytest.mark.parametrize("output", ["--logtostderr", "-q"])
def test_output_formats(format: str, output: str):
    result = runner.invoke(app, [STRATEGY_NAME, output, "-f", format])
    try:
        assert result.exit_code == 0, result.exc_info
    except AssertionError as e:
        raise e from result.exception