Add plugins and importers from private config

The importers are missing tests, because not having any specifications
for the import formats means  we must use real, private data as test inputs

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()