# Copyright 2016 Tecnativa - Carlos Dauden # Copyright 2017 Tecnativa - Pedro M. Baeza # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl). from odoo import fields from odoo.exceptions import ValidationError from odoo.tests import common class TestContractBase(common.SavepointCase): @classmethod def setUpClass(cls): super(TestContractBase, cls).setUpClass() cls.partner = cls.env.ref('base.res_partner_2') cls.product = cls.env.ref('product.product_product_2') cls.product.taxes_id += cls.env['account.tax'].search( [('type_tax_use', '=', 'sale')], limit=1) cls.product.description_sale = 'Test description sale' cls.template_vals = { 'recurring_rule_type': 'yearly', 'recurring_interval': 12345, 'name': 'Test Contract Template', } cls.template = cls.env['account.analytic.contract'].create( cls.template_vals, ) cls.contract = cls.env['account.analytic.account'].create({ 'name': 'Test Contract', 'partner_id': cls.partner.id, 'pricelist_id': cls.partner.property_product_pricelist.id, 'recurring_invoices': True, 'date_start': '2016-02-15', 'recurring_next_date': '2016-02-29', }) cls.line_vals = { 'analytic_account_id': cls.contract.id, 'product_id': cls.product.id, 'name': 'Services from #START# to #END#', 'quantity': 1, 'uom_id': cls.product.uom_id.id, 'price_unit': 100, 'discount': 50, } cls.acct_line = cls.env['account.analytic.invoice.line'].create( cls.line_vals, ) class TestContract(TestContractBase): def _add_template_line(self, overrides=None): if overrides is None: overrides = {} vals = self.line_vals.copy() vals['analytic_account_id'] = self.template.id vals.update(overrides) return self.env['account.analytic.contract.line'].create(vals) def test_check_discount(self): with self.assertRaises(ValidationError): self.acct_line.write({'discount': 120}) def test_contract(self): self.assertAlmostEqual(self.acct_line.price_subtotal, 50.0) res = self.acct_line._onchange_product_id() self.assertIn('uom_id', res['domain']) self.acct_line.price_unit = 100.0 self.contract.partner_id = False with self.assertRaises(ValidationError): self.contract.recurring_create_invoice() self.contract.partner_id = self.partner.id self.contract.recurring_create_invoice() self.invoice_monthly = self.env['account.invoice'].search( [('contract_id', '=', self.contract.id)]) self.assertTrue(self.invoice_monthly) self.assertEqual(self.contract.recurring_next_date, '2016-03-29') self.inv_line = self.invoice_monthly.invoice_line_ids[0] self.assertTrue(self.inv_line.invoice_line_tax_ids) self.assertAlmostEqual(self.inv_line.price_subtotal, 50.0) self.assertEqual(self.contract.partner_id.user_id, self.invoice_monthly.user_id) def test_contract_daily(self): self.contract.recurring_next_date = '2016-02-29' self.contract.recurring_rule_type = 'daily' self.contract.pricelist_id = False self.contract.cron_recurring_create_invoice() invoice_daily = self.env['account.invoice'].search( [('contract_id', '=', self.contract.id)]) self.assertTrue(invoice_daily) self.assertEqual(self.contract.recurring_next_date, '2016-03-01') def test_contract_weekly(self): self.contract.recurring_next_date = '2016-02-29' self.contract.recurring_rule_type = 'weekly' self.contract.recurring_invoicing_type = 'post-paid' self.contract.recurring_create_invoice() invoices_weekly = self.env['account.invoice'].search( [('contract_id', '=', self.contract.id)]) self.assertTrue(invoices_weekly) self.assertEqual( self.contract.recurring_next_date, '2016-03-07') def test_contract_yearly(self): self.contract.recurring_next_date = '2016-02-29' self.contract.recurring_rule_type = 'yearly' self.contract.recurring_create_invoice() invoices_weekly = self.env['account.invoice'].search( [('contract_id', '=', self.contract.id)]) self.assertTrue(invoices_weekly) self.assertEqual( self.contract.recurring_next_date, '2017-02-28') def test_contract_monthly_lastday(self): self.contract.recurring_next_date = '2016-02-29' self.contract.recurring_invoicing_type = 'post-paid' self.contract.recurring_rule_type = 'monthlylastday' self.contract.recurring_create_invoice() invoices_monthly_lastday = self.env['account.invoice'].search( [('contract_id', '=', self.contract.id)]) self.assertTrue(invoices_monthly_lastday) self.assertEqual(self.contract.recurring_next_date, '2016-03-31') def test_onchange_partner_id(self): self.contract._onchange_partner_id() self.assertEqual(self.contract.pricelist_id, self.contract.partner_id.property_product_pricelist) def test_onchange_recurring_invoices(self): self.contract.recurring_next_date = False self.contract._onchange_recurring_invoices() self.assertEqual(self.contract.recurring_next_date, self.contract.date_start) def test_uom(self): uom_litre = self.env.ref('product.product_uom_litre') self.acct_line.uom_id = uom_litre.id self.acct_line._onchange_product_id() self.assertEqual(self.acct_line.uom_id, self.acct_line.product_id.uom_id) def test_onchange_product_id(self): line = self.env['account.analytic.invoice.line'].new() res = line._onchange_product_id() self.assertFalse(res['domain']['uom_id']) def test_no_pricelist(self): self.contract.pricelist_id = False self.acct_line.quantity = 2 self.assertAlmostEqual(self.acct_line.price_subtotal, 100.0) def test_check_journal(self): contract_no_journal = self.contract.copy() contract_no_journal.journal_id = False journal = self.env['account.journal'].search([('type', '=', 'sale')]) journal.write({'type': 'general'}) with self.assertRaises(ValidationError): contract_no_journal.recurring_create_invoice() def test_onchange_contract_template_id(self): """It should change the contract values to match the template.""" self.contract.contract_template_id = self.template self.contract._onchange_contract_template_id() res = { 'recurring_rule_type': self.contract.recurring_rule_type, 'recurring_interval': self.contract.recurring_interval, } del self.template_vals['name'] self.assertDictEqual(res, self.template_vals) def test_onchange_contract_template_id_lines(self): """It should create invoice lines for the contract lines.""" self.acct_line.unlink() self.line_vals['analytic_account_id'] = self.template.id self.env['account.analytic.contract.line'].create(self.line_vals) self.contract.contract_template_id = self.template self.assertFalse(self.contract.recurring_invoice_line_ids, 'Recurring lines were not removed.') self.contract._onchange_contract_template_id() del self.line_vals['analytic_account_id'] self.assertEqual(len(self.contract.recurring_invoice_line_ids), 1) for key, value in self.line_vals.items(): test_value = self.contract.recurring_invoice_line_ids[0][key] try: test_value = test_value.id except AttributeError: pass self.assertEqual(test_value, value) def test_send_mail_contract(self): result = self.contract.action_contract_send() self.assertEqual(result['res_model'], 'mail.compose.message') def test_contract_onchange_product_id_domain_blank(self): """It should return a blank UoM domain when no product.""" line = self.env['account.analytic.contract.line'].new() res = line._onchange_product_id() self.assertFalse(res['domain']['uom_id']) def test_contract_onchange_product_id_domain(self): """It should return UoM category domain.""" line = self._add_template_line() res = line._onchange_product_id() self.assertEqual( res['domain']['uom_id'][0], ('category_id', '=', self.product.uom_id.category_id.id), ) def test_contract_onchange_product_id_uom(self): """It should update the UoM for the line.""" line = self._add_template_line( {'uom_id': self.env.ref('product.product_uom_litre').id} ) line.product_id.uom_id = self.env.ref('product.product_uom_day').id line._onchange_product_id() self.assertEqual(line.uom_id, line.product_id.uom_id) def test_contract_onchange_product_id_name(self): """It should update the name for the line.""" line = self._add_template_line() line.product_id.description_sale = 'Test' line._onchange_product_id() self.assertEqual(line.name, '\n'.join([line.product_id.name, line.product_id.description_sale, ])) def test_contract_count(self): """It should return contract count.""" count = self.partner.contract_count + 2 self.contract.copy() self.contract.copy() self.assertEqual(self.partner.contract_count, count) def test_date_end(self): """It should don't create invoices from finished contract.""" AccountInvoice = self.env['account.invoice'] self.contract.date_end = '2015-12-31' with self.assertRaises(ValidationError): self.contract.recurring_create_invoice() init_count = AccountInvoice.search_count( [('contract_id', '=', self.contract.id)]) self.contract.cron_recurring_create_invoice() last_count = AccountInvoice.search_count( [('contract_id', '=', self.contract.id)]) self.assertEqual(last_count, init_count) def test_same_date_start_and_date_end(self): """It should create one invoice with same start and end date.""" AccountInvoice = self.env['account.invoice'] self.contract.date_start = self.contract.date_end = fields.Date.today() self.contract.recurring_next_date = self.contract.date_start init_count = AccountInvoice.search_count( [('contract_id', '=', self.contract.id)]) self.contract.cron_recurring_create_invoice() last_count = AccountInvoice.search_count( [('contract_id', '=', self.contract.id)]) self.assertEqual(last_count, init_count + 1) with self.assertRaises(ValidationError): self.contract.recurring_create_invoice()