Testování výstupu click.echo

[vanclmil]

Ahoj, chtěl jsem se zeptat, jestli jde nějak v otestovat, zdali funkce dává správný výstup clicku. Konkrétně funkce vypíše něco stylem click.echo('cosi') a já chci udělat test na to, jestli bylo vypsáno 'cosi'. Našel jsem jen řešení stylem přes CliRunner().invoke() s result.output, což ale nechci, protože chci udělat akorát jednotkový test té funkce. Jde to nějak? Díky

[hroncok]

Fixture capsys https://docs.pytest.org/en/stable/capture.html

Taky se hodí click.unstyle, pokud píšeš barevně.

import click

def print_red_warning():
    click.secho('WARNING', fg='red')

def test_output_contains_exactly_warning(capsys):
    print_red_warning()
    out, err = capsys.readouterr()
    assert click.unstyle(out.strip()) == 'WARNING'

[vanclmil]

Jo díky, to jsem potřeboval.