blob: 77418405ba1496fdb2375a87c3a9bbe7bc705980 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
# Copyright (C) 2026 Danilo M. <danix@danix.xyz> GPL-2.0-only
import unittest
from unittest.mock import MagicMock
from types import SimpleNamespace
import firefly_cli.commands.budget as budget
def _ctx(client=None, resolver=None):
return SimpleNamespace(client=client or MagicMock(),
resolver=resolver or MagicMock(), human=False)
class TestBudgetList(unittest.TestCase):
def test_list_defaults_to_current_month(self):
client = MagicMock()
client.request.return_value = {"data": []}
args = SimpleNamespace(start=None, end=None)
rc = budget.cmd_list(args, _ctx(client=client))
self.assertEqual(rc, 0)
_, kwargs = client.request.call_args
params = kwargs["params"]
self.assertTrue(params["start"].endswith("-01"))
self.assertEqual(params["start"][:7], params["end"][:7])
def test_list_explicit_range(self):
client = MagicMock()
client.request.return_value = {"data": []}
args = SimpleNamespace(start="2026-01-01", end="2026-01-31")
budget.cmd_list(args, _ctx(client=client))
_, kwargs = client.request.call_args
self.assertEqual(kwargs["params"],
{"start": "2026-01-01", "end": "2026-01-31"})
if __name__ == "__main__":
unittest.main()
|
