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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
import io, json, unittest
from contextlib import redirect_stdout
from firefly_cli.output import unwrap, emit
class TestOutput(unittest.TestCase):
def test_unwrap_list_returns_clean_objects(self):
resp = {"data": [
{"id": "1", "type": "accounts", "attributes": {"name": "Checking"}},
{"id": "2", "type": "accounts", "attributes": {"name": "Savings"}},
]}
self.assertEqual(unwrap(resp),
[{"id": "1", "name": "Checking"}, {"id": "2", "name": "Savings"}])
def test_unwrap_single_object(self):
resp = {"data": {"id": "5", "type": "accounts",
"attributes": {"name": "Wallet"}}}
self.assertEqual(unwrap(resp), {"id": "5", "name": "Wallet"})
def test_emit_json_default(self):
buf = io.StringIO()
with redirect_stdout(buf):
emit([{"id": "1", "name": "x"}], human=False)
self.assertEqual(json.loads(buf.getvalue()), [{"id": "1", "name": "x"}])
def test_emit_human_table_contains_values(self):
buf = io.StringIO()
with redirect_stdout(buf):
emit([{"id": "1", "name": "Checking"}], human=True)
out = buf.getvalue()
self.assertIn("Checking", out)
self.assertIn("id", out)
def test_emit_human_drops_nested_columns(self):
# dict/list-valued fields would dump unreadable blobs; they must be cut.
buf = io.StringIO()
with redirect_stdout(buf):
emit([{"id": "1", "name": "x", "junk": {"a": 1}, "tags": [1, 2]}],
human=True)
out = buf.getvalue()
self.assertIn("name", out)
self.assertNotIn("junk", out)
self.assertNotIn("tags", out)
def test_emit_human_transaction_flattens_splits(self):
# The useful fields live in the nested `transactions` split list, and the
# raw 12-decimal amount should be trimmed to 2 dp.
tx = {"id": "77", "transactions": [{
"type": "withdrawal", "date": "2026-06-28T00:00:00+02:00",
"amount": "7.400000000000", "currency_code": "EUR",
"description": "McDonald", "source_name": "BBVA",
"destination_name": "McDonald's", "category_name": "Food"}]}
buf = io.StringIO()
with redirect_stdout(buf):
emit([tx], human=True)
out = buf.getvalue()
self.assertIn("28/06/2026", out) # Italian date, no time/zone
self.assertIn("7.40", out) # amount trimmed to 2 dp
self.assertNotIn("7.400000", out)
self.assertIn("McDonald's", out) # destination surfaced
self.assertIn("Food", out)
self.assertNotIn("import_hash", out) # raw blob not dumped
def test_emit_color_only_on_tty(self):
tx = {"id": "1", "transactions": [{"type": "withdrawal",
"date": "2026-06-28", "amount": "1", "currency_code": "EUR",
"description": "x", "source_name": "a", "destination_name": "b",
"category_name": "c"}]}
plain = io.StringIO()
emit([tx], human=True, stream=plain)
self.assertNotIn("\033[", plain.getvalue()) # piped: no ANSI
class TTY(io.StringIO):
def isatty(self):
return True
tty = TTY()
emit([tx], human=True, stream=tty)
self.assertIn("\033[31m", tty.getvalue()) # tty: withdrawal is red
|
