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