diff options
Diffstat (limited to 'tests/unit/test_commands_transaction.py')
| -rw-r--r-- | tests/unit/test_commands_transaction.py | 22 |
1 files changed, 18 insertions, 4 deletions
diff --git a/tests/unit/test_commands_transaction.py b/tests/unit/test_commands_transaction.py index 7366b5c..ec4e54c 100644 --- a/tests/unit/test_commands_transaction.py +++ b/tests/unit/test_commands_transaction.py @@ -274,7 +274,8 @@ class TestTxEdit(unittest.TestCase): ctx, client, resolver = make_ctx() client.request.return_value = {"data": {"id": "9", "attributes": {}}} args = MagicMock(id="9", amount="12.00", date=None, desc="fixed", - source=None, dest=None, category=None, tags=None, type=None) + source=None, dest=None, category=None, tags=None, + budget=None, type=None) rc = tx.cmd_edit(args, ctx) self.assertEqual(rc, 0) method, path = client.request.call_args[0][:2] @@ -291,7 +292,8 @@ class TestTxEdit(unittest.TestCase): }[n] client.request.return_value = {"data": {"id": "9", "attributes": {}}} args = MagicMock(id="9", amount=None, date=None, desc=None, - source="BBVA", dest="Medio", category=None, tags=None, type=None) + source="BBVA", dest="Medio", category=None, tags=None, + budget=None, type=None) tx.cmd_edit(args, ctx) split = client.request.call_args[1]["body"]["transactions"][0] self.assertEqual(split, {"source_id": "3", "destination_id": "4"}) @@ -300,7 +302,8 @@ class TestTxEdit(unittest.TestCase): ctx, client, resolver = make_ctx() client.request.return_value = {"data": {"id": "9", "attributes": {}}} args = MagicMock(id="9", amount=None, date=None, desc=None, source=None, - dest=None, category="Cat", tags="a, b", type="transfer") + dest=None, category="Cat", tags="a, b", budget=None, + type="transfer") tx.cmd_edit(args, ctx) split = client.request.call_args[1]["body"]["transactions"][0] self.assertEqual(split, @@ -311,11 +314,22 @@ class TestTxEdit(unittest.TestCase): from firefly_cli.errors import FireflyError ctx, client, _ = make_ctx() args = MagicMock(id="9", amount=None, date=None, desc=None, source=None, - dest=None, category=None, tags=None, type=None) + dest=None, category=None, tags=None, budget=None, type=None) with self.assertRaises(FireflyError): tx.cmd_edit(args, ctx) client.request.assert_not_called() + def test_edit_budget_ref_sets_budget_id(self): + ctx, client, resolver = make_ctx() + resolver.budget.return_value = {"id": "12", "name": "Groceries"} + client.request.return_value = {"data": {"id": "9", "attributes": {}}} + args = MagicMock(id="9", amount=None, date=None, desc=None, source=None, + dest=None, category=None, tags=None, budget="Groceries", + type=None) + tx.cmd_edit(args, ctx) + split = client.request.call_args[1]["body"]["transactions"][0] + self.assertEqual(split, {"budget_id": "12"}) + class TestTxDelete(unittest.TestCase): def test_delete_requires_yes(self): |