from click.testing import CliRunner
from trading_cli.main import cli


def test_cli_help():
    runner = CliRunner()
    result = runner.invoke(cli, ["--help"])
    assert result.exit_code == 0
    assert "Usage" in result.output


def test_cli_data_group():
    runner = CliRunner()
    result = runner.invoke(cli, ["data", "--help"])
    assert result.exit_code == 0
    assert "collect" in result.output
    assert "history" in result.output