testing new features

tests/test_cli.py

@@ -1,21 +1,35 @@
 from typer.testing import CliRunner
 
-from cli.main import app
+from cli.main import MODEL_PATH, app
 
 runner = CliRunner()
 
 
+def _cleanup():
+    if MODEL_PATH.exists():
+        MODEL_PATH.unlink()
+
+
 class TestCli:
     def test_init(self):
+        _cleanup()
         result = runner.invoke(app, ["init"])
         assert result.exit_code == 0
         assert "Пустая модель" in result.stdout
 
     def test_forecast_after_init(self):
+        _cleanup()
         runner.invoke(app, ["init"])
         result = runner.invoke(app, ["forecast", "--months", "3"])
         assert result.exit_code == 0
 
+    def test_forecast_with_currency(self):
+        _cleanup()
+        runner.invoke(app, ["init"])
+        result = runner.invoke(app, ["forecast", "--months", "3", "--currency", "USD"])
+        assert result.exit_code == 0
+        assert "$" in result.stdout
+
     def test_unknown_scenario(self):
         result = runner.invoke(app, ["scenario", "unknown"])
         assert result.exit_code != 0
@@ -23,3 +37,132 @@ class TestCli:
     def test_help(self):
         result = runner.invoke(app, ["--help"])
         assert result.exit_code == 0
+
+    def test_info_empty(self):
+        _cleanup()
+        runner.invoke(app, ["init"])
+        result = runner.invoke(app, ["info"])
+        assert result.exit_code == 0
+        assert "Базовая валюта" in result.stdout
+
+    def test_config_base_currency_set(self):
+        _cleanup()
+        runner.invoke(app, ["init"])
+        result = runner.invoke(app, ["config", "base-currency", "EUR"])
+        assert result.exit_code == 0
+        assert "EUR" in result.stdout
+
+        info = runner.invoke(app, ["info"])
+        assert "EUR" in info.stdout
+
+    def test_config_base_currency_show(self):
+        _cleanup()
+        runner.invoke(app, ["init"])
+        result = runner.invoke(app, ["config", "base-currency"])
+        assert result.exit_code == 0
+        assert "RUB" in result.stdout
+
+    def test_config_account_crud(self):
+        _cleanup()
+        runner.invoke(app, ["init"])
+
+        args = ["config", "account-add", "--name", "Тестовый", "--balance", "1000"]
+        add = runner.invoke(app, args)
+        assert add.exit_code == 0
+        assert "Тестовый" in add.stdout
+
+        lst = runner.invoke(app, ["config", "account-list"])
+        assert "Тестовый" in lst.stdout
+
+        remove = runner.invoke(app, ["config", "account-remove", "Тестовый"])
+        assert remove.exit_code == 0
+
+        lst2 = runner.invoke(app, ["config", "account-list"])
+        assert "Тестовый" not in lst2.stdout
+
+    def test_config_transaction_crud(self):
+        _cleanup()
+        runner.invoke(app, ["init"])
+        runner.invoke(app, ["config", "account-add", "--name", "Счёт", "--balance", "0"])
+
+        add = runner.invoke(app, [
+            "config", "transaction-add",
+            "--account", "Счёт",
+            "--amount", "5000",
+            "--category", "income",
+            "--description", "Зарплата",
+        ])
+        assert add.exit_code == 0
+        assert "доход" in add.stdout
+
+        lst = runner.invoke(app, ["config", "transaction-list"])
+        assert "Зарплата" in lst.stdout
+
+    def test_config_asset_crud(self):
+        _cleanup()
+        runner.invoke(app, ["init"])
+
+        add = runner.invoke(app, [
+            "config", "asset-add",
+            "--name", "Квартира",
+            "--value", "5000000",
+            "--growth", "5",
+        ])
+        assert add.exit_code == 0
+        assert "Квартира" in add.stdout
+
+        lst = runner.invoke(app, ["config", "asset-list"])
+        assert "Квартира" in lst.stdout
+
+    def test_config_liability_crud(self):
+        _cleanup()
+        runner.invoke(app, ["init"])
+
+        add = runner.invoke(app, [
+            "config", "liability-add",
+            "--name", "Кредит",
+            "--balance", "100000",
+            "--interest", "10",
+            "--payment", "5000",
+        ])
+        assert add.exit_code == 0
+        assert "Кредит" in add.stdout
+
+    def test_config_recurring_crud(self):
+        _cleanup()
+        runner.invoke(app, ["init"])
+
+        add = runner.invoke(app, [
+            "config", "recurring-add",
+            "--amount", "-500",
+            "--category", "аренда",
+            "--frequency", "monthly",
+        ])
+        assert add.exit_code == 0
+
+        lst = runner.invoke(app, ["config", "recurring-list"])
+        assert "аренда" in lst.stdout
+
+    def test_config_rate_crud(self):
+        _cleanup()
+        runner.invoke(app, ["init"])
+
+        set_r = runner.invoke(app, ["config", "rate-set", "EUR", "RUB", "90"])
+        assert set_r.exit_code == 0
+
+        lst = runner.invoke(app, ["config", "rate-list"])
+        assert "EUR" in lst.stdout
+        assert "90" in lst.stdout
+
+    def test_info_with_data(self):
+        _cleanup()
+        runner.invoke(app, ["init"])
+        acc_args = ["config", "account-add", "--name", "Основной", "--balance", "50000"]
+        runner.invoke(app, acc_args)
+        asset_args = ["config", "asset-add", "--name", "Акции",
+                       "--value", "100000", "--growth", "8.0"]
+        runner.invoke(app, asset_args)
+
+        result = runner.invoke(app, ["info"])
+        assert "Основной" in result.stdout
+        assert "Акции" in result.stdout