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