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

diff --git a/beancount_extras_kris7t/plugins/default_tolerance_test.py b/beancount_extras_kris7t/plugins/default_tolerance_test.py
new file mode 100644
index 0000000..2e5c629
--- /dev/null
+++ b/beancount_extras_kris7t/plugins/default_tolerance_test.py
@@ -0,0 +1,104 @@
+__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 DefaultToleranceTest(cmptest.TestCase):
+
+    @loader.load_doc()
+    def test_account_with_tolerance(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.default_tolerance"
+
+        2020-01-01 open Assets:Checking
+          default-balance-tolerance: 10
+
+        2020-01-01 balance Assets:Checking  0 USD
+        '''
+        self.assertEqualEntries('''
+        plugin "beancount_extras_kris7t.plugins.default_tolerance"
+
+        2020-01-01 open Assets:Checking
+          default-balance-tolerance: 10
+
+        2020-01-01 balance Assets:Checking  0 ~ 10 USD
+        ''', entries)
+
+    @loader.load_doc()
+    def test_account_with_tolerance_override(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.default_tolerance"
+
+        2020-01-01 open Assets:Checking
+          default-balance-tolerance: 10
+
+        2020-01-01 balance Assets:Checking  10 ~ 20 USD
+        '''
+        self.assertEqualEntries('''
+        plugin "beancount_extras_kris7t.plugins.default_tolerance"
+
+        2020-01-01 open Assets:Checking
+          default-balance-tolerance: 10
+
+        2020-01-01 balance Assets:Checking  10 ~ 20 USD
+        ''', entries)
+
+    @loader.load_doc()
+    def test_account_with_tolerance_override_zero(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.default_tolerance"
+
+        2020-01-01 open Assets:Checking
+          default-balance-tolerance: 10
+
+        2020-01-01 balance Assets:Checking  0 ~ 0 USD
+        '''
+        self.assertEqualEntries('''
+        plugin "beancount_extras_kris7t.plugins.default_tolerance"
+
+        2020-01-01 open Assets:Checking
+          default-balance-tolerance: 10
+
+        2020-01-01 balance Assets:Checking  0 ~ 0 USD
+        ''', entries)
+
+    @loader.load_doc()
+    def test_account_without_tolerance(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.default_tolerance"
+
+        2020-01-01 open Assets:Checking
+
+        2020-01-01 balance Assets:Checking  0 USD
+
+        2020-01-02 balance Assets:Checking  10 ~ 20 USD
+        '''
+        self.assertEqualEntries('''
+        plugin "beancount_extras_kris7t.plugins.default_tolerance"
+
+        2020-01-01 open Assets:Checking
+
+        2020-01-01 balance Assets:Checking  0 USD
+
+        2020-01-02 balance Assets:Checking  10 ~ 20 USD
+        ''', entries)
+
+    @loader.load_doc(expect_errors=True)
+    def test_account_with_invalid_tolerance(self, entries, errors, options_map):
+        '''
+        plugin "beancount_extras_kris7t.plugins.default_tolerance"
+
+        2020-01-01 open Assets:Checking
+          default-balance-tolerance: TRUE
+
+        2020-01-01 balance Assets:Checking  0 USD
+        '''
+        self.assertRegex(errors[0].message, '^default-balance-tolerance must be decimal')
+
+
+if __name__ == '__main__':
+    unittest.main()