test_main.py

import pytest

from core import *
from main import find_unbalanced_matches


m = MbankTransaction(amount=5, description="ble", accounting_date="2022-10-02")
h = HledgerTransaction(amount=5, description="ble", ledger_id=5)
m5 = MbankTransaction(amount=5, description="fuj", accounting_date="2022-10-02")
h5 = HledgerTransaction(amount=5, description="fuj", ledger_id=6)
m7 = MbankTransaction(amount=7, description="bork", accounting_date="2022-10-02")
h7 = HledgerTransaction(amount=7, description="bork", ledger_id=7)


@pytest.mark.parametrize("additional_m,additional_h", [
    (set(), set()),
    ({m7}, {h7}),
])
@pytest.mark.parametrize("mbank_transactions,hledger_transactions,result", [
    (set(), set(), []),
    ({m}, set(), [TransactionsMatch(mbank_transactions=[m])]),
    (set(), {h}, [TransactionsMatch(hledger_transactions=[h])]),
    ({m}, {h}, []),
    ({m, m5}, {h}, [TransactionsMatch(mbank_transactions={m5, m}, hledger_transactions={h})]),
    ({m}, {h, h5}, [TransactionsMatch(mbank_transactions={m}, hledger_transactions={h, h5})]),
])
def test_find_unbalanched_matches(mbank_transactions, hledger_transactions, result, additional_h, additional_m):
    mbank_transactions.union(additional_m)
    hledger_transactions.union(additional_h)

    assert result == find_unbalanced_matches(mbank_transactions=mbank_transactions, hledger_transactions=hledger_transactions)


def test_mk_transaction():
    HledgerTransaction(amount=5.12, ledger_id=3, description="aaa")