blob: 1206938979238636fb85fea8e72a95f7afed7960 (
plain)
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
|
#!/usr/bin/env python3
import unittest
import operator
from ledger import *
from datetime import *
class TransactionTestCase(unittest.TestCase):
def setUp(self):
self.journal = read_journal_from_string("""
2012-03-01 KFC
;this is a note
Expenses:Food $21.34
Assets:Cash
2012-03-02 MJT
Expenses:Museum $45.67
Assets:Cash
""")
def tearDown(self):
close_journal_files()
def testAddRemovePosts(self):
xacts = [xact for xact in self.journal]
x1_posts = [post for post in xacts[1]]
for post in x1_posts:
xacts[0].add_post(post)
xacts[1].remove_post(post)
x0_posts = [post for post in xacts[0]]
x1_posts = [post for post in xacts[1]]
self.assertEqual(len(x0_posts), 4)
self.assertEqual(len(x1_posts), 0)
def testSetNote(self):
xacts = [xact for xact in self.journal]
self.assertEqual(xacts[0].note, 'this is a note')
xacts[0].note = 'this is also a note'
self.assertEqual(xacts[0].note, 'this is also a note')
xacts[0].note += 'so is this'
self.assertEqual(xacts[0].note, 'this is also a noteso is this')
def suite():
return unittest.TestLoader().loadTestsFromTestCase(TransactionTestCase)
if __name__ == '__main__':
unittest.main()
|
