[MIG][9.0] contract_recurring_plans module

contract_recurring_analytic_distribution/tests/test_contract_recurring_distribution.py

@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*-
+# Copyright 2015 Tecnativa - Pedro M. Baeza
+# Copyright 2017 Tecnativa - Vicent Cubells
+# License AGPL-3 - See http://www.gnu.org/licenses/agpl-3.0.html
+
+import openerp.tests.common as common
+
+
+class TestContractRecurringDistribution(common.SavepointCase):
+    @classmethod
+    def setUpClass(cls):
+        super(TestContractRecurringDistribution, cls).setUpClass()
+        cls.partner = cls.env['res.partner'].create({'name': 'Test'})
+        cls.product = cls.env['product.product'].create({
+            'name': 'Test product',
+        })
+        cls.account1 = cls.env['account.analytic.account'].create({
+            'name': 'Test account #1',
+        })
+        cls.account2 = cls.env['account.analytic.account'].create({
+            'name': 'Test account #2',
+        })
+        cls.uom = cls.env.ref('product.product_uom_hour')
+        cls.contract = cls.env['account.analytic.account'].create({
+            'name': 'Test contract',
+            'partner_id': cls.partner.id,
+            'type': 'contract',
+            'recurring_invoices': 1,
+            'recurring_interval': 1,
+            'recurring_invoice_line_ids': [
+                (0, 0, {'quantity': 2.0,
+                        'price_unit': 100.0,
+                        'name': 'Test',
+                        'product_id': cls.product.id,
+                        'uom_id': cls.uom.id})],
+        })
+        cls.distribution = cls.env['account.analytic.distribution'].create({
+            'name': 'Test distribution',
+            'rule_ids': [
+                (0, 0, {
+                    'sequence': 10,
+                    'percent': 75.00,
+                    'analytic_account_id': cls.account1.id,
+                }),
+                (0, 0, {
+                    'sequence': 20,
+                    'percent': 25.00,
+                    'analytic_account_id': cls.account2.id,
+                }),
+            ]
+        })
+
+    def test_invoice_without_distribution(self):
+        self.contract.recurring_create_invoice()
+        invoice = self.env['account.invoice'].search(
+            [('partner_id', '=', self.partner.id)])
+        self.assertEqual(
+            invoice.invoice_line_ids[0].account_analytic_id, self.contract)
+
+    def test_invoice_with_distribution(self):
+        self.contract.recurring_invoice_line_ids.analytic_distribution_id = (
+            self.distribution.id)
+        self.contract.recurring_create_invoice()
+        invoice = self.env['account.invoice'].search(
+            [('partner_id', '=', self.partner.id)])
+        self.assertFalse(invoice.invoice_line_ids[0].account_analytic_id)
+        self.assertEqual(
+            invoice.invoice_line_ids[0].analytic_distribution_id,
+            self.distribution)