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

diff --git a/beancount_extras_kris7t/plugins/selective_implicit_prices_test.py b/beancount_extras_kris7t/plugins/selective_implicit_prices_test.py
new file mode 100644
index 0000000..6ead45d
--- /dev/null
+++ b/beancount_extras_kris7t/plugins/selective_implicit_prices_test.py
@@ -0,0 +1,410 @@
+__copyright__ = "Copyright (C) 2015-2017  Martin Blais, " + \
+    "2020  Kristóf Marussy <kristof@marussy.com>"
+__license__ = "GNU GPLv2"
+
+import unittest
+
+from beancount.core.number import D
+from beancount.core import data
+from beancount.parser import cmptest
+from beancount import loader
+
+from beancount_extras_kris7t.plugins import selective_implicit_prices as implicit_prices
+
+
+class TestImplicitPrices(cmptest.TestCase):
+
+    @loader.load_doc()
+    def test_add_implicit_prices__all_cases(self, entries, _, options_map):
+        """
+        1702-04-02 commodity USD
+          implicit-prices: TRUE
+
+        2013-01-01 commodity HOOL
+          implicit-prices: TRUE
+
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Account2
+        2013-01-01 open Assets:Other
+
+        ;; An explicit price directive.
+        2013-02-01 price  USD  1.10 CAD
+
+        2013-04-01 * "A transaction with a price conversion."
+          Assets:Account1             150 USD @ 1.12 CAD
+          Assets:Other
+
+        ;; This should book at price at the cost.
+        2013-04-02 * "A transaction with a cost."
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Other
+
+        ;; This one should be IGNORED because it books against the above.
+        2013-04-03 * "A transaction with a cost that reduces an existing position"
+          Assets:Account1            -500 HOOL {520 USD}
+          Assets:Other
+
+        ;; This one should generate the price, even if it is reducing.
+        2013-04-04 * "A transaction with a cost that reduces existing position, with price"
+          Assets:Account1            -100 HOOL {520 USD} @ 530 USD
+          Assets:Other
+
+        ;; This is not reducing and should also book a price at cost.
+        2013-04-05 * "A transaction with another cost that is not reducing."
+          Assets:Account1             500 HOOL {540 USD}
+          Assets:Other
+
+        ;; The price here overrides the cost and should create an entry.
+        2013-04-06 * "A transaction with a cost and a price."
+          Assets:Account1             500 HOOL {540 USD} @ 560 USD
+          Assets:Other
+        """
+        self.assertEqual(12, len(entries))
+        new_entries, _ = implicit_prices.add_implicit_prices(entries, options_map)
+        price_entries = [entry for entry in new_entries if isinstance(entry, data.Price)]
+
+        self.assertEqualEntries("""
+        1702-04-02 commodity USD
+          implicit-prices: TRUE
+
+        2013-01-01 commodity HOOL
+          implicit-prices: TRUE
+
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Account2
+        2013-01-01 open Assets:Other
+
+        2013-02-01 price USD 1.10 CAD
+
+        2013-04-01 * "A transaction with a price conversion."
+          Assets:Account1                        150 USD                        @ 1.12 CAD
+          Assets:Other                          -168.00 CAD
+
+        2013-04-01 price USD 1.12 CAD
+
+        2013-04-02 * "A transaction with a cost."
+          Assets:Account1                      1500 HOOL     {520 USD}
+          Assets:Other                       -780000 USD
+
+        2013-04-02 price HOOL 520 USD
+
+        2013-04-03 * "A transaction with a cost that reduces an existing position"
+          Assets:Account1                      -500 HOOL     {520 USD}
+          Assets:Other                        260000 USD
+
+        2013-04-04 * "A transaction with a cost that reduces existing position, with price"
+          Assets:Account1                      -100 HOOL     {520 USD}     @ 530 USD
+          Assets:Other                         52000 USD
+
+        2013-04-04 price HOOL 530 USD
+
+        2013-04-05 * "A transaction with another cost that is not reducing."
+          Assets:Account1                       500 HOOL     {540 USD}
+          Assets:Other                       -270000 USD
+
+        2013-04-05 price HOOL 540 USD
+
+        2013-04-06 * "A transaction with a cost and a price."
+          Assets:Account1                       500 HOOL     {540 USD}     @ 560 USD
+          Assets:Other                       -270000 USD
+
+        2013-04-06 price HOOL 560 USD
+        """, new_entries)
+
+        self.assertEqual(6, len(price_entries))
+        expected_values = [(x[0], x[1], D(x[2])) for x in [
+            ('USD', 'CAD', '1.10'),
+            ('USD', 'CAD', '1.12'),
+            ('HOOL', 'USD', '520.00'),
+            ('HOOL', 'USD', '530.00'),
+            ('HOOL', 'USD', '540.00'),
+            ('HOOL', 'USD', '560.00')
+            ]]
+        for expected, price in zip(expected_values, price_entries):
+            actual = (price.currency, price.amount.currency, price.amount.number)
+            self.assertEqual(expected, actual)
+
+    @loader.load_doc()
+    def test_add_implicit_prices__other_account(self, entries, errors, options_map):
+        """
+        2013-01-01 commodity HOOL
+          implicit-prices: TRUE
+
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Account2  "NONE"
+        2013-01-01 open Assets:Other
+
+        2013-04-01 *
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Other
+
+        2013-04-02 *
+          Assets:Account2             1500 HOOL {530 USD}
+          Assets:Other
+
+        2013-04-10 * "Reduces existing position in account 1"
+          Assets:Account1            -100 HOOL {520 USD}
+          Assets:Other              52000 USD
+
+        2013-04-11 * "Does not find an existing position in account 2"
+          Assets:Account2            -200 HOOL {531 USD}
+          Assets:Other             106200 USD
+
+        """
+        new_entries, _ = implicit_prices.add_implicit_prices(entries, options_map)
+        self.assertEqualEntries("""
+        2013-01-01 commodity HOOL
+          implicit-prices: TRUE
+
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Account2  "NONE"
+        2013-01-01 open Assets:Other
+
+        2013-04-01 *
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Other             -780000 USD
+
+        2013-04-02 *
+          Assets:Account2             1500 HOOL {530 USD}
+          Assets:Other             -795000 USD
+
+        2013-04-01 price HOOL 520 USD
+
+        2013-04-02 price HOOL 530 USD
+
+        2013-04-10 * "Reduces existing position in account 1"
+          Assets:Account1                       -100 HOOL {520 USD}
+          Assets:Other                         52000 USD
+
+        2013-04-11 * "Does not find an existing position in account 2"
+          Assets:Account2                       -200 HOOL {531 USD}
+          Assets:Other                        106200 USD
+
+        ;; Because a match was not found against the inventory, a price will be added.
+        2013-04-11 price HOOL 531 USD
+
+        """, new_entries)
+
+    @loader.load_doc()
+    def test_add_implicit_prices__duplicates_on_same_transaction(self,
+                                                                 entries, _, options_map):
+        """
+        2013-01-01 commodity HOOL
+          implicit-prices: TRUE
+
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Account2
+        2013-01-01 open Assets:Other
+
+        2013-04-01 * "Allowed because of same price"
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Account2             1500 HOOL {520 USD}
+          Assets:Other
+
+        2013-04-02 * "Second one is disallowed because of different price"
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Account2             1500 HOOL {530 USD}
+          Assets:Other
+
+        """
+        new_entries, errors = implicit_prices.add_implicit_prices(entries, options_map)
+        self.assertEqual([], [type(error) for error in errors])
+        self.assertEqualEntries("""
+        2013-01-01 commodity HOOL
+          implicit-prices: TRUE
+
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Account2
+        2013-01-01 open Assets:Other
+
+        2013-04-01 * "Allowed because of same price"
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Account2             1500 HOOL {520 USD}
+          Assets:Other            -1560000 USD
+
+        2013-04-01 price HOOL 520 USD
+
+        2013-04-02 * "Second one is disallowed because of different price"
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Account2             1500 HOOL {530 USD}
+          Assets:Other            -1575000 USD
+
+        2013-04-02 price HOOL 520 USD
+        2013-04-02 price HOOL 530 USD  ;; Allowed for now.
+
+        """, new_entries)
+
+    @loader.load_doc()
+    def test_add_implicit_prices__duplicates_on_different_transactions(self,
+                                                                       entries, _,
+                                                                       options_map):
+        """
+        2013-01-01 commodity HOOL
+          implicit-prices: TRUE
+
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Account2
+        2013-01-01 open Assets:Other
+
+        2013-04-01 * "Allowed because of same price #1"
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Other
+
+        2013-04-01 * "Allowed because of same price #2"
+          Assets:Account2             1500 HOOL {520 USD}
+          Assets:Other
+
+        2013-04-02 * "Second one is disallowed because of different price #1"
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Other
+
+        2013-04-02 * "Second one is disallowed because of different price #2"
+          Assets:Account2             1500 HOOL {530 USD}
+          Assets:Other
+
+        """
+        new_entries, errors = implicit_prices.add_implicit_prices(entries, options_map)
+        self.assertEqual([], [type(error) for error in errors])
+        self.assertEqualEntries("""
+        2013-01-01 commodity HOOL
+          implicit-prices: TRUE
+
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Account2
+        2013-01-01 open Assets:Other
+
+        2013-04-01 * "Allowed because of same price #1"
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Other             -780000 USD
+
+        2013-04-01 * "Allowed because of same price #2"
+          Assets:Account2             1500 HOOL {520 USD}
+          Assets:Other             -780000 USD
+
+        2013-04-01 price HOOL 520 USD
+
+        2013-04-02 * "Second one is disallowed because of different price #1"
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Other             -780000 USD
+
+        2013-04-02 * "Second one is disallowed because of different price #2"
+          Assets:Account2             1500 HOOL {530 USD}
+          Assets:Other             -795000 USD
+
+        2013-04-02 price HOOL 520 USD
+        2013-04-02 price HOOL 530 USD  ;; Allowed for now.
+
+        """, new_entries)
+
+    @loader.load_doc()
+    def test_add_implicit_prices__duplicates_overloaded(self, entries, _, options_map):
+        """
+        2013-01-01 commodity HOOL
+          implicit-prices: TRUE
+
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Other
+
+        2013-04-01 * "Allowed, sets the price for that day"
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Other
+
+        2013-04-01 * "Will be ignored, price for the day already set"
+          Assets:Account1             1500 HOOL {530 USD}
+          Assets:Other
+
+        2013-04-01 * "Should be ignored too, price for the day already set"
+          Assets:Account1             1500 HOOL {530 USD}
+          Assets:Other
+        """
+        new_entries, errors = implicit_prices.add_implicit_prices(entries, options_map)
+        self.assertEqual([], [type(error) for error in errors])
+        self.assertEqualEntries("""
+        2013-01-01 commodity HOOL
+          implicit-prices: TRUE
+
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Other
+
+        2013-04-01 * "Allowed, sets the price for that day"
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Other             -780000 USD
+
+        2013-04-01 * "Will be ignored, price for the day already set"
+          Assets:Account1             1500 HOOL {530 USD}
+          Assets:Other             -795000 USD
+
+        2013-04-01 * "Should be ignored too, price for the day already set"
+          Assets:Account1             1500 HOOL {530 USD}
+          Assets:Other             -795000 USD
+
+        2013-04-01 price HOOL 520 USD
+        2013-04-01 price HOOL 530 USD
+
+        """, new_entries)
+
+    @loader.load_doc()
+    def test_add_implicit_prices__not_enabled(self, entries, errors, options_map):
+        """
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Other
+
+        2013-04-01 *
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Other
+        """
+        new_entries, _ = implicit_prices.add_implicit_prices(entries, options_map)
+        self.assertEqualEntries("""
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Other
+
+        2013-04-01 *
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Other             -780000 USD
+        """, new_entries)
+
+    @loader.load_doc()
+    def test_add_implicit_prices__disabled(self, entries, errors, options_map):
+        """
+        2013-01-01 commodity HOOL
+          implicit-prices: FALSE
+
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Other
+
+        2013-04-01 *
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Other
+        """
+        new_entries, _ = implicit_prices.add_implicit_prices(entries, options_map)
+        self.assertEqualEntries("""
+        2013-01-01 commodity HOOL
+          implicit-prices: FALSE
+
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Other
+
+        2013-04-01 *
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Other             -780000 USD
+        """, new_entries)
+
+    @loader.load_doc()
+    def test_add_implicit_prices__invalid(self, entries, errors, options_map):
+        """
+        2013-01-01 commodity HOOL
+          implicit-prices: "yes"
+
+        2013-01-01 open Assets:Account1
+        2013-01-01 open Assets:Other
+
+        2013-04-01 *
+          Assets:Account1             1500 HOOL {520 USD}
+          Assets:Other
+        """
+        _, new_errors = implicit_prices.add_implicit_prices(entries, options_map)
+        self.assertRegex(new_errors[0].message, '^implicit-prices must be Boolean')
+
+
+if __name__ == '__main__':
+    unittest.main()