153 lines
5.5 KiB
Python
153 lines
5.5 KiB
Python
"""Tests for the YNAB import helpers in main.py."""
|
|
from main import (
|
|
_wf_to_ynab_millis, _build_import_ids, _convert_wf_to_ynab_transactions,
|
|
_classify_for_import, _format_amount_dollars, _preview_import,
|
|
)
|
|
|
|
|
|
def test_wf_to_ynab_millis():
|
|
# Credit card charge (no debitCreditType) -> outflow (negative)
|
|
assert _wf_to_ynab_millis({"transactionAmount": 9.11}) == -9110
|
|
# Credit card refund (debitCreditType=CREDIT) -> inflow (positive)
|
|
assert _wf_to_ynab_millis({"transactionAmount": 1.64, "debitCreditType": "CREDIT"}) == 1640
|
|
# Debit card charge (debitCreditType=DEBIT) -> outflow (negative)
|
|
assert _wf_to_ynab_millis({"transactionAmount": 194.32, "debitCreditType": "DEBIT"}) == -194320
|
|
# Debit card credit (debitCreditType=CREDIT) -> inflow (positive)
|
|
assert _wf_to_ynab_millis({"transactionAmount": 225.0, "debitCreditType": "CREDIT"}) == 225000
|
|
# Pending (no debitCreditType) -> outflow (negative)
|
|
assert _wf_to_ynab_millis({"transactionAmount": 2.48}) == -2480
|
|
print("test_wf_to_ynab_millis: PASS")
|
|
|
|
|
|
def test_build_import_ids():
|
|
amounts = [-9110, -9110, -5000]
|
|
dates = ["2026-03-18", "2026-03-18", "2026-03-17"]
|
|
ids = _build_import_ids(amounts, dates)
|
|
assert ids == [
|
|
"YNAB:-9110:2026-03-18:1",
|
|
"YNAB:-9110:2026-03-18:2",
|
|
"YNAB:-5000:2026-03-17:1",
|
|
], f"Got {ids}"
|
|
print("test_build_import_ids: PASS")
|
|
|
|
|
|
def test_format_amount_dollars():
|
|
assert _format_amount_dollars(-9110) == "$-9.11"
|
|
assert _format_amount_dollars(225000) == "$+225.00"
|
|
assert _format_amount_dollars(-194320) == "$-194.32"
|
|
print("test_format_amount_dollars: PASS")
|
|
|
|
|
|
def test_convert_wf_to_ynab_transactions():
|
|
wf_txns = [
|
|
{
|
|
"transactionAmount": 9.11,
|
|
"transactionDate": 1773817200000,
|
|
"transactionDescription": "WALGREENS #10276 MORRISVILLE NC",
|
|
"status": "POSTED",
|
|
},
|
|
{
|
|
"transactionAmount": 2.48,
|
|
"transactionDate": 1774076400000,
|
|
"transactionDescription": "CULVERS",
|
|
"status": "PENDING",
|
|
},
|
|
]
|
|
result = _convert_wf_to_ynab_transactions(wf_txns, "acct-123")
|
|
|
|
assert result[0]["amount"] == -9110
|
|
assert result[0]["cleared"] == "cleared"
|
|
assert result[0]["flag_color"] == "blue"
|
|
assert result[0]["account_id"] == "acct-123"
|
|
assert result[0]["payee_name"] == "WALGREENS #10276 MORRISVILLE NC"
|
|
assert "YNAB:-9110:" in result[0]["import_id"]
|
|
|
|
assert result[1]["amount"] == -2480
|
|
assert result[1]["cleared"] == "uncleared"
|
|
assert result[1]["flag_color"] == "blue"
|
|
print("test_convert_wf_to_ynab_transactions: PASS")
|
|
|
|
|
|
def test_classify_for_import():
|
|
wf_txns = [
|
|
{
|
|
"transactionAmount": 9.11,
|
|
"transactionDate": 1773817200000,
|
|
"transactionDescription": "WALGREENS #10276",
|
|
"status": "POSTED",
|
|
},
|
|
{
|
|
"transactionAmount": 2.48,
|
|
"transactionDate": 1774076400000,
|
|
"transactionDescription": "CULVERS",
|
|
"status": "PENDING",
|
|
},
|
|
{
|
|
"transactionAmount": 50.00,
|
|
"transactionDate": 1773817200000,
|
|
"transactionDescription": "NEW TXN",
|
|
"status": "POSTED",
|
|
},
|
|
]
|
|
ynab_ready = _convert_wf_to_ynab_transactions(wf_txns, "acct-123")
|
|
|
|
# Derive dates to build matching YNAB transactions
|
|
from main import _wf_date_to_iso
|
|
date0 = _wf_date_to_iso(1773817200000)
|
|
date1 = _wf_date_to_iso(1774076400000)
|
|
|
|
ynab_txns = [
|
|
# Uncleared txn matching posted WALGREENS -> pending_to_posted
|
|
{
|
|
"id": "ynab-1", "date": date0, "payee_name": "WALGREENS",
|
|
"category_name": None, "amount_milliCurrency": -9110, "memo": None,
|
|
"account_id": "acct-123", "account_name": "Credit", "cleared": "uncleared",
|
|
"import_id": "YNAB:-9110:" + date0 + ":1",
|
|
},
|
|
# Cleared txn with same import_id as CULVERS -> duplicate
|
|
{
|
|
"id": "ynab-2", "date": date1, "payee_name": "CULVERS",
|
|
"category_name": None, "amount_milliCurrency": -2480, "memo": None,
|
|
"account_id": "acct-123", "account_name": "Credit", "cleared": "cleared",
|
|
"import_id": "YNAB:-2480:" + date1 + ":1",
|
|
},
|
|
]
|
|
|
|
classification = _classify_for_import(ynab_ready, ynab_txns)
|
|
|
|
assert len(classification.pending_to_posted) == 1
|
|
assert classification.pending_to_posted[0][1]["id"] == "ynab-1"
|
|
|
|
assert len(classification.duplicates) == 1
|
|
assert classification.duplicates[0]["flag_color"] == "red"
|
|
|
|
assert len(classification.new) == 1
|
|
assert classification.new[0]["payee_name"] == "NEW TXN"
|
|
|
|
print("test_classify_for_import: PASS")
|
|
|
|
|
|
def test_preview():
|
|
"""Visual test - just make sure it doesn't crash."""
|
|
wf_txns = [
|
|
{"transactionAmount": 9.11, "transactionDate": 1773817200000,
|
|
"transactionDescription": "WALGREENS #10276", "status": "POSTED"},
|
|
{"transactionAmount": 50.00, "transactionDate": 1773817200000,
|
|
"transactionDescription": "NEW TXN", "status": "POSTED"},
|
|
]
|
|
ynab_ready = _convert_wf_to_ynab_transactions(wf_txns, "acct-123")
|
|
classification = _classify_for_import(ynab_ready, [])
|
|
_preview_import(classification)
|
|
print("test_preview: PASS")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_wf_to_ynab_millis()
|
|
test_build_import_ids()
|
|
test_format_amount_dollars()
|
|
test_convert_wf_to_ynab_transactions()
|
|
test_classify_for_import()
|
|
test_preview()
|
|
print("\nALL TESTS PASSED")
|
|
|