From e0843da28eaa5828665a700b8a06f7fe1e27d2b8 Mon Sep 17 00:00:00 2001 From: "Danilo M." Date: Fri, 3 Jul 2026 17:46:25 +0200 Subject: feat(budget): delete, enable, disable Co-Authored-By: Claude Opus 4.8 --- tests/unit/test_commands_budget.py | 37 +++++++++++++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) (limited to 'tests/unit') diff --git a/tests/unit/test_commands_budget.py b/tests/unit/test_commands_budget.py index 26e9dae..0b1d89c 100644 --- a/tests/unit/test_commands_budget.py +++ b/tests/unit/test_commands_budget.py @@ -3,6 +3,7 @@ import unittest from unittest.mock import MagicMock from types import SimpleNamespace import firefly_cli.commands.budget as budget +from firefly_cli.errors import FireflyError def _ctx(client=None, resolver=None): @@ -82,5 +83,41 @@ class TestBudgetSpent(unittest.TestCase): self.assertEqual(captured["rows"][0]["spent"], "-15.00") +class TestBudgetLifecycle(unittest.TestCase): + def _resolver(self): + r = MagicMock() + r.budget.return_value = {"id": "5", "name": "Rent"} + return r + + def test_delete_requires_yes(self): + args = SimpleNamespace(ref="Rent", yes=False) + with self.assertRaises(FireflyError): + budget.cmd_delete(args, _ctx(resolver=self._resolver())) + + def test_delete_with_yes_calls_delete(self): + client = MagicMock() + args = SimpleNamespace(ref="Rent", yes=True) + rc = budget.cmd_delete(args, _ctx(client=client, resolver=self._resolver())) + self.assertEqual(rc, 0) + client.request.assert_called_with("DELETE", "/api/v1/budgets/5") + + def test_enable_sets_active_true(self): + client = MagicMock() + client.request.return_value = {"data": {"id": "5", "attributes": {}}} + args = SimpleNamespace(ref="Rent") + budget.cmd_enable(args, _ctx(client=client, resolver=self._resolver())) + method, path = client.request.call_args[0][:2] + body = client.request.call_args[1]["body"] + self.assertEqual((method, path), ("PUT", "/api/v1/budgets/5")) + self.assertTrue(body["active"]) + + def test_disable_sets_active_false(self): + client = MagicMock() + client.request.return_value = {"data": {"id": "5", "attributes": {}}} + args = SimpleNamespace(ref="Rent") + budget.cmd_disable(args, _ctx(client=client, resolver=self._resolver())) + self.assertFalse(client.request.call_args[1]["body"]["active"]) + + if __name__ == "__main__": unittest.main() -- cgit v1.2.3