[ADD][11.0] account_spread_contract

account_spread_contract/tests/test_account_spread_contract.py

@@ -0,0 +1,105 @@
+# Copyright 2018-2019 Onestein (<https://www.onestein.eu>)
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
+
+from odoo.tools import convert_file
+from odoo.modules.module import get_module_resource
+from odoo.addons.contract.tests.test_contract import TestContractBase
+
+
+class TestAccountSpreadContract(TestContractBase):
+
+    def _load(self, module, *args):
+        convert_file(
+            self.cr,
+            'account_spread_contract',
+            get_module_resource(module, *args),
+            {}, 'init', False, 'test', self.registry._assertion_report)
+
+    def setUp(self):
+        super().setUp()
+        self._load('account', 'test', 'account_minimal_test.xml')
+
+        self.contract.recurring_next_date = '2016-02-29'
+        self.contract.recurring_invoicing_type = 'pre-paid'
+        self.contract.recurring_rule_type = 'monthly'
+
+        self.receivable_account = self.env['account.account'].search([(
+            'user_type_id',
+            '=',
+            self.env.ref('account.data_account_type_receivable').id)],
+            limit=1)
+        self.sales_journal_journal_id = self.ref(
+            'account_spread_contract.sales_journal')
+
+        self.sale_template = self.env['account.spread.template'].create({
+            'name': 'test',
+            'spread_type': 'sale',
+            'spread_account_id': self.receivable_account.id,
+            'spread_journal_id': self.sales_journal_journal_id,
+        })
+
+    def test_01_create_recurring_invoice_with_spread(self):
+        self.assertTrue(self.receivable_account)
+
+        self.assertEqual(len(self.contract.recurring_invoice_line_ids), 1)
+        contract_line = self.contract.recurring_invoice_line_ids
+        self.assertEqual(contract_line.spread_check, 'unlinked')
+
+        contract_line.spread_template_id = self.sale_template
+        self.assertEqual(contract_line.spread_check, 'linked')
+
+        self.contract.recurring_create_invoice()
+        invoice_monthly = self.env['account.invoice'].search(
+            [('contract_id', '=', self.contract.id)])
+        self.assertEqual(len(invoice_monthly), 1)
+
+        self.assertEqual(len(invoice_monthly.invoice_line_ids), 1)
+        spread = invoice_monthly.invoice_line_ids.spread_id
+        self.assertTrue(spread)
+        self.assertEqual(spread.template_id, self.sale_template)
+        self.assertEqual(contract_line.spread_template_id, self.sale_template)
+
+    def test_02_open_wizard(self):
+
+        contract_line = self.contract.recurring_invoice_line_ids
+
+        res_action = contract_line.spread_details()
+        self.assertTrue(isinstance(res_action, dict))
+        self.assertFalse(res_action.get('res_id'))
+        self.assertTrue(res_action.get('context'))
+
+        contract_line.spread_template_id = self.sale_template
+
+        res_action = contract_line.spread_details()
+        self.assertTrue(isinstance(res_action, dict))
+        self.assertTrue(res_action.get('res_id'))
+        self.assertTrue(res_action.get('context'))
+
+    def test_03_wizard_create(self):
+        my_company = self.env.user.company_id
+        contract_line = self.contract.recurring_invoice_line_ids
+        self.assertFalse(contract_line.spread_template_id)
+
+        Wizard = self.env['account.spread.contract.line.link.wizard']
+        wizard = Wizard.with_context(
+            default_contract_line_id=contract_line.id,
+            default_company_id=my_company.id,
+        ).create({
+            'spread_template_id': self.sale_template.id,
+        })
+
+        self.assertEqual(wizard.contract_line_id, contract_line)
+        self.assertEqual(wizard.contract_id, self.contract)
+        self.assertEqual(wizard.contract_type, 'sale')
+        self.assertEqual(wizard.spread_template_id, self.sale_template)
+        self.assertEqual(wizard.company_id, my_company)
+        self.assertEqual(contract_line.spread_check, 'unlinked')
+
+        wizard.confirm()
+        self.assertEqual(contract_line.spread_template_id, self.sale_template)
+        self.assertEqual(contract_line.spread_check, 'linked')
+
+        ctx = {'force_contract_line_id': contract_line.id}
+        self.sale_template.with_context(ctx).action_unlink_contract_line()
+        self.assertFalse(contract_line.spread_template_id)
+        self.assertEqual(contract_line.spread_check, 'unlinked')