__copyright__ = 'Copyright (c) 2020 Kristóf Marussy <kristof@marussy.com>'
__license__ = 'GNU GPLv2'
import unittest
from beancount.loader import load_doc
from beancount.parser import cmptest
class DefaultToleranceTest(cmptest.TestCase):
@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)
@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)
@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)
@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)
@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()
