1 files changed, 124 insertions, 0 deletions

diff --git a/beancount_extras_kris7t/plugins/closing_balance_test.py b/beancount_extras_kris7t/plugins/closing_balance_test.py
new file mode 100644
index 0000000..40115bf
--- /dev/null
+++ b/beancount_extras_kris7t/plugins/closing_balance_test.py
@@ -0,0 +1,124 @@
+__copyright__ = 'Copyright (c) 2020  Kristóf Marussy <kristof@marussy.com>'
+__license__ = 'GNU GPLv2'
+
+import unittest
+
+from beancount import loader
+from beancount.parser import cmptest
+
+
+class TestClosingBalance(cmptest.TestCase):
+
+    @loader.load_doc()
+    def test_close_account_correct(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.closing_balance"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Equity:Opening-Balances
+        2020-01-01 open Equity:Closing-Balances
+
+        2020-01-01 * "Opening balances"
+          Assets:Checking  100 USD
+          Equity:Opening-Balances  -100 USD
+
+        2020-03-15 balance Assets:Checking  100 USD
+          close-to: Equity:Closing-Balances
+        '''
+        self.assertEqualEntries('''
+        plugin "beancount_extras_kris7t.plugins.closing_balance"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Equity:Opening-Balances
+        2020-01-01 open Equity:Closing-Balances
+
+        2020-01-01 * "Opening balances"
+          Assets:Checking  100 USD
+          Equity:Opening-Balances  -100 USD
+
+        2020-03-15 balance Assets:Checking  100 USD
+          close-to: Equity:Closing-Balances
+
+        2020-03-15 * "Closing Assets:Checking"
+          Assets:Checking  -100 USD
+            closing: TRUE
+          Equity:Closing-Balances  100 USD
+
+        2020-03-15 close Assets:Checking
+        ''', entries)
+
+    @loader.load_doc(expect_errors=True)
+    def test_close_account_incorrect(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.closing_balance"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Equity:Opening-Balances
+        2020-01-01 open Equity:Closing-Balances
+
+        2020-01-01 * "Opening balances"
+          Assets:Checking  100 USD
+          Equity:Opening-Balances  -100 USD
+
+        2020-03-15 balance Assets:Checking  80 USD
+          close-to: Equity:Closing-Balances
+        '''
+        self.assertEqualEntries('''
+        plugin "beancount_extras_kris7t.plugins.closing_balance"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Equity:Opening-Balances
+        2020-01-01 open Equity:Closing-Balances
+
+        2020-01-01 * "Opening balances"
+          Assets:Checking  100 USD
+          Equity:Opening-Balances  -100 USD
+
+        2020-03-15 balance Assets:Checking  80 USD
+          close-to: Equity:Closing-Balances
+
+        2020-03-15 * "Closing Assets:Checking"
+          Assets:Checking  -80 USD
+            closing: TRUE
+          Equity:Closing-Balances  80 USD
+
+        2020-03-15 close Assets:Checking
+        ''', entries)
+        self.assertRegex(errors[0].message, '^Balance failed for \'Assets:Checking\'')
+
+    @loader.load_doc()
+    def test_close_account_zero(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.closing_balance"
+
+        2020-01-01 open Assets:Checking
+
+        2020-03-15 balance Assets:Checking  0 USD
+          close-to: Equity:Closing-Balances
+        '''
+        self.assertEqualEntries('''
+        plugin "beancount_extras_kris7t.plugins.closing_balance"
+
+        2020-01-01 open Assets:Checking
+
+        2020-03-15 balance Assets:Checking  0 USD
+          close-to: Equity:Closing-Balances
+
+        2020-03-15 close Assets:Checking
+        ''', entries)
+
+    @loader.load_doc(expect_errors=True)
+    def test_invalid_close_to(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.closing_balance"
+
+        2020-01-01 open Assets:Checking
+
+        2020-03-15 balance Assets:Checking  100 USD
+          close-to: TRUE
+        '''
+        self.assertRegex(errors[0].message, '^close-to must be a string')
+
+
+if __name__ == '__main__':
+    unittest.main()