contract/contract_recurring_analytic_distribution/tests/test_contract_recurring_distribution.py

# -*- 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 odoo.tests.common as common


@common.at_install(False)
@common.post_install(True)
class TestContractRecurringDistribution(common.HttpCase):
    def setUp(self):
        super(TestContractRecurringDistribution, self).setUp()
        self.partner = self.env['res.partner'].create({'name': 'Test'})
        self.product = self.env['product.product'].create({
            'name': 'Test product',
        })
        self.account1 = self.env['account.analytic.account'].create({
            'name': 'Test account #1',
        })
        self.account2 = self.env['account.analytic.account'].create({
            'name': 'Test account #2',
        })
        self.uom = self.env.ref('product.product_uom_hour')
        self.contract = self.env['account.analytic.account'].create({
            'name': 'Test contract',
            'partner_id': self.partner.id,
            'recurring_invoices': 1,
            'recurring_interval': 1,
            'recurring_invoice_line_ids': [
                (0, 0, {'quantity': 2.0,
                        'price_unit': 100.0,
                        'name': 'Test',
                        'product_id': self.product.id,
                        'uom_id': self.uom.id})],
        })
        self.distribution = self.env['account.analytic.distribution'].create({
            'name': 'Test distribution',
            'rule_ids': [
                (0, 0, {
                    'sequence': 10,
                    'percent': 75.00,
                    'analytic_account_id': self.account1.id,
                }),
                (0, 0, {
                    'sequence': 20,
                    'percent': 25.00,
                    'analytic_account_id': self.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)