@@ -39,45 +39,45 @@ INVALID_VALUES = {
def test_valid_values_on_postings(src_value, set_value):
txn = testutil.Transaction(postings=[
('Assets:Cash', -25),
('Expenses:General', 25, {'expenseAllocation': src_value}),
])
checker = meta_expense_allocation.MetaExpenseAllocation()
errors = checker.run(txn, txn.postings[-1])
errors = checker.run(txn, txn.postings[-1], -1)
assert not errors
assert txn.postings[-1].meta.get('expenseAllocation') == set_value
@pytest.mark.parametrize('src_value', INVALID_VALUES)
def test_invalid_values_on_postings(src_value):
assert errors
@pytest.mark.parametrize('src_value,set_value', VALID_VALUES.items())
def test_valid_values_on_transactions(src_value, set_value):
txn = testutil.Transaction(expenseAllocation=src_value, postings=[
('Expenses:General', 25),
def test_invalid_values_on_transactions(src_value):
@pytest.mark.parametrize('account', [
'Accrued:AccountsReceivable',
'Assets:Cash',
'Income:Donations',
@@ -87,13 +87,13 @@ def test_invalid_values_on_transactions(src_value):
def test_non_expense_accounts_skipped(account):
(account, -25),
('Expenses:General', 25, {'expenseAllocation': 'program'}),
errors = checker.run(txn, txn.postings[0])
errors = checker.run(txn, txn.postings[0], 0)
@pytest.mark.parametrize('account,set_value', [
('Expenses:Services:Accounting', 'administration'),
('Expenses:Services:Administration', 'administration'),
('Expenses:Services:Advocacy', 'program'),
@@ -103,13 +103,13 @@ def test_non_expense_accounts_skipped(account):
def test_default_values(account, set_value):
('Liabilites:CreditCard', -25),
(account, 25),
assert txn.postings[-1].meta['expenseAllocation'] == set_value
@pytest.mark.parametrize('date,set_value', [
(testutil.EXTREME_FUTURE_DATE, False),
(testutil.FUTURE_DATE, True),
@@ -120,10 +120,10 @@ def test_default_values(account, set_value):
def test_default_value_set_in_date_range(date, set_value):
txn = testutil.Transaction(date=date, postings=[
got_value = (txn.postings[-1].meta or {}).get('expenseAllocation')
assert bool(got_value) == bool(set_value)