diff options
Diffstat (limited to 'tests/unit/test_output.py')
| -rw-r--r-- | tests/unit/test_output.py | 84 |
1 files changed, 84 insertions, 0 deletions
diff --git a/tests/unit/test_output.py b/tests/unit/test_output.py index c71a424..2e3e5ea 100644 --- a/tests/unit/test_output.py +++ b/tests/unit/test_output.py @@ -29,3 +29,87 @@ class TestOutput(unittest.TestCase): 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 + + def _emit(self, row): + buf = io.StringIO() + emit([row], human=True, stream=buf) + return buf.getvalue() + + def test_view_account_shows_balance_drops_plumbing(self): + # account_role signature -> the account view. + out = self._emit({"id": "6", "name": "BBVA", "type": "asset", + "account_role": "defaultAsset", + "current_balance": "1590.92", "currency_code": "EUR", + "active": True, "iban": "IT68...", "notes": "secret"}) + self.assertIn("BBVA", out) + self.assertIn("1590.92", out) + self.assertIn("currency_code", out) + self.assertNotIn("iban", out) # plumbing column dropped + self.assertNotIn("secret", out) + + def test_view_tag(self): + out = self._emit({"id": "9", "tag": "2026", "description": "yr", + "zoom_level": None, "latitude": None}) + self.assertIn("2026", out) + self.assertIn("description", out) + self.assertNotIn("zoom_level", out) + + def test_view_account_balance(self): + # The balance handler emits id+name+current_balance (no account_role). + out = self._emit({"id": "6", "name": "BBVA", + "current_balance": "1590.92"}) + self.assertIn("current_balance", out) + self.assertIn("1590.92", out) + + def test_view_category_name_only(self): + out = self._emit({"id": "2", "name": "Food", "spent": [], + "primary_currency_code": "EUR", "notes": "junk"}) + self.assertIn("Food", out) + self.assertNotIn("primary_currency_code", out) + self.assertNotIn("spent", out) |
