contract/contract_section/tests/test_contract.py

# Copyright 2016 Tecnativa - Carlos Dauden
# Copyright 2017 Tecnativa - Pedro M. Baeza
# Copyright 2018 Roel Adriaans - roel@road-support.nl
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from odoo.addons.contract.tests.test_contract import TestContractBase


class TestContract(TestContractBase):
    @classmethod
    def setUpClass(cls):
        # Reuse the tests from the contract module, but replace the
        # contract line
        super(TestContract, cls).setUpClass()
        cls.acct_line.unlink()
        cls.line_vals.update({
            'display_type': 'line_section',
        })
        cls.acct_line = cls.env['account.analytic.invoice.line'].create(
            cls.line_vals,
        )

    def test_create_invoice(self):
        """ Create contract with section"""
        self.contract.recurring_create_invoice()
        invoice_id = self.env['account.invoice'].search(
            [('contract_id', '=', self.contract.id)])
        self.assertEqual(invoice_id.invoice_line_ids.display_type,
                         self.line_vals['display_type'])