from typer.testing import CliRunner

from cli.main import app

runner = CliRunner()


class TestCli:
    def test_init(self):
        result = runner.invoke(app, ["init"])
        assert result.exit_code == 0
        assert "Пустая модель" in result.stdout

    def test_forecast_after_init(self):
        runner.invoke(app, ["init"])
        result = runner.invoke(app, ["forecast", "--months", "3"])
        assert result.exit_code == 0

    def test_unknown_scenario(self):
        result = runner.invoke(app, ["scenario", "unknown"])
        assert result.exit_code != 0

    def test_help(self):
        result = runner.invoke(app, ["--help"])
        assert result.exit_code == 0