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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
import datetime
import ledgerhelpers as m
import ledgerhelpers.legacy as mc
try:
import ledgerhelpers.journal as journal
except ImportError:
journal = None
import tests.test_base as base
import tempfile
import time
import unittest
from unittest import TestCase as T
@unittest.skipIf(journal is None, reason="ledger-python is not available on this system")
class TestJournal(T):
def test_journal_with_simple_transaction(self):
c = base.datapath("simple_transaction.dat")
j = journal.Journal.from_file(c, None)
payees = j.all_payees()
self.assertListEqual(payees, ["beer"])
accts, commos = j.accounts_and_last_commodity_for_account()
expaccts = ["Accounts:Cash", "Expenses:Drinking"]
self.assertListEqual(accts, expaccts)
self.assertEqual(commos["Expenses:Drinking"], "1.00 CHF")
def test_reload_works(self):
with tempfile.NamedTemporaryFile(mode="w") as f:
with open(base.datapath("simple_transaction.dat")) as transaction_data:
data = transaction_data.read()
f.write(data)
f.flush()
j = journal.Journal.from_file(f.name, None)
_, commos = j.accounts_and_last_commodity_for_account()
self.assertEqual(commos["Expenses:Drinking"], "1.00 CHF")
data = data.replace("CHF", "EUR")
time.sleep(0.01) # Wait a few ms to make sure st_mtime of the file changes
f.write(data)
f.flush()
_, commos = j.accounts_and_last_commodity_for_account()
self.assertEqual(commos["Expenses:Drinking"], "1.00 EUR")
def test_transactions_with_payee_match(self):
c = base.datapath("simple_transaction.dat")
j = journal.Journal.from_file(c, None)
ts = journal.transactions_with_payee("beer", j.internal_parsing())
self.assertEqual(ts[0].payee, "beer")
def test_transaction_with_zero_posting(self):
c = base.datapath("zero.dat")
j = journal.Journal.from_file(c, None)
_, commos = j.accounts_and_last_commodity_for_account()
self.assertEqual(commos["rest"], "1 USD")
class TestGenerateRecord(T):
def test_no_spurious_whitespace(self):
title = "x"
date = datetime.date(2014, 1, 1)
cleared_date = None
accountamounts = [
("assets", "56 CHF"),
("expenses", ""),
]
res = mc.generate_record(title, date, cleared_date, "", accountamounts)
self.assertListEqual(
res,
"""
2014-01-01 x
assets 56 CHF
expenses
""".splitlines())
def test_no_cleared_date_when_cleared_date_not_supplied(self):
cases = [
("2014-01-01 x", (datetime.date(2014, 1, 1), None), ""),
("2014-01-01 * x", (datetime.date(2014, 1, 1), datetime.date(2014, 1, 1)), "*"),
("2014-01-01=2015-01-01 ! x", (datetime.date(2014, 1, 1), datetime.date(2015, 1, 1)), "!"),
]
accountamounts = [("assets", "56 CHF"), ("expenses", "")]
for expected_line, (date, cleared), statechar in cases:
res = mc.generate_record("x", date, cleared, statechar, accountamounts)[1]
self.assertEqual(res, expected_line)
def test_empty_record_auto_goes_last(self):
accountamounts = [("expenses", ""), ("assets:cash", "56 CHF")]
res = mc.generate_record("x", datetime.date(2014, 1, 1),
None, "", accountamounts)
self.assertListEqual(
res,
"""
2014-01-01 x
assets:cash 56 CHF
expenses
""".splitlines())
|
