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, 326 insertions, 0 deletions

diff --git a/beancount_extras_kris7t/plugins/templates_test.py b/beancount_extras_kris7t/plugins/templates_test.py
new file mode 100644
index 0000000..5f63e6c
--- /dev/null
+++ b/beancount_extras_kris7t/plugins/templates_test.py
@@ -0,0 +1,326 @@
+__copyright__ = 'Copyright (c) 2020  Kristóf Marussy <kristof@marussy.com>'
+__license__ = 'GNU GPLv2'
+
+import unittest
+
+from beancount import loader
+from beancount.parser import cmptest
+import pytest
+
+
+class TestClosingBalance(cmptest.TestCase):
+
+    @loader.load_doc()
+    def test_use_template_simple(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+
+        2020-01-01 * "Eating out" #tag ^link
+          template: "eating-out"
+          Assets:Checking  25 USD
+          Expenses:Food  -25 USD
+
+        2020-03-15 custom "template-use" "eating-out"
+
+        2020-04-12 custom "template-use" "eating-out"
+        '''
+        self.assertEqualEntries('''
+        plugin "beancount_extras_kris7t.plugins.closing_balance"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+
+        2020-03-15 * "Eating out" #tag ^link ^template_eating-out
+          template: "eating-out"
+          Assets:Checking  25 USD
+          Expenses:Food  -25 USD
+
+        2020-04-12 * "Eating out" #tag ^link ^template_eating-out
+          template: "eating-out"
+          Assets:Checking  25 USD
+          Expenses:Food  -25 USD
+        ''', entries)
+
+    @loader.load_doc()
+    def test_use_template_metadata(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+
+        2020-01-01 * "Eating out" #tag ^link
+          template: "eating-out"
+          meta1: "data"
+          meta2: TRUE
+          Assets:Checking  25 USD
+          Expenses:Food  -25 USD
+
+        2020-03-15 custom "template-use" "eating-out"
+          meta1: "foo"
+          meta3: 3.14
+        '''
+        self.assertEqualEntries('''
+        plugin "beancount_extras_kris7t.plugins.closing_balance"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+
+        2020-03-15 * "Eating out" #tag ^link ^template_eating-out
+          template: "eating-out"
+          meta1: "foo"
+          meta2: TRUE
+          meta3: 2.14
+          Assets:Checking  25 USD
+          Expenses:Food  -25 USD
+        ''', entries)
+
+    @loader.load_doc()
+    def test_use_template_scaled(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+
+        2020-01-01 * "Eating out" #tag ^link
+          template: "eating-out"
+          Assets:Checking  1 USD
+          Expenses:Food  -1 USD
+
+        2020-03-15 custom "template-use" "eating-out" 25
+
+        2020-04-12 custom "template-use" "eating-out" 27
+        '''
+        self.assertEqualEntries('''
+        plugin "beancount_extras_kris7t.plugins.closing_balance"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+
+        2020-03-15 * "Eating out" #tag ^link ^template_eating-out
+          template: "eating-out"
+          Assets:Checking  25 USD
+          Expenses:Food  -25 USD
+
+        2020-04-12 * "Eating out" #tag ^link ^template_eating-out
+          template: "eating-out"
+          Assets:Checking  27 USD
+          Expenses:Food  -27 USD
+        ''', entries)
+
+    @loader.load_doc()
+    def test_use_template_overwritten(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+        2020-01-01 open Expenses:Tax
+
+        2020-04-01 * "Eating out" #tag ^link
+          template: "eating-out"
+          Assets:Checking  1.10 USD
+          Expenses:Food  -1 USD
+          Expenses:Tax  -0.10 USD
+
+        2020-01-01 * "Eating out" #tag ^link
+          template: "eating-out"
+          Assets:Checking  1 USD
+          Expenses:Food  -1 USD
+
+        2020-03-15 custom "template-use" "eating-out" 25
+
+        2020-04-12 custom "template-use" "eating-out" 27
+        '''
+        self.assertEqualEntries('''
+        plugin "beancount_extras_kris7t.plugins.closing_balance"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+        2020-01-01 open Expenses:Tax
+
+        2020-03-15 * "Eating out" #tag ^link ^template_eating-out
+          template: "eating-out"
+          Assets:Checking  25 USD
+          Expenses:Food  -25 USD
+
+        2020-04-12 * "Eating out" #tag ^link ^template_eating-out
+          template: "eating-out"
+          Assets:Checking  29.70 USD
+          Expenses:Food  -27 USD
+          Expenses:Tax  -2.70 USD
+        ''', entries)
+
+    @loader.load_doc(expect_errors=True)
+    def test_invalid_name(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+
+        2020-01-01 * "Eating out"
+          template: TRUE
+          Assets:Checking  25 USD
+          Expenses:Food  -25 USD
+        '''
+        self.assertRegex(errors[0].message, "^template must be a string")
+
+    @pytest.mark.xfail(reason="Empty custom directive fails in beancount.ops.pad")
+    @loader.load_doc(expect_errors=True)
+    def test_use_missing_name(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+
+        2020-01-01 * "Eating out"
+          template: "eating-out"
+          Assets:Checking  25 USD
+          Expenses:Food  -25 USD
+
+        2020-03-15 custom "template-use"
+        '''
+        self.assertRegex(errors[0].message, "^Template name missing")
+
+    @loader.load_doc(expect_errors=True)
+    def test_use_too_many_arguments(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+
+        2020-01-01 * "Eating out"
+          template: "eating-out"
+          Assets:Checking  25 USD
+          Expenses:Food  -25 USD
+
+        2020-03-15 custom "template-use" "eating-out" 2 "another"
+        '''
+        self.assertRegex(errors[0].message, "^Too many template-use arguments")
+
+    @loader.load_doc(expect_errors=True)
+    def test_use_invalid_name(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-03-15 custom "template-use" TRUE
+        '''
+        self.assertRegex(errors[0].message, "^Template name must be a string")
+
+    @loader.load_doc(expect_errors=True)
+    def test_use_unknown(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-03-15 custom "template-use" "taxi"
+        '''
+        self.assertRegex(errors[0].message, "^Unknown template")
+
+    @loader.load_doc(expect_errors=True)
+    def test_use_invalid_scale_factor(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+
+        2020-01-01 * "Eating out"
+          template: "eating-out"
+          Assets:Checking  25 USD
+          Expenses:Food  -25 USD
+
+        2020-03-15 custom "template-use" "eating-out" TRUE
+        '''
+        self.assertRegex(errors[0].message, "^Invalid scale factor")
+
+    @loader.load_doc(expect_errors=True)
+    def test_use_zero_scale_factor(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+
+        2020-01-01 * "Eating out"
+          template: "eating-out"
+          Assets:Checking  25 USD
+          Expenses:Food  -25 USD
+
+        2020-03-15 custom "template-use" "eating-out" 0.0
+        '''
+        self.assertRegex(errors[0].message, "^Scale factor must not be 0")
+
+    @loader.load_doc(expect_errors=True)
+    def test_template_delete(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+
+        2020-01-01 * "Eating out"
+          template: "eating-out"
+          Assets:Checking  25 USD
+          Expenses:Food  -25 USD
+
+        2020-03-01 custom "template-delete" "eating-out"
+
+        2020-03-15 custom "template-use" "eating-out"
+        '''
+        self.assertRegex(errors[0].message, "^Unknown template")
+
+    @pytest.mark.xfail(reason="Empty custom directive fails in beancount.ops.pad")
+    @loader.load_doc(expect_errors=True)
+    def test_template_delete_too_few_arguments(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-03-01 custom "template-delete"
+        '''
+        self.assertRegex(errors[0].message, "^template-delete takes a single argument")
+
+    @loader.load_doc(expect_errors=True)
+    def test_template_delete_too_many_arguments(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-01-01 open Assets:Checking
+        2020-01-01 open Expenses:Food
+
+        2020-01-01 * "Eating out"
+          template: "eating-out"
+          Assets:Checking  25 USD
+          Expenses:Food  -25 USD
+
+        2020-03-01 custom "template-delete" "eating-out" TRUE
+        '''
+        self.assertRegex(errors[0].message, "^template-delete takes a single argument")
+
+    @loader.load_doc(expect_errors=True)
+    def test_template_delete_invalid_argument(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-03-01 custom "template-delete" TRUE
+        '''
+        self.assertRegex(errors[0].message, "^Template name must be a string")
+
+    @loader.load_doc(expect_errors=True)
+    def test_template_delete_unknown(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.templates"
+
+        2020-03-01 custom "template-delete" "taxi"
+        '''
+        self.assertRegex(errors[0].message, "^Unknown template")
+
+
+if __name__ == '__main__':
+    unittest.main()