Python: Transaction.remove_post should call xact_base_t::remove_post

2 files changed, 22 insertions, 7 deletions

diff --git a/src/py_xact.cc b/src/py_xact.cc
index 3474c127..5f062294 100644
--- a/src/py_xact.cc
+++ b/src/py_xact.cc
@@ -100,7 +100,7 @@ void export_xact()
          return_internal_reference<>())
 
     .def("add_post", &xact_base_t::add_post, with_custodian_and_ward<1, 2>())
-    .def("remove_post", &xact_base_t::add_post)
+    .def("remove_post", &xact_base_t::remove_post)
 
     .def("finalize", &xact_base_t::finalize)
 
diff --git a/test/python/TransactionTest.py b/test/python/TransactionTest.py
index 257eb377..6827720d 100644
--- a/test/python/TransactionTest.py
+++ b/test/python/TransactionTest.py
@@ -6,18 +6,33 @@ import operator
 from ledger import *
 from datetime import *
 
-class JournalTestCase(unittest.TestCase):
+class TransactionTestCase(unittest.TestCase):
     def setUp(self):
-        pass
+        self.journal = read_journal_from_string("""
+2012-03-01 KFC
+    Expenses:Food      $21.34
+    Assets:Cash
+2012-03-02 MJT
+    Expenses:Museum      $45.67
+    Assets:Cash
+""")
 
     def tearDown(self):
         close_journal_files()
 
-    def test_(self):
-        pass
- 
+    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 suite():
-    return unittest.TestLoader().loadTestsFromTestCase(JournalTestCase)
+    return unittest.TestLoader().loadTestsFromTestCase(TransactionTestCase)
 
 if __name__ == '__main__':
     unittest.main()